add plasma java client metrics api (apache#7)

zhouyuan · Mar 22, 2021 · 608ee97 · 608ee97
1 parent 104e918
commit 608ee97
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/java/plasma/src/main/java/org/apache/arrow/plasma/ObjectStoreLink.java b/java/plasma/src/main/java/org/apache/arrow/plasma/ObjectStoreLink.java
@@ -128,4 +128,9 @@ default byte[] get(byte[] objectId, int timeoutMs, boolean isMetadata) {
    * List all objects in the PlasmaStore.
    */
   List<byte[]> list();
+
+  /**
+   * Get metrics in the PlasmaStore.
+   */
+  int metrics(long[] metrics);
 }
diff --git a/java/plasma/src/main/java/org/apache/arrow/plasma/PlasmaClient.java b/java/plasma/src/main/java/org/apache/arrow/plasma/PlasmaClient.java
@@ -181,4 +181,14 @@ public void delete(byte[] objectId) {
   public boolean contains(byte[] objectId) {
     return PlasmaClientJNI.contains(conn, objectId);
   }
+
+  /**
+   * Get metrics in the PlasmaStore.
+   *
+   * @param metrics used to return metrics, array size is 4, elements are: total memory size,
+   *        used memory size, total external size, used external size.
+   */
+  public int metrics(long[] metrics) {
+    return PlasmaClientJNI.metrics(conn, metrics);
+  }
 }