add generic parameter for RowLock

Change-Id: I309b89846ca15d58ea39095d406a0b0b5de510e7
apache · Dec 21, 2019 · f424dc9 · f424dc9
1 parent 49bff5e
commit f424dc9
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 7 deletions.
diff --git a/src/main/java/com/baidu/hugegraph/concurrent/LockGroup.java b/src/main/java/com/baidu/hugegraph/concurrent/LockGroup.java
@@ -71,11 +71,14 @@ public KeyLock keyLock(String lockName, int size) {
         return (KeyLock) this.locksMap.get(lockName);
     }
 
-    public RowLock rowLock(String lockName) {
+    public <K extends Comparable<K>> RowLock<K> rowLock(String lockName) {
         if (!this.locksMap.containsKey(lockName)) {
-            this.locksMap.putIfAbsent(lockName, new RowLock());
+            this.locksMap.putIfAbsent(lockName, new RowLock<>());
         }
-        return (RowLock) this.locksMap.get(lockName);
+        Object value = this.locksMap.get(lockName);
+        @SuppressWarnings("unchecked")
+        RowLock<K> lock = (RowLock<K>) value;
+        return lock;
     }
 
     public String name() {

diff --git a/src/test/java/com/baidu/hugegraph/unit/concurrent/LockGroupTest.java b/src/test/java/com/baidu/hugegraph/unit/concurrent/LockGroupTest.java
@@ -28,8 +28,8 @@
 
 import com.baidu.hugegraph.concurrent.AtomicLock;
 import com.baidu.hugegraph.concurrent.KeyLock;
-import com.baidu.hugegraph.concurrent.RowLock;
 import com.baidu.hugegraph.concurrent.LockGroup;
+import com.baidu.hugegraph.concurrent.RowLock;
 import com.baidu.hugegraph.testutil.Assert;
 
 public class LockGroupTest {
@@ -80,9 +80,9 @@ public void testKeyLockWithSize() {
 
     @Test
     public void testRowLock() {
-        RowLock lock = this.group.rowLock("lock");
+        RowLock<?> lock = this.group.rowLock("lock");
         Assert.assertNotNull(lock);
-        RowLock lock1 = this.group.rowLock("lock");
+        RowLock<?> lock1 = this.group.rowLock("lock");
         Assert.assertSame(lock, lock1);
     }
 

diff --git a/src/test/java/com/baidu/hugegraph/unit/concurrent/RowLockTest.java b/src/test/java/com/baidu/hugegraph/unit/concurrent/RowLockTest.java
@@ -65,7 +65,7 @@ public void testRowLockMultiRows() {
     @SuppressWarnings("unchecked")
     @Test
     public void testRowLockWithMultiThreads() {
-        RowLock lock = new RowLock();
+        RowLock<Integer> lock = new RowLock<>();
         Set<String> names = new HashSet<>(THREADS_NUM);
         List<Integer> keys = new ArrayList<>(5);
         Random random = new Random();