1 files changed, 51 insertions, 39 deletions
diff --git a/src/main/java/org/onap/music/datastore/MusicDataStoreHandle.java b/src/main/java/org/onap/music/datastore/MusicDataStoreHandle.java
index dc1c43a8..b4412bf6 100644
--- a/src/main/java/org/onap/music/datastore/MusicDataStoreHandle.java
+++ b/src/main/java/org/onap/music/datastore/MusicDataStoreHandle.java
@@ -32,6 +32,7 @@ import org.onap.music.eelf.logging.EELFLoggerDelegate;
 import org.onap.music.exceptions.MusicServiceException;
 import org.onap.music.main.MusicUtil;
 
+import com.datastax.driver.core.KeyspaceMetadata;
 import com.datastax.driver.core.ResultSet;
 import com.datastax.driver.core.TableMetadata;
 
@@ -49,45 +50,45 @@ public class MusicDataStoreHandle {
     * @param remoteIp
     * @return
     */
-   public static MusicDataStore getDSHandle(String remoteIp) {
-       logger.info(EELFLoggerDelegate.metricsLogger,"Acquiring data store handle");
-       long start = System.currentTimeMillis();
-       if (mDstoreHandle == null) {
-           mDstoreHandle = new MusicDataStore(remoteIp);
-       }
-       long end = System.currentTimeMillis();
-       logger.info(EELFLoggerDelegate.metricsLogger,"Time taken to acquire data store handle:" + (end - start) + " ms");
-       return mDstoreHandle;
-   }
+    public static MusicDataStore getDSHandle(String remoteIp) {
+        logger.info(EELFLoggerDelegate.metricsLogger,"Acquiring data store handle");
+        long start = System.currentTimeMillis();
+        if (mDstoreHandle == null) {
+            mDstoreHandle = new MusicDataStore(remoteIp);
+        }
+        long end = System.currentTimeMillis();
+        logger.info(EELFLoggerDelegate.metricsLogger,"Time taken to acquire data store handle:" + (end - start) + " ms");
+        return mDstoreHandle;
+    }
 
    /**
     *
     * @return
     * @throws MusicServiceException
     */
-   public static MusicDataStore getDSHandle() throws MusicServiceException {
-       logger.info(EELFLoggerDelegate.metricsLogger,"Acquiring data store handle");
-       long start = System.currentTimeMillis();
-       if (mDstoreHandle == null) {
-           // Quick Fix - Best to put this into every call to getDSHandle?
+    public static MusicDataStore getDSHandle() throws MusicServiceException {
+        logger.info(EELFLoggerDelegate.metricsLogger,"Acquiring data store handle");
+        long start = System.currentTimeMillis();
+        if (mDstoreHandle == null) {
+            // Quick Fix - Best to put this into every call to getDSHandle?
            if (!"localhost".equals(MusicUtil.getMyCassaHost())) {
-               mDstoreHandle = new MusicDataStore(MusicUtil.getMyCassaHost());
-           } else {
-               mDstoreHandle = new MusicDataStore();
-           }
-       }
-       if(mDstoreHandle.getSession() == null) {
-           String message = "Connection to Cassandra has not been enstablished."
-                   + " Please check connection properites and reboot.";
-           logger.info(EELFLoggerDelegate.applicationLogger, message);
-           throw new MusicServiceException(message);
-       }
-       long end = System.currentTimeMillis();
-       logger.info(EELFLoggerDelegate.metricsLogger,"Time taken to acquire data store handle:" + (end - start) + " ms");
-       return mDstoreHandle;
-   }
-   
-   
+                mDstoreHandle = new MusicDataStore(MusicUtil.getMyCassaHost());
+            } else {
+                mDstoreHandle = new MusicDataStore();
+            }
+        }
+        if(mDstoreHandle.getSession() == null) {
+            String message = "Connection to Cassandra has not been enstablished."
+                + " Please check connection properites and reboot.";
+            logger.info(EELFLoggerDelegate.applicationLogger, message);
+            throw new MusicServiceException(message);
+        }
+        long end = System.currentTimeMillis();
+        logger.info(EELFLoggerDelegate.metricsLogger,"Time taken to acquire data store handle:" + (end - start) + " ms");
+        return mDstoreHandle;
+    }
+
+
    /**
    *
    * @param keyspace
@@ -95,18 +96,29 @@ public class MusicDataStoreHandle {
    * @return
    * @throws MusicServiceException
    */
-  public static TableMetadata returnColumnMetadata(String keyspace, String tablename) throws MusicServiceException {
-      return getDSHandle().returnColumnMetadata(keyspace, tablename);
-  }
-  
+    public static TableMetadata returnColumnMetadata(String keyspace, String tablename) throws MusicServiceException {
+        return getDSHandle().returnColumnMetadata(keyspace, tablename);
+    }
+    
+    /**
+    *
+    * @param keyspace
+    * @param tablename
+    * @return
+    * @throws MusicServiceException
+    */
+     public static KeyspaceMetadata returnkeyspaceMetadata(String keyspace) throws MusicServiceException {
+         return getDSHandle().returnKeyspaceMetadata(keyspace);
+     }
+
   /**
   *
   * @param results
   * @return
   * @throws MusicServiceException
   */
- public static Map<String, HashMap<String, Object>> marshallResults(ResultSet results) throws MusicServiceException {
-     return getDSHandle().marshalData(results);
- }
+    public static Map<String, HashMap<String, Object>> marshallResults(ResultSet results) throws MusicServiceException {
+        return getDSHandle().marshalData(results);
+    }
 
 }