fix(analytical): Fix the garbage collection of GAE Java (#4001)

As titled.
alibaba · Jan 14, 2025 · bb9a603 · bb9a603
1 parent 19d728e
commit bb9a603
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/...giraph/src/main/java/com/alibaba/graphscope/parallel/mm/impl/GiraphMpiMessageManager.java b/...giraph/src/main/java/com/alibaba/graphscope/parallel/mm/impl/GiraphMpiMessageManager.java
@@ -113,6 +113,7 @@ public void receiveMessages() {
             ///////////////////////////////////////////
             bytesOfReceivedMsg += tmpVector.size();
         }
+        tmpVector.delete();
         logger.info(
                 "Frag [{}] totally Received [{}] bytes from others starting deserialization",
                 fragId,
@@ -256,5 +257,9 @@ public void postSuperstep() {
     }
 
     @Override
-    public void postApplication() {}
+    public void postApplication() {
+        for (int i = 0; i < fragNum; ++i) {
+            cacheOut[i].close();
+        }
+    }
 }
diff --git a/...ape-jdk/src/main/java/com/alibaba/graphscope/serialization/FFIByteVectorOutputStream.java b/...ape-jdk/src/main/java/com/alibaba/graphscope/serialization/FFIByteVectorOutputStream.java
@@ -38,6 +38,11 @@ public FFIByteVectorOutputStream(FFIByteVector vector) {
         offset = 0;
     }
 
+    @Override
+    public void close() {
+        vector.delete();
+    }
+
     public void resize(long size) {
         vector.resize(size);
     }