functionalj.stream.StreamPlusWithLimit Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of functionalj-core Show documentation
Show all versions of functionalj-core Show documentation
The module for FunctionalJ Core.
// ============================================================================
// Copyright (c) 2017-2021 Nawapunth Manusitthipol (NawaMan - http://nawaman.net).
// ----------------------------------------------------------------------------
// MIT License
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.
// ============================================================================
package functionalj.stream;
import static functionalj.stream.StreamPlusHelper.sequential;
import java.util.Spliterators;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.function.BiPredicate;
import java.util.function.Consumer;
import java.util.function.Predicate;
import java.util.stream.StreamSupport;
import functionalj.function.aggregator.AggregationToBoolean;
import functionalj.stream.markers.Sequential;
import lombok.val;
public interface StreamPlusWithLimit {
public StreamPlus streamPlus();
/** Limit the size of the stream to the given size. */
public default StreamPlus limit(Long maxSize) {
val streamPlus = streamPlus();
return ((maxSize == null) || (maxSize.longValue() < 0))
? streamPlus
: streamPlus
.limit((long)maxSize);
}
/** Skip to the given offset position. */
public default StreamPlus skip(Long offset) {
val streamPlus = streamPlus();
return ((offset == null) || (offset.longValue() < 0))
? streamPlus
: streamPlus
.skip((long)offset);
}
/** Skip any value while the condition is true. */
@Sequential
public default StreamPlus skipWhile(Predicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, stream -> {
val isStillTrue = new AtomicBoolean(true);
return stream.filter(e -> {
if (!isStillTrue.get())
return true;
if (!condition.test(e))
isStillTrue.set(false);
return !isStillTrue.get();
});
});
}
/** Skip any value while the condition is true. */
@Sequential
public default StreamPlus skipWhile(AggregationToBoolean aggregationCondition) {
val condition = aggregationCondition.newAggregator();
return skipWhile(condition);
}
/** Skip any value while the condition is true. */
@Sequential
public default StreamPlus skipWhile(BiPredicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, orgStreamPlus -> {
val orgSpliterator = orgStreamPlus.spliterator();
val newSpliterator = new Spliterators.AbstractSpliterator(orgSpliterator.estimateSize(), 0) {
boolean isStillSkipping = true;
boolean isFirst = true;
DATA prevValue = null;
@Override
public boolean tryAdvance(Consumer consumer) {
Consumer action = elem -> {
if (isStillSkipping) {
if (!isFirst) {
if (condition.test(prevValue, elem)) {
isStillSkipping = false;
}
} else {
isFirst = false;
}
if (!isStillSkipping) {
consumer.accept(prevValue);
}
prevValue = elem;
} else {
consumer.accept(prevValue);
prevValue = elem;
}
};
boolean hadNext = orgSpliterator.tryAdvance(action);
if (!isStillSkipping && !hadNext) {
consumer.accept(prevValue);
}
return hadNext;
}
};
val newStream = StreamSupport.stream(newSpliterator, false);
return StreamPlus.from(newStream);
});
}
/** Skip any value until the condition is true. */
@Sequential
public default StreamPlus skipUntil(Predicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, stream -> {
val isStillTrue = new AtomicBoolean(true);
return stream.filter(e -> {
if (!isStillTrue.get())
return true;
if (condition.test(e))
isStillTrue.set(false);
return !isStillTrue.get();
});
});
}
/** Skip any value until the condition is true. */
@Sequential
public default StreamPlus skipUntil(AggregationToBoolean aggregationCondition) {
val condition = aggregationCondition.newAggregator();
return skipUntil(condition);
}
/** Skip any value until the condition is true. */
@Sequential
public default StreamPlus skipUntil(BiPredicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, orgStreamPlus -> {
val orgSpliterator = orgStreamPlus.spliterator();
val newSpliterator = new Spliterators.AbstractSpliterator(orgSpliterator.estimateSize(), 0) {
boolean isStillSkipping = true;
boolean isFirst = true;
DATA prevValue = null;
@Override
public boolean tryAdvance(Consumer consumer) {
Consumer action = elem -> {
if (isStillSkipping) {
if (!isFirst) {
if (!condition.test(prevValue, elem)) {
isStillSkipping = false;
}
} else {
isFirst = false;
}
if (!isStillSkipping) {
consumer.accept(prevValue);
}
prevValue = elem;
} else {
consumer.accept(prevValue);
prevValue = elem;
}
};
boolean hadNext = orgSpliterator.tryAdvance(action);
if (!isStillSkipping && !hadNext) {
consumer.accept(prevValue);
}
return hadNext;
}
};
val newStream = StreamSupport.stream(newSpliterator, false);
return StreamPlus.from(newStream);
});
}
/** Accept any value while the condition is true. */
@Sequential
public default StreamPlus takeWhile(Predicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, orgStreamPlus -> {
val orgSpliterator = orgStreamPlus.spliterator();
val newSpliterator = new Spliterators.AbstractSpliterator(orgSpliterator.estimateSize(), 0) {
boolean stillGoing = true;
@Override
public boolean tryAdvance(Consumer consumer) {
if (stillGoing) {
Consumer action = elem -> {
if (condition.test(elem)) {
consumer.accept(elem);
} else {
stillGoing = false;
}
};
boolean hadNext = orgSpliterator.tryAdvance(action);
return hadNext && stillGoing;
}
return false;
}
};
val newStream = StreamSupport.stream(newSpliterator, false);
return StreamPlus.from(newStream);
});
}
/** Accept any value while the condition is true. */
@Sequential
public default StreamPlus takeWhile(AggregationToBoolean aggregationCondition) {
val condition = aggregationCondition.newAggregator();
return takeWhile(condition);
}
/** Accept any value while the condition is true. */
@Sequential
public default StreamPlus takeWhile(BiPredicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, orgStreamPlus -> {
val orgSpliterator = orgStreamPlus.spliterator();
val newSpliterator = new Spliterators.AbstractSpliterator(orgSpliterator.estimateSize(), 0) {
boolean stillGoing = true;
boolean isFirst = true;
DATA prevValue = null;
@Override
public boolean tryAdvance(Consumer consumer) {
if (stillGoing) {
Consumer action = elem -> {
if (!isFirst) {
if (condition.test(prevValue, elem)) {
consumer.accept(elem);
} else {
stillGoing = false;
}
} else {
consumer.accept(elem);
isFirst = false;
}
prevValue = elem;
};
boolean hadNext = orgSpliterator.tryAdvance(action);
return hadNext && stillGoing;
}
return false;
}
};
val newStream = StreamSupport.stream(newSpliterator, false);
return StreamPlus.from(newStream);
});
}
/** Accept any value until the condition is true. */
@Sequential
public default StreamPlus takeUntil(Predicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, orgStreamPlus -> {
val orgSpliterator = orgStreamPlus.spliterator();
val newSpliterator = new Spliterators.AbstractSpliterator(orgSpliterator.estimateSize(), 0) {
boolean stillGoing = true;
@Override
public boolean tryAdvance(Consumer consumer) {
if (stillGoing) {
Consumer action = elem -> {
if (!condition.test(elem)) {
consumer.accept(elem);
} else {
stillGoing = false;
}
};
boolean hadNext = orgSpliterator.tryAdvance(action);
return hadNext && stillGoing;
}
return false;
}
};
val newStream = StreamSupport.stream(newSpliterator, false);
return StreamPlus.from(newStream);
});
}
/** Accept any value until the condition is true. */
@Sequential
public default StreamPlus takeUntil(AggregationToBoolean aggregationCondition) {
val condition = aggregationCondition.newAggregator();
return takeWhile(condition);
}
/** Accept any value until the condition is true. */
@Sequential
public default StreamPlus takeUntil(BiPredicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, orgStreamPlus -> {
val orgSpliterator = orgStreamPlus.spliterator();
val newSpliterator = new Spliterators.AbstractSpliterator(orgSpliterator.estimateSize(), 0) {
boolean stillGoing = true;
boolean isFirst = true;
DATA prevValue = null;
@Override
public boolean tryAdvance(Consumer consumer) {
if (stillGoing) {
Consumer action = elem -> {
if (!isFirst) {
if (!condition.test(prevValue, elem)) {
consumer.accept(elem);
} else {
stillGoing = false;
}
} else {
consumer.accept(elem);
isFirst = false;
}
prevValue = elem;
};
boolean hadNext = orgSpliterator.tryAdvance(action);
return hadNext && stillGoing;
}
return false;
}
};
val newStream = StreamSupport.stream(newSpliterator, false);
return StreamPlus.from(newStream);
});
}
/** Accept any value until the condition is false - include the item that the condition is false. */
@Sequential
public default StreamPlus dropAfter(Predicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, orgStreamPlus -> {
val orgSpliterator = orgStreamPlus.spliterator();
val newSpliterator = new Spliterators.AbstractSpliterator(orgSpliterator.estimateSize(), 0) {
boolean stillGoing = true;
@Override
public boolean tryAdvance(Consumer consumer) {
if (stillGoing) {
Consumer action = elem -> {
consumer.accept(elem);
if (condition.test(elem)) {
stillGoing = false;
}
};
boolean hadNext = orgSpliterator.tryAdvance(action);
return hadNext;
}
return false;
}
};
val newStream = StreamSupport.stream(newSpliterator, false);
return StreamPlus.from(newStream);
});
}
/** Accept any value until the condition is false - include the item that the condition is false. */
@Sequential
public default StreamPlus dropAfter(AggregationToBoolean aggregationCondition) {
val condition = aggregationCondition.newAggregator();
return dropAfter(condition);
}
/** Accept any value until the condition is false - include the item that the condition is false. */
@Sequential
public default StreamPlus dropAfter(BiPredicate condition) {
val streamPlus = streamPlus();
return sequential(streamPlus, orgStreamPlus -> {
val orgSpliterator = orgStreamPlus.spliterator();
val newSpliterator = new Spliterators.AbstractSpliterator(orgSpliterator.estimateSize(), 0) {
boolean stillGoing = true;
boolean isFirst = true;
DATA prevValue = null;
@Override
public boolean tryAdvance(Consumer consumer) {
if (stillGoing) {
Consumer action = elem -> {
if (!isFirst) {
consumer.accept(elem);
if (condition.test(prevValue, elem)) {
stillGoing = false;
}
} else {
consumer.accept(elem);
isFirst = false;
}
prevValue = elem;
};
boolean hadNext = orgSpliterator.tryAdvance(action);
return hadNext;
}
return false;
}
};
val newStream = StreamSupport.stream(newSpliterator, false);
return StreamPlus.from(newStream);
});
}
}