diff options
Diffstat (limited to 'catalog-be/src/main/java/org/openecomp/sdc/be/components/impl/CassandraHealthCheck.java')
| -rw-r--r-- | catalog-be/src/main/java/org/openecomp/sdc/be/components/impl/CassandraHealthCheck.java | 336 |
1 files changed, 167 insertions, 169 deletions
diff --git a/catalog-be/src/main/java/org/openecomp/sdc/be/components/impl/CassandraHealthCheck.java b/catalog-be/src/main/java/org/openecomp/sdc/be/components/impl/CassandraHealthCheck.java index 1f5770856b..31db9651c6 100644 --- a/catalog-be/src/main/java/org/openecomp/sdc/be/components/impl/CassandraHealthCheck.java +++ b/catalog-be/src/main/java/org/openecomp/sdc/be/components/impl/CassandraHealthCheck.java @@ -1,5 +1,18 @@ package org.openecomp.sdc.be.components.impl; +import com.datastax.driver.core.Cluster; +import com.datastax.driver.core.KeyspaceMetadata; +import com.datastax.driver.core.Metadata; +import com.datastax.driver.core.Session; +import org.openecomp.sdc.be.config.ConfigurationManager; +import org.openecomp.sdc.be.dao.cassandra.schema.SdcSchemaUtils; +import org.openecomp.sdc.be.dao.cassandra.schema.Table; +import org.openecomp.sdc.common.util.GeneralUtility; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.stereotype.Component; + +import javax.annotation.PostConstruct; import java.io.FileInputStream; import java.io.InputStream; import java.util.ArrayList; @@ -10,178 +23,163 @@ import java.util.Map; import java.util.Properties; import java.util.Set; -import javax.annotation.PostConstruct; +@Component("cassandra-health-check") +public class CassandraHealthCheck { -import org.openecomp.sdc.be.config.ConfigurationManager; -import org.openecomp.sdc.be.dao.cassandra.schema.SdcSchemaUtils; -import org.openecomp.sdc.be.dao.cassandra.schema.Table; -import org.openecomp.sdc.common.util.GeneralUtility; -import org.slf4j.Logger; -import org.slf4j.LoggerFactory; -import org.springframework.stereotype.Component; -import com.datastax.driver.core.Cluster; -import com.datastax.driver.core.KeyspaceMetadata; -import com.datastax.driver.core.Metadata; -import com.datastax.driver.core.Session; + private static final Logger log = LoggerFactory.getLogger(CassandraHealthCheck.class); -@Component("cassandra-health-check") -public class CassandraHealthCheck { - - - private static Logger log = LoggerFactory.getLogger(CassandraHealthCheck.class.getName()); - - private String localDataCenterName = null; - - private Set<String> sdcKeyspaces = new HashSet<String>(); - - private int HC_FormulaNumber; - - @PostConstruct - private void init() { - - //Initialize local data center name - this field must be filled by DevOps - localDataCenterName = ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().getLocalDataCenter(); - - if (GeneralUtility.isEmptyString(localDataCenterName)) { - log.error("localDataCenter Name in configuration.yaml is missing."); - return; - } - - //Collect all SDC keyspaces - for (Table table : Table.values()) { - sdcKeyspaces.add(table.getTableDescription().getKeyspace()); - } - - String titanCfgFile = ConfigurationManager.getConfigurationManager().getConfiguration().getTitanCfgFile(); - Properties prop = new Properties(); - InputStream titanProp = null; - try { - //load a properties file - titanProp = new FileInputStream(titanCfgFile); - prop.load(titanProp); - //Add titan keyspace - String titanKeyspace = prop.getProperty("storage.cassandra.keyspace"); - if (!GeneralUtility.isEmptyString(titanKeyspace)) { - sdcKeyspaces.add(titanKeyspace); - } - } catch (Exception e) { - log.error("Failed to open titen.properties file , url is : {}", titanCfgFile); - } - - log.info("All sdc keyspaces are : {}", sdcKeyspaces); - - //Calculate the Formula of Health Check - Cluster cluster = null; - try { - - log.info("creating cluster for Cassandra Health Check."); - //Create cluster from nodes in cassandra configuration - cluster = SdcSchemaUtils.createCluster(); - if (cluster == null) { - log.error("Failure create cassandra cluster."); - return; - } - - Metadata metadata = cluster.getMetadata(); - - if (metadata == null) { - log.error("Failure get cassandra metadata."); - return; - } - - log.info("Cluster Metadata: {}", metadata.toString()); - List<KeyspaceMetadata> keyspaces = metadata.getKeyspaces(); - List<Integer> replactionFactorList = new ArrayList<Integer>(); - - //Collect the keyspaces Replication Factor of current localDataCenter - for (KeyspaceMetadata keyspace : keyspaces) { - - if (sdcKeyspaces.contains(keyspace.getName())) { - - log.info("keyspace : {} , replication: {}", keyspace.getName(), keyspace.getReplication()); - Map<String, String> replicationOptions = keyspace.getReplication(); - - //In 1 site with one data center - if (replicationOptions.containsKey("replication_factor")) { - replactionFactorList.add(Integer.parseInt(replicationOptions.get("replication_factor"))); - } - //In multiple sites with some data center - else if (replicationOptions.containsKey(localDataCenterName)) { - replactionFactorList.add(Integer.parseInt(replicationOptions.get(localDataCenterName))); - } - } - } - - if (replactionFactorList.size() == 0) { - log.error("Replication factor NOT found in all keyspaces"); - return; - } - - int maxReplicationFactor = Collections.max(replactionFactorList); - log.info("maxReplication Factor is: {}", maxReplicationFactor); - - int localQuorum = maxReplicationFactor/2 + 1; - log.info("localQuorum is: {}", localQuorum); - - HC_FormulaNumber = maxReplicationFactor - localQuorum; - - log.info("Health Check formula : Replication Factor – Local_Quorum = {}", HC_FormulaNumber); - - - } catch (Exception e) { - log.error("create cassandra cluster failed with exception.", e); - } finally { - if (cluster != null) { - cluster.close(); - } - } - - } - - public boolean getCassandraStatus() { - - if (GeneralUtility.isEmptyString(localDataCenterName)) { - log.error("localDataCenter Name in configuration.yaml is missing."); - return false; - } - - Cluster cluster = null; - Session session = null; - try { - log.info("creating cluster for Cassandra for monitoring."); - cluster = SdcSchemaUtils.createCluster(); - if (cluster == null) { - log.error("Failure create cassandra cluster."); - return false; - } - session = cluster.connect(); - Metadata metadata = cluster.getMetadata(); - - if (metadata == null) { - log.error("Failure get cassandra metadata."); - return false; - } - - log.info("The number of cassandra nodes is:{}", metadata.getAllHosts().size()); - - //Count the number of data center nodes that are down - Long downHostsNumber = metadata.getAllHosts().stream() - .filter(x -> x.getDatacenter().equals(localDataCenterName) && !x.isUp()).count(); - - log.info("The cassandra down nodes number is {}", downHostsNumber.toString()); - return (HC_FormulaNumber >= downHostsNumber); - - } catch (Exception e) { - log.error("create cassandra cluster failed with exception.", e); - return false; - } finally { - if (session != null) { + private String localDataCenterName = null; + + private Set<String> sdcKeyspaces = new HashSet<String>(); + + private int HC_FormulaNumber; + + @PostConstruct + private void init() { + + //Initialize local data center name - this field must be filled by DevOps + localDataCenterName = ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().getLocalDataCenter(); + + if (GeneralUtility.isEmptyString(localDataCenterName)) { + log.error("localDataCenter Name in configuration.yaml is missing."); + return; + } + + //Collect all SDC keyspaces + for (Table table : Table.values()) { + sdcKeyspaces.add(table.getTableDescription().getKeyspace()); + } + + String titanCfgFile = ConfigurationManager.getConfigurationManager().getConfiguration().getTitanCfgFile(); + Properties prop = new Properties(); + InputStream titanProp = null; + try { + //load a properties file + titanProp = new FileInputStream(titanCfgFile); + prop.load(titanProp); + //Add titan keyspace + String titanKeyspace = prop.getProperty("storage.cassandra.keyspace"); + if (!GeneralUtility.isEmptyString(titanKeyspace)) { + sdcKeyspaces.add(titanKeyspace); + } + } catch (Exception e) { + log.error("Failed to open titen.properties file , url is : {}", titanCfgFile, e); + } + + log.info("All sdc keyspaces are : {}", sdcKeyspaces); + + //Calculate the Formula of Health Check + Cluster cluster = null; + try { + + log.info("creating cluster for Cassandra Health Check."); + //Create cluster from nodes in cassandra configuration + cluster = SdcSchemaUtils.createCluster(); + if (cluster == null) { + log.error("Failure create cassandra cluster."); + return; + } + + Metadata metadata = cluster.getMetadata(); + + if (metadata == null) { + log.error("Failure get cassandra metadata."); + return; + } + + log.info("Cluster Metadata: {}", metadata); + List<KeyspaceMetadata> keyspaces = metadata.getKeyspaces(); + List<Integer> replactionFactorList = new ArrayList<Integer>(); + + //Collect the keyspaces Replication Factor of current localDataCenter + for (KeyspaceMetadata keyspace : keyspaces) { + + if (sdcKeyspaces.contains(keyspace.getName())) { + + log.info("keyspace : {} , replication: {}", keyspace.getName(), keyspace.getReplication()); + Map<String, String> replicationOptions = keyspace.getReplication(); + + //In 1 site with one data center + if (replicationOptions.containsKey("replication_factor")) { + replactionFactorList.add(Integer.parseInt(replicationOptions.get("replication_factor"))); + } + //In multiple sites with some data center + else if (replicationOptions.containsKey(localDataCenterName)) { + replactionFactorList.add(Integer.parseInt(replicationOptions.get(localDataCenterName))); + } + } + } + + if (replactionFactorList.size() == 0) { + log.error("Replication factor NOT found in all keyspaces"); + return; + } + + int maxReplicationFactor = Collections.max(replactionFactorList); + log.info("maxReplication Factor is: {}", maxReplicationFactor); + + int localQuorum = maxReplicationFactor/2 + 1; + log.info("localQuorum is: {}", localQuorum); + + HC_FormulaNumber = maxReplicationFactor - localQuorum; + + log.info("Health Check formula : Replication Factor – Local_Quorum = {}", HC_FormulaNumber); + + + } catch (Exception e) { + log.error("create cassandra cluster failed with exception.", e); + } finally { + if (cluster != null) { + cluster.close(); + } + } + + } + + public boolean getCassandraStatus() { + + if (GeneralUtility.isEmptyString(localDataCenterName)) { + log.error("localDataCenter Name in configuration.yaml is missing."); + return false; + } + + Cluster cluster = null; + Session session = null; + try { + log.info("creating cluster for Cassandra for monitoring."); + cluster = SdcSchemaUtils.createCluster(); + if (cluster == null) { + log.error("Failure create cassandra cluster."); + return false; + } + session = cluster.connect(); + Metadata metadata = cluster.getMetadata(); + + if (metadata == null) { + log.error("Failure get cassandra metadata."); + return false; + } + + log.info("The number of cassandra nodes is:{}", metadata.getAllHosts().size()); + + //Count the number of data center nodes that are down + Long downHostsNumber = metadata.getAllHosts().stream() + .filter(x -> x.getDatacenter().equals(localDataCenterName) && !x.isUp()).count(); + + log.info("The cassandra down nodes number is {}", downHostsNumber); + return HC_FormulaNumber >= downHostsNumber; + + } catch (Exception e) { + log.error("create cassandra cluster failed with exception.", e); + return false; + } finally { + if (session != null) { session.close(); } - if (cluster != null) { - cluster.close(); - } - } - } + if (cluster != null) { + cluster.close(); + } + } + } } |