gistfile1.java

public class TimeLimitStep<S> extends MapStep<S, S> {

    private final AtomicLong startTime = new AtomicLong(-1);

    public TimeLimitStep(final Traversal traversal, final long timeLimit) {
        super(traversal);
        super.setFunction(traverser -> {
            if (this.startTime.get() == -1l)
                this.startTime.set(System.currentTimeMillis());
            if ((System.currentTimeMillis() - this.startTime.get()) >= timeLimit)
                throw FastNoSuchElementException.instance();
            return traverser.get();
        });
    }
}