1 files changed, 90 insertions, 0 deletions
diff --git a/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/cassandra/schema/SdcSchemaUtils.java b/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/cassandra/schema/SdcSchemaUtils.java
new file mode 100644
index 0000000000..f52faa5886
--- /dev/null
+++ b/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/cassandra/schema/SdcSchemaUtils.java
@@ -0,0 +1,90 @@
+package org.openecomp.sdc.be.dao.cassandra.schema;
+
+import com.datastax.driver.core.*;
+import org.openecomp.sdc.be.config.ConfigurationManager;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.List;
+
+public class SdcSchemaUtils {
+
+    private static Logger log = LoggerFactory.getLogger(SdcSchemaUtils.class.getName());
+
+    /**
+     * the method creates the cluster object using the supplied cassandra nodes
+     * in the configuration
+     *
+     * @return cluster object our null in case of an invalid configuration
+     */
+    public static Cluster createCluster() {
+        List<String> nodes = ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().getCassandraHosts();
+        if (nodes == null) {
+            log.info("no nodes were supplied in configuration.");
+            return null;
+        }
+        log.info("connecting to node:{}.", nodes);
+        Cluster.Builder clusterBuilder = Cluster.builder();
+        nodes.forEach(host -> clusterBuilder.addContactPoint(host));
+
+        clusterBuilder.withMaxSchemaAgreementWaitSeconds(60);
+
+        boolean authenticate = ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().isAuthenticate();
+        if (authenticate) {
+            String username = ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().getUsername();
+            String password = ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().getPassword();
+            if (username == null || password == null) {
+                log.info("authentication is enabled but username or password were not supplied.");
+                return null;
+            }
+            clusterBuilder.withCredentials(username, password);
+        }
+        boolean ssl = ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().isSsl();
+        if (ssl) {
+            String truststorePath = ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().getTruststorePath();
+            String truststorePassword = ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().getTruststorePassword();
+            if (truststorePath == null || truststorePassword == null) {
+                log.info("ssl is enabled but truststorePath or truststorePassword were not supplied.");
+                return null;
+            }
+            System.setProperty("javax.net.ssl.trustStore", truststorePath);
+            System.setProperty("javax.net.ssl.trustStorePassword", truststorePassword);
+            clusterBuilder.withSSL();
+        }
+        return clusterBuilder.build();
+    }
+
+    public static boolean executeStatement(String statement) {
+        return executeStatements(statement);
+    }
+
+    public static boolean executeStatements(String ... statements) {
+        Cluster cluster = null;
+        Session session = null;
+        try {
+            cluster = createCluster();
+            if (cluster == null) {
+                return false;
+            }
+            session = cluster.connect();
+            for (String statement : statements) {
+                session.execute(statement);
+            }
+            return true;
+        } catch (RuntimeException e) {
+            log.error(String.format("could not execute statements"), e);
+            return false;
+        } finally {
+            if (session != null) {
+                session.close();
+            }
+            if (cluster != null) {
+                cluster.close();
+            }
+
+        }
+    }
+
+
+
+}