Add retry logic for acquireLock in atomicPut

acquireLock might fail right after generating a
new lock reference because we are not making a SERIAL get.
atomicPut should keep retrying until it acquires the lock.

Change-Id: I7b0f85a0d0229e28a56cdd41ec69fcde8d8398fe
Issue-ID: MUSIC-148
Signed-off-by: Mohammad Salehe <salehe@cs.toronto.edu>

1 files changed, 4 insertions, 2 deletions

diff --git a/src/main/java/org/onap/music/lockingservice/cassandra/CassaLockStore.java b/src/main/java/org/onap/music/lockingservice/cassandra/CassaLockStore.java
index 51a78264..58163a8f 100644
--- a/src/main/java/org/onap/music/lockingservice/cassandra/CassaLockStore.java
+++ b/src/main/java/org/onap/music/lockingservice/cassandra/CassaLockStore.java
@@ -201,7 +201,7 @@ public class CassaLockStore {
 	 * @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.
+	 * @return the lock reference.
 	 * @throws MusicServiceException
 	 * @throws MusicQueryException
 	 */
@@ -217,11 +217,13 @@ public class CassaLockStore {
 			queryObject.appendQueryString(selectQuery);
 			ResultSet results = dsHandle.executeOneConsistencyGet(queryObject);
 			Row row = results.one();
+			if (row == null)
+			    return null;
 			String lockReference = "" + row.getLong("lockReference");
 			String createTime = row.getString("createTime");
 			String acquireTime = row.getString("acquireTime");
 
-			return new LockObject(lockReference, createTime,acquireTime);
+			return new LockObject(lockReference, createTime, acquireTime);
 		}
 		finally {
 			TimeMeasureInstance.instance().exit();