music as a service impl v3

Change-Id: I7f96eaa48afed96d62ad7662139167e8a17d00d7
Issue-ID: MUSIC-204
Signed-off-by: srupane kondreddy <sk5300@research.att.com>

5 files changed, 174 insertions, 276 deletions

diff --git a/src/main/java/org/onap/music/datastore/CassaLockStore.java b/src/main/java/org/onap/music/datastore/CassaLockStore.java
deleted file mode 100644
index 100d93bc..00000000
--- a/src/main/java/org/onap/music/datastore/CassaLockStore.java
+++ /dev/null
@@ -1,230 +0,0 @@
-package org.onap.music.datastore;
-
-import java.util.ArrayList;
-import java.util.List;
-
-import org.onap.music.eelf.logging.EELFLoggerDelegate;
-import org.onap.music.exceptions.MusicQueryException;
-import org.onap.music.exceptions.MusicServiceException;
-
-import com.datastax.driver.core.ResultSet;
-import com.datastax.driver.core.Row;
-
-/*
- * This is the lock store that is built on top of Cassandra that is used by MUSIC to maintain lock state. 
- */
-
-public class CassaLockStore {
-	
-	private EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(CassaLockStore.class);
-	private static String table_prepend_name = "lockQ_";
-	
-	public class LockObject{
-		public String lockRef;
-		public String createTime;
-		public String acquireTime;
-		public LockObject(String lockRef, String createTime, 	String acquireTime) {
-			this.lockRef = lockRef;
-			this.acquireTime = acquireTime;
-			this.createTime = createTime;
-			
-		}
-	}
-	CassaDataStore dsHandle;
-	public CassaLockStore() {
-		dsHandle = new CassaDataStore();
-	}
-	
-	public CassaLockStore(CassaDataStore dsHandle) {
-		this.dsHandle=dsHandle;
-	}
-
-    
-	/**
-	 * 
-	 * This method creates a shadow locking table for every main table in Cassandra. This table tracks all information regarding locks. 
-	 * @param keyspace of the application. 
-	 * @param table of the application. 
-	 * @return true if the operation was successful.
-	 * @throws MusicServiceException
-	 * @throws MusicQueryException
-	 */
-	public boolean createLockQueue(String keyspace, String table) throws MusicServiceException, MusicQueryException {
-        logger.info(EELFLoggerDelegate.applicationLogger,
-                "Create lock queue/table for " +  keyspace+"."+table);
-        table = table_prepend_name+table;
-		String tabQuery = "CREATE TABLE IF NOT EXISTS "+keyspace+"."+table
-				+ " ( key text, lockReference bigint, createTime text, acquireTime text, guard bigint static, PRIMARY KEY ((key), lockReference) ) "
-				+ "WITH CLUSTERING ORDER BY (lockReference ASC);";
-		PreparedQueryObject queryObject = new PreparedQueryObject();
-		
-		queryObject.appendQueryString(tabQuery);
-		boolean result;
-		result = dsHandle.executePut(queryObject, "eventual");
-		return result;
-	}
-
-	/**
-	 * This method creates a lock reference for each invocation. The lock references are monotonically increasing timestamps.
-	 * @param keyspace of the locks.
-	 * @param table of the locks.
-	 * @param lockName is the primary key of the lock table
-	 * @return the UUID lock reference.
-	 * @throws MusicServiceException
-	 * @throws MusicQueryException
-	 */
-	public String genLockRefandEnQueue(String keyspace, String table, String lockName) throws MusicServiceException, MusicQueryException {
-        logger.info(EELFLoggerDelegate.applicationLogger,
-                "Create lock reference for " +  keyspace + "." + table + "." + lockName);
-        table = table_prepend_name+table;
-
-
-		PreparedQueryObject queryObject = new PreparedQueryObject();
-		String selectQuery = "SELECT guard FROM " + keyspace + "." + table + " WHERE key=?;";
-
-		queryObject.addValue(lockName);
-		queryObject.appendQueryString(selectQuery);
-		ResultSet gqResult = dsHandle.executeOneConsistencyGet(queryObject);
-		List<Row> latestGuardRow = gqResult.all();
-
-		long prevGuard = 0;
-		long lockRef = 1;
-        if (latestGuardRow.size() > 0) {
-        	prevGuard = latestGuardRow.get(0).getLong(0);
-        	lockRef = prevGuard + 1;
-		}
-
-        long lockEpochMillis = System.currentTimeMillis();
-
-//        System.out.println("guard(" + lockName + "): " + prevGuard + "->" + lockRef);
-		logger.info(EELFLoggerDelegate.applicationLogger,
-				"Created lock reference for " +  keyspace + "." + table + "." + lockName + ":" + lockRef);
-
-        queryObject = new PreparedQueryObject();
-		String insQuery = "BEGIN BATCH" +
-                " UPDATE " + keyspace + "." + table +
-				" SET guard=? WHERE key=? IF guard = " + (prevGuard == 0 ? "NULL" : "?") +";" +
-                " INSERT INTO " + keyspace + "." + table +
-				"(key, lockReference, createTime, acquireTime) VALUES (?,?,?,?) IF NOT EXISTS; APPLY BATCH;";
-
-        queryObject.addValue(lockRef);
-        queryObject.addValue(lockName);
-        if (prevGuard != 0)
-        	queryObject.addValue(prevGuard);
-
-        queryObject.addValue(lockName);
-        queryObject.addValue(lockRef);
-        queryObject.addValue(String.valueOf(lockEpochMillis));
-        queryObject.addValue("0");
-        queryObject.appendQueryString(insQuery);
-        boolean pResult = dsHandle.executePut(queryObject, "critical");
-		return String.valueOf(lockRef);
-	}
-	
-	/**
-	 * Returns a result set containing the list of clients waiting for a particular lock
-	 * @param keyspace
-	 * @param table
-	 * @param key
-	 * @return list of lockrefs in the queue
-	 * @throws MusicServiceException
-	 * @throws MusicQueryException
-	 */
-	public List<String> getLockQueue(String keyspace, String table, String key)
-			throws MusicServiceException, MusicQueryException {
-		logger.info(EELFLoggerDelegate.applicationLogger,
-                "Getting the queue for " +  keyspace+"."+table+"."+key);
-		table = table_prepend_name+table;
-		String selectQuery = "select * from " + keyspace + "." + table + " where key='" + key + "';";
-		PreparedQueryObject queryObject = new PreparedQueryObject();
-        queryObject.appendQueryString(selectQuery);
-		ResultSet rs = dsHandle.executeOneConsistencyGet(queryObject);
-		ArrayList<String> lockQueue = new ArrayList<>();
-		for (Row row: rs) {
-			lockQueue.add(Long.toString(row.getLong("lockReference")));
-		}
-		return lockQueue;
-	}
-	
-	
-	/**
-	 * Returns a result set containing the list of clients waiting for a particular lock
-	 * @param keyspace
-	 * @param table
-	 * @param key
-	 * @return size of lockrefs queue
-	 * @throws MusicServiceException
-	 * @throws MusicQueryException
-	 */
-	public long getLockQueueSize(String keyspace, String table, String key)
-			throws MusicServiceException, MusicQueryException {
-		logger.info(EELFLoggerDelegate.applicationLogger,
-                "Getting the queue size for " +  keyspace+"."+table+"."+key);
-		table = table_prepend_name+table;
-		String selectQuery = "select count(*) from " + keyspace + "." + table + " where key='" + key + "';";
-		PreparedQueryObject queryObject = new PreparedQueryObject();
-        queryObject.appendQueryString(selectQuery);
-		ResultSet rs = dsHandle.executeOneConsistencyGet(queryObject);
-		System.err.println(rs);
-		
-		return rs.one().getLong("count");
-	}
-	
-	
-	/**
-	 * This method returns the top of lock table/queue for the key. 
-	 * @param keyspace of the application. 
-	 * @param table of the application. 
-	 * @param key is the primary key of the application table
-	 * @return the UUID lock reference.
-	 * @throws MusicServiceException
-	 * @throws MusicQueryException
-	 */
-	public LockObject peekLockQueue(String keyspace, String table, String key) throws MusicServiceException, MusicQueryException{
-        logger.info(EELFLoggerDelegate.applicationLogger,
-                "Peek in lock table for " +  keyspace+"."+table+"."+key);
-        table = table_prepend_name+table; 
-		String selectQuery = "select * from "+keyspace+"."+table+" where key='"+key+"' LIMIT 1;";	
-        PreparedQueryObject queryObject = new PreparedQueryObject();
-        queryObject.appendQueryString(selectQuery);
-		ResultSet results = dsHandle.executeOneConsistencyGet(queryObject);
-		Row row = results.one();
-		String lockReference = "" + row.getLong("lockReference");
-		String createTime = row.getString("createTime");
-		String acquireTime = row.getString("acquireTime");
-
-		return new LockObject(lockReference, createTime,acquireTime);
-	}
-	
-	
-	/**
-	 * This method removes the lock ref from the lock table/queue for the key. 
-	 * @param keyspace of the application. 
-	 * @param table of the application. 
-	 * @param key is the primary key of the application table
-	 * @param lockReference the lock reference that needs to be dequeued.
-	 * @throws MusicServiceException
-	 * @throws MusicQueryException
-	 */	
-	public void deQueueLockRef(String keyspace, String table, String key, String lockReference) throws MusicServiceException, MusicQueryException{
-		table = table_prepend_name+table; 
-        PreparedQueryObject queryObject = new PreparedQueryObject();
-        Long lockReferenceL = Long.parseLong(lockReference);
-        String deleteQuery = "delete from "+keyspace+"."+table+" where key='"+key+"' AND lockReference ="+lockReferenceL+" IF EXISTS;";
-        queryObject.appendQueryString(deleteQuery);
-		dsHandle.executePut(queryObject, "critical");	
-	}
-	
-
-	public void updateLockAcquireTime(String keyspace, String table, String key, String lockReference) throws MusicServiceException, MusicQueryException{
-		table = table_prepend_name+table;
-        PreparedQueryObject queryObject = new PreparedQueryObject();
-        Long lockReferenceL = Long.parseLong(lockReference);
-        String updateQuery = "update "+keyspace+"."+table+" set acquireTime='"+ System.currentTimeMillis()+"' where key='"+key+"' AND lockReference = "+lockReferenceL+" IF EXISTS;";
-        queryObject.appendQueryString(updateQuery);
-		dsHandle.executePut(queryObject, "eventual");	
-
-	}
-	
-
-}
diff --git a/src/main/java/org/onap/music/datastore/Condition.java b/src/main/java/org/onap/music/datastore/Condition.java
new file mode 100644
index 00000000..23994900
--- /dev/null
+++ b/src/main/java/org/onap/music/datastore/Condition.java
@@ -0,0 +1,54 @@
+/*
+ * ============LICENSE_START==========================================
+ * org.onap.music
+ * ===================================================================
+ *  Copyright (c) 2017 AT&T Intellectual Property
+ * ===================================================================
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ * 
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ * 
+ * ============LICENSE_END=============================================
+ * ====================================================================
+ */
+package org.onap.music.datastore;
+
+import java.util.Map;
+
+import org.onap.music.eelf.logging.EELFLoggerDelegate;
+import org.onap.music.main.MusicCore;
+import org.onap.music.service.MusicCoreService;
+import org.onap.music.service.impl.MusicCassaCore;
+
+import com.datastax.driver.core.ResultSet;
+import com.datastax.driver.core.Row;
+
+
+
+    public class Condition {
+        Map<String, Object> conditions;
+        PreparedQueryObject selectQueryForTheRow;
+        private static EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(Condition.class);
+       //private static MusicCoreService musicCore = MusicCassaCore.getInstance();
+
+        public Condition(Map<String, Object> conditions, PreparedQueryObject selectQueryForTheRow) {
+            this.conditions = conditions;
+            this.selectQueryForTheRow = selectQueryForTheRow;
+        }
+
+        public boolean testCondition() throws Exception {
+            // first generate the row
+            ResultSet results = MusicCore.quorumGet(selectQueryForTheRow);
+            Row row = results.one();
+            return MusicDataStoreHandle.getDSHandle().doesRowSatisfyCondition(row, conditions);
+        }
+    }
+
diff --git a/src/main/java/org/onap/music/datastore/CassaDataStore.java b/src/main/java/org/onap/music/datastore/MusicDataStore.java
index a56cf63a..4c4e2b8e 100644
--- a/src/main/java/org/onap/music/datastore/CassaDataStore.java
+++ b/src/main/java/org/onap/music/datastore/MusicDataStore.java
@@ -76,7 +76,7 @@ import com.datastax.driver.core.exceptions.NoHostAvailableException;
  * @author bharathb
  *
  */
-public class CassaDataStore {
+public class MusicDataStore {
 
     public static final String CONSISTENCY_LEVEL_ONE = "ONE";
     public static final String CONSISTENCY_LEVEL_QUORUM = "QUORUM";
@@ -107,12 +107,12 @@ public class CassaDataStore {
 
 
 
-    private EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(CassaDataStore.class);
+    private EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(MusicDataStore.class);
 
     /**
      * 
      */
-    public CassaDataStore() {
+    public MusicDataStore() {
         connectToCassaCluster();
     }
 
@@ -121,7 +121,7 @@ public class CassaDataStore {
      * @param cluster
      * @param session
      */
-    public CassaDataStore(Cluster cluster, Session session) {
+    public MusicDataStore(Cluster cluster, Session session) {
         this.session = session;
         this.cluster = cluster;
     }
@@ -131,7 +131,7 @@ public class CassaDataStore {
      * @param remoteIp
      * @throws MusicServiceException
      */
-    public CassaDataStore(String remoteIp) {
+    public MusicDataStore(String remoteIp) {
         try {
             connectToCassaCluster(remoteIp);
         } catch (MusicServiceException e) {
diff --git a/src/main/java/org/onap/music/datastore/MusicDataStoreHandle.java b/src/main/java/org/onap/music/datastore/MusicDataStoreHandle.java
new file mode 100644
index 00000000..a2d9386b
--- /dev/null
+++ b/src/main/java/org/onap/music/datastore/MusicDataStoreHandle.java
@@ -0,0 +1,115 @@
+/*
+ * ============LICENSE_START==========================================
+ * org.onap.music
+ * ===================================================================
+ *  Copyright (c) 2017 AT&T Intellectual Property
+ * ===================================================================
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ * 
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ * 
+ * ============LICENSE_END=============================================
+ * ====================================================================
+ */
+package org.onap.music.datastore;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import org.onap.music.eelf.logging.EELFLoggerDelegate;
+import org.onap.music.exceptions.MusicServiceException;
+import org.onap.music.main.MusicUtil;
+import org.onap.music.service.impl.MusicCassaCore;
+
+import com.datastax.driver.core.ResultSet;
+import com.datastax.driver.core.TableMetadata;
+
+public class MusicDataStoreHandle {
+	
+	 public static MusicDataStore mDstoreHandle = null;
+	 private static EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(MusicDataStoreHandle.class);
+	
+    /**
+     * 
+     * @param remoteIp
+     * @return
+     */
+    public static MusicDataStore getDSHandle(String remoteIp) {
+        logger.info(EELFLoggerDelegate.applicationLogger,"Acquiring data store handle");
+        long start = System.currentTimeMillis();
+        if (mDstoreHandle == null) {
+        	try {
+    			MusicUtil.loadProperties();
+    		} catch (Exception e) {
+    			logger.error(EELFLoggerDelegate.errorLogger, "No properties file defined. Falling back to default.");
+    		}
+            mDstoreHandle = new MusicDataStore(remoteIp);
+        }
+        long end = System.currentTimeMillis();
+        logger.info(EELFLoggerDelegate.applicationLogger,"Time taken to acquire data store handle:" + (end - start) + " ms");
+        return mDstoreHandle;
+    }
+
+    /**
+     * 
+     * @return
+     * @throws MusicServiceException 
+     */
+    public static MusicDataStore getDSHandle() throws MusicServiceException {
+		
+        logger.info(EELFLoggerDelegate.applicationLogger,"Acquiring data store handle");
+        long start = System.currentTimeMillis();
+        if (mDstoreHandle == null) {
+        	try {
+    			MusicUtil.loadProperties();
+    		} catch (Exception e) {
+    			logger.error(EELFLoggerDelegate.errorLogger, "No properties file defined. Falling back to default.");
+    		}
+            // Quick Fix - Best to put this into every call to getDSHandle?
+            if (! MusicUtil.getMyCassaHost().equals("localhost") ) {
+                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.applicationLogger,"Time taken to acquire data store handle:" + (end - start) + " ms");
+        return mDstoreHandle;
+    }
+    
+    /**
+     * 
+     * @param keyspace
+     * @param tablename
+     * @return
+     * @throws MusicServiceException 
+     */
+    public static TableMetadata returnColumnMetadata(String keyspace, String tablename) throws MusicServiceException {
+        return MusicDataStoreHandle.getDSHandle().returnColumnMetadata(keyspace, tablename);
+    }
+    
+    /**
+     * 
+     * @param results
+     * @return
+     * @throws MusicServiceException 
+     */
+    public static Map<String, HashMap<String, Object>> marshallResults(ResultSet results) throws MusicServiceException {
+        return MusicDataStoreHandle.getDSHandle().marshalData(results);
+    }
+
+}
diff --git a/src/main/java/org/onap/music/datastore/MusicLockState.java b/src/main/java/org/onap/music/datastore/MusicLockState.java
deleted file mode 100644
index 60a15b13..00000000
--- a/src/main/java/org/onap/music/datastore/MusicLockState.java
+++ /dev/null
@@ -1,41 +0,0 @@
-package org.onap.music.datastore;
-public class MusicLockState {
-    public enum LockStatus {
-        UNLOCKED, BEING_LOCKED, LOCKED
-    };// captures the state of the lock
-    
-    private LockStatus lockStatus;
-    private String lockHolder;
-    private String errorMessage = null;
-    
-    public String getErrorMessage() {
-		return errorMessage;
-	}
-
-	public void setErrorMessage(String errorMessage) {
-		this.errorMessage = errorMessage;
-	}
-
-	public MusicLockState(LockStatus lockStatus, String lockHolder) {
-		super();
-        this.lockStatus = lockStatus;
-		this.lockHolder = lockHolder;
-	}
-
-	public LockStatus getLockStatus() {
-        return lockStatus;
-    }
-
-    public void setLockStatus(LockStatus lockStatus) {
-        this.lockStatus = lockStatus;
-    }
-
-    public String getLockHolder() {
-        return lockHolder;
-    }
-
-    public void setLockHolder(String lockHolder) {
-        this.lockHolder = lockHolder;
-    }
-
-}