Expose profile in Result. Partially implements #17

rethinkdb · Feb 17, 2020 · 10cff9f · 10cff9f
1 parent 120a4bd
commit 10cff9f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/src/main/java/com/rethinkdb/model/Profile.java b/src/main/java/com/rethinkdb/model/Profile.java
@@ -6,20 +6,20 @@
 
 public class Profile {
 
-    private List<Object> profileObj;
+    private List<Object> values;
 
     private Profile(List<Object> profileObj) {
-        this.profileObj = profileObj;
+        this.values = profileObj;
     }
 
-    public static @Nullable Profile fromList(List<Object> profileObj) {
-        if(profileObj == null || profileObj.size() == 0){
+    public static @Nullable Profile fromList(List<Object> list) {
+        if(list == null || list.size() == 0){
             return null;
         }
-        return new Profile(profileObj);
+        return new Profile(list);
     }
 
-    public List<Object> getProfileObj() {
-        return profileObj;
+    public List<Object> getValues() {
+        return values;
     }
 }
diff --git a/src/main/java/com/rethinkdb/net/Result.java b/src/main/java/com/rethinkdb/net/Result.java
@@ -6,7 +6,9 @@
 import com.rethinkdb.gen.exc.ReqlDriverError;
 import com.rethinkdb.gen.exc.ReqlRuntimeError;
 import com.rethinkdb.gen.proto.ResponseType;
+import com.rethinkdb.model.Profile;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 import java.io.Closeable;
 import java.util.Iterator;
@@ -167,6 +169,10 @@ public Iterator<T> iterator() {
         return this;
     }
 
+    public @Nullable Profile profile() {
+        return currentResponse.get().profile;
+    }
+
     protected void handleFirstResponse() {
         if (firstRes.isWaitComplete()) {
             completed.complete(true);