add slist.getMember for async client (#340)

distkv-project · Dec 16, 2019 · 89a5cc1 · 89a5cc1
1 parent bb8f6b4
commit 89a5cc1
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/client/src/main/java/com/distkv/dst/asyncclient/DstAsyncSortedListProxy.java b/client/src/main/java/com/distkv/dst/asyncclient/DstAsyncSortedListProxy.java
@@ -89,4 +89,15 @@ public CompletableFuture<SortedListProtocol.PutMemberResponse> putMember(
             service.putMember(requestBuilder.build());
     return future;
   }
+
+  public CompletableFuture<SortedListProtocol.GetMemberResponse> getMember(
+          String key, String member) {
+    SortedListProtocol.GetMemberRequest.Builder requestBuilder =
+            SortedListProtocol.GetMemberRequest.newBuilder();
+    requestBuilder.setKey(key);
+    requestBuilder.setMember(member);
+    CompletableFuture<SortedListProtocol.GetMemberResponse> future =
+            service.getMember(requestBuilder.build());
+    return future;
+  }
 }
diff --git a/test/src/test/java/com/distkv/dst/test/client/AsyncSortedListTest.java b/test/src/test/java/com/distkv/dst/test/client/AsyncSortedListTest.java
@@ -59,6 +59,15 @@ public void testAsyncSortedList() throws ExecutionException, InterruptedExceptio
       Assert.assertEquals(r.getList(1).getMember(), "wlll");
     });
 
+    // TestGetMember
+    CompletableFuture<SortedListProtocol.GetMemberResponse> getMemberFuture =
+            client.sortedLists().getMember("k1", "55");
+    getMemberFuture.whenComplete((r, t) -> {
+      Assert.assertEquals(r.getEntity().getMember(), "55");
+      Assert.assertEquals(r.getEntity().getScore(), 6);
+      Assert.assertEquals(r.getCount(), 4);
+    });
+
     //TestDrop
     CompletableFuture<CommonProtocol.DropResponse> dropFuture =
             client.sortedLists().drop("k1");
@@ -71,6 +80,7 @@ public void testAsyncSortedList() throws ExecutionException, InterruptedExceptio
     putItemFuture.get();
     removeFuture.get();
     topFuture.get();
+    getMemberFuture.get();
     dropFuture.get();
     client.disconnect();
   }