Merge pull request apache#2 from bjchambers/splittable-do-fn

DoFnInvokers work for Splittable DoFn.
cosmoskitten · Sep 2, 2016 · 788b7db · 788b7db
2 parents 5fd1a3c + 52b174b
commit 788b7db
Show file tree

Hide file tree

Showing 3 changed files with 277 additions and 129 deletions.
diff --git a/sdks/java/core/src/main/java/org/apache/beam/sdk/transforms/DoFn.java b/sdks/java/core/src/main/java/org/apache/beam/sdk/transforms/DoFn.java
@@ -21,6 +21,7 @@
 import static com.google.common.base.Preconditions.checkNotNull;
 import static com.google.common.base.Preconditions.checkState;
 
+import com.google.common.base.Objects;
 import java.io.Serializable;
 import java.lang.annotation.Documented;
 import java.lang.annotation.ElementType;
@@ -620,6 +621,19 @@ public ProcessContinuation withResumeDelay(Duration resumeDelay) {
     public ProcessContinuation withFutureOutputWatermark(Instant futureOutputWatermark) {
       return new ProcessContinuation(resumeDelay, futureOutputWatermark);
     }
+
+    @Override
+    public boolean equals(Object obj) {
+      if (this == obj) {
+        return true;
+      } else if (!(obj instanceof ProcessContinuation)) {
+        return false;
+      } else {
+        ProcessContinuation that = (ProcessContinuation) obj;
+        return Objects.equal(this.resumeDelay, that.resumeDelay)
+            && Objects.equal(this.futureOutputWatermark, that.futureOutputWatermark);
+      }
+    }
   }
 
   /**