1 files changed, 189 insertions, 0 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
new file mode 100644
index 0000000000..237ec6d5fe
--- /dev/null
+++ b/catalog-be/src/main/java/org/openecomp/sdc/be/components/impl/CassandraHealthCheck.java
@@ -0,0 +1,189 @@
+package org.openecomp.sdc.be.components.impl;
+
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Map;
+import java.util.Properties;
+import java.util.Set;
+
+import javax.annotation.PostConstruct;
+
+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;
+
+@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) {
+                session.close();
+            }
+			if (cluster != null) {
+				cluster.close();
+			}
+		}
+	}
+}