(incubator-hugegraph-computer) branch master updated: feat(core): support output filter (#303)

[jin]

This is an automated email from the ASF dual-hosted git repository.

jin pushed a commit to branch master
in repository 
https://gitbox.apache.org/repos/asf/incubator-hugegraph-computer.git


The following commit(s) were added to refs/heads/master by this push:
     new e241d169 feat(core): support output filter (#303)
e241d169 is described below

commit e241d1690f76db70ccb151ab575a7fdadedc8c22
Author: diaohancai <36229835+diaohan...@users.noreply.github.com>
AuthorDate: Sun Aug 11 00:45:08 2024 +0800

    feat(core): support output filter (#303)
    
    ## Main Changes
    
    1. computation results output supports custom write.
    2. `SingleSourceShortestPathOutput` implements 
`org.apache.hugegraph.computer.core.output.ComputerOutput#filter`
    
    ---------
    
    Co-authored-by: imbajin <j...@apache.org>
---
 .../algorithm/path/shortest/SingleSourceShortestPath.java        | 4 ++++
 .../algorithm/path/shortest/SingleSourceShortestPathOutput.java  | 8 ++++++++
 .../apache/hugegraph/computer/core/output/ComputerOutput.java    | 9 +++++++++
 .../hugegraph/computer/core/compute/FileGraphPartition.java      | 4 +++-
 4 files changed, 24 insertions(+), 1 deletion(-)

diff --git 
a/computer-algorithm/src/main/java/org/apache/hugegraph/computer/algorithm/path/shortest/SingleSourceShortestPath.java
 
b/computer-algorithm/src/main/java/org/apache/hugegraph/computer/algorithm/path/shortest/SingleSourceShortestPath.java
index c9b49cfd..b9b1f12b 100644
--- 
a/computer-algorithm/src/main/java/org/apache/hugegraph/computer/algorithm/path/shortest/SingleSourceShortestPath.java
+++ 
b/computer-algorithm/src/main/java/org/apache/hugegraph/computer/algorithm/path/shortest/SingleSourceShortestPath.java
@@ -278,4 +278,8 @@ public class SingleSourceShortestPath implements 
Computation<SingleSourceShortes
         }
         return false;
     }
+
+    public IdSet getTargetIdSet() {
+        return this.targetIdSet;
+    }
 }
diff --git 
a/computer-algorithm/src/main/java/org/apache/hugegraph/computer/algorithm/path/shortest/SingleSourceShortestPathOutput.java
 
b/computer-algorithm/src/main/java/org/apache/hugegraph/computer/algorithm/path/shortest/SingleSourceShortestPathOutput.java
index 4a646cf4..943babb2 100644
--- 
a/computer-algorithm/src/main/java/org/apache/hugegraph/computer/algorithm/path/shortest/SingleSourceShortestPathOutput.java
+++ 
b/computer-algorithm/src/main/java/org/apache/hugegraph/computer/algorithm/path/shortest/SingleSourceShortestPathOutput.java
@@ -20,8 +20,10 @@ package 
org.apache.hugegraph.computer.algorithm.path.shortest;
 import java.util.HashMap;
 import java.util.Map;
 
+import org.apache.hugegraph.computer.core.config.Config;
 import org.apache.hugegraph.computer.core.graph.vertex.Vertex;
 import org.apache.hugegraph.computer.core.output.hg.HugeGraphOutput;
+import org.apache.hugegraph.computer.core.worker.Computation;
 import org.apache.hugegraph.util.JsonUtil;
 
 public class SingleSourceShortestPathOutput extends HugeGraphOutput<String> {
@@ -45,4 +47,10 @@ public class SingleSourceShortestPathOutput extends 
HugeGraphOutput<String> {
         map.put("total_weight", value.totalWeight());
         return JsonUtil.toJson(map);
     }
+
+    @Override
+    public boolean filter(Config config, Computation computation, Vertex 
vertex) {
+        SingleSourceShortestPath sssp = (SingleSourceShortestPath) computation;
+        return sssp.getTargetIdSet() == null || 
sssp.getTargetIdSet().contains(vertex.id());
+    }
 }
diff --git 
a/computer-api/src/main/java/org/apache/hugegraph/computer/core/output/ComputerOutput.java
 
b/computer-api/src/main/java/org/apache/hugegraph/computer/core/output/ComputerOutput.java
index c2c2cc20..3522853f 100644
--- 
a/computer-api/src/main/java/org/apache/hugegraph/computer/core/output/ComputerOutput.java
+++ 
b/computer-api/src/main/java/org/apache/hugegraph/computer/core/output/ComputerOutput.java
@@ -19,6 +19,7 @@ package org.apache.hugegraph.computer.core.output;
 
 import org.apache.hugegraph.computer.core.config.Config;
 import org.apache.hugegraph.computer.core.graph.vertex.Vertex;
+import org.apache.hugegraph.computer.core.worker.Computation;
 
 /**
  * Computer output is used to output computer results. There is an output 
object
@@ -37,6 +38,14 @@ public interface ComputerOutput {
      */
     void write(Vertex vertex);
 
+    /**
+     * Write filter.
+     * True to commit the computation result, otherwise not to commit.
+     */
+    default boolean filter(Config config, Computation computation, Vertex 
vertex) {
+        return true;
+    }
+
     /**
      * Merge output files of multiple partitions if applicable.
      */
diff --git 
a/computer-core/src/main/java/org/apache/hugegraph/computer/core/compute/FileGraphPartition.java
 
b/computer-core/src/main/java/org/apache/hugegraph/computer/core/compute/FileGraphPartition.java
index 25312868..a2e61a8a 100644
--- 
a/computer-core/src/main/java/org/apache/hugegraph/computer/core/compute/FileGraphPartition.java
+++ 
b/computer-core/src/main/java/org/apache/hugegraph/computer/core/compute/FileGraphPartition.java
@@ -255,7 +255,9 @@ public class FileGraphPartition {
             Edges edges = this.edgesInput.edges(this.vertexInput.idPointer());
             vertex.edges(edges);
 
-            output.write(vertex);
+            if (output.filter(this.context.config(), this.computation, 
vertex)) {
+                output.write(vertex);
+            }
         }
 
         try {