Open NoOpBarrierStep for extensibility

Signed-off-by: Oleksandr Porunov <[email protected]>
apache · May 18, 2024 · 5f2aff6 · 5f2aff6
1 parent 1fa67c2
commit 5f2aff6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/CHANGELOG.asciidoc b/CHANGELOG.asciidoc
@@ -25,7 +25,7 @@ image::https://raw.githubusercontent.com/apache/tinkerpop/master/docs/static/ima
 
 This release also includes changes from <<release-3-6-8, 3.6.8>>.
 
-
+Open `NoOpBarrierStep` for extensibility (removed `final` keyword)
 
 [[release-3-7-2]]
 === TinkerPop 3.7.2 (April 8, 2024)

diff --git a/...rc/main/java/org/apache/tinkerpop/gremlin/process/traversal/step/map/NoOpBarrierStep.java b/...rc/main/java/org/apache/tinkerpop/gremlin/process/traversal/step/map/NoOpBarrierStep.java
@@ -35,19 +35,23 @@
 /**
  * @author Marko A. Rodriguez (http://markorodriguez.com)
  */
-public final class NoOpBarrierStep<S> extends AbstractStep<S, S> implements LocalBarrier<S> {
+public class NoOpBarrierStep<S> extends AbstractStep<S, S> implements LocalBarrier<S> {
 
-    private int maxBarrierSize;
-    private TraverserSet<S> barrier;
+    protected int maxBarrierSize;
+    protected TraverserSet<S> barrier;
 
     public NoOpBarrierStep(final Traversal.Admin traversal) {
         this(traversal, Integer.MAX_VALUE);
     }
 
     public NoOpBarrierStep(final Traversal.Admin traversal, final int maxBarrierSize) {
+        this(traversal, maxBarrierSize, (TraverserSet<S>) traversal.getTraverserSetSupplier().get());
+    }
+
+    public NoOpBarrierStep(final Traversal.Admin traversal, final int maxBarrierSize, TraverserSet<S> barrier) {
         super(traversal);
         this.maxBarrierSize = maxBarrierSize;
-        this.barrier = (TraverserSet<S>) this.traversal.getTraverserSetSupplier().get();
+        this.barrier = barrier;
     }
 
     @Override