diff --git a/CHANGELOG.asciidoc b/CHANGELOG.asciidoc index 43f551f6b67..caf5e430c4d 100644 --- 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 <>. - +Open `NoOpBarrierStep` for extensibility (removed `final` keyword) [[release-3-7-2]] === TinkerPop 3.7.2 (April 8, 2024) diff --git a/gremlin-core/src/main/java/org/apache/tinkerpop/gremlin/process/traversal/step/map/NoOpBarrierStep.java b/gremlin-core/src/main/java/org/apache/tinkerpop/gremlin/process/traversal/step/map/NoOpBarrierStep.java index b3f7c35e8ad..3f617b3a05a 100644 --- a/gremlin-core/src/main/java/org/apache/tinkerpop/gremlin/process/traversal/step/map/NoOpBarrierStep.java +++ b/gremlin-core/src/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 extends AbstractStep implements LocalBarrier { +public class NoOpBarrierStep extends AbstractStep implements LocalBarrier { - private int maxBarrierSize; - private TraverserSet barrier; + protected int maxBarrierSize; + protected TraverserSet 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) traversal.getTraverserSetSupplier().get()); + } + + public NoOpBarrierStep(final Traversal.Admin traversal, final int maxBarrierSize, TraverserSet barrier) { super(traversal); this.maxBarrierSize = maxBarrierSize; - this.barrier = (TraverserSet) this.traversal.getTraverserSetSupplier().get(); + this.barrier = barrier; } @Override