com.landawn.abacus.util.stream.ParallelIteratorLongStream Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of abacus-android Show documentation
Show all versions of abacus-android Show documentation
A general and simple library for Android
/*
* Copyright (C) 2016, 2017, 2018, 2019 HaiYang Li
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
* in compliance with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software distributed under the License
* is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
* or implied. See the License for the specific language governing permissions and limitations under
* the License.
*/
package com.landawn.abacus.util.stream;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Deque;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import com.landawn.abacus.util.AsyncExecutor;
import com.landawn.abacus.util.ContinuableFuture;
import com.landawn.abacus.util.LongIterator;
import com.landawn.abacus.util.MutableBoolean;
import com.landawn.abacus.util.MutableLong;
import com.landawn.abacus.util.N;
import com.landawn.abacus.util.Nth;
import com.landawn.abacus.util.Pair;
import com.landawn.abacus.util.Try;
import com.landawn.abacus.util.u.Holder;
import com.landawn.abacus.util.u.OptionalLong;
import com.landawn.abacus.util.function.BiConsumer;
import com.landawn.abacus.util.function.BinaryOperator;
import com.landawn.abacus.util.function.Consumer;
import com.landawn.abacus.util.function.Function;
import com.landawn.abacus.util.function.LongBiFunction;
import com.landawn.abacus.util.function.LongBinaryOperator;
import com.landawn.abacus.util.function.LongConsumer;
import com.landawn.abacus.util.function.LongFunction;
import com.landawn.abacus.util.function.LongPredicate;
import com.landawn.abacus.util.function.LongTernaryOperator;
import com.landawn.abacus.util.function.LongToDoubleFunction;
import com.landawn.abacus.util.function.LongToFloatFunction;
import com.landawn.abacus.util.function.LongToIntFunction;
import com.landawn.abacus.util.function.LongUnaryOperator;
import com.landawn.abacus.util.function.ObjLongConsumer;
import com.landawn.abacus.util.function.Predicate;
import com.landawn.abacus.util.function.Supplier;
import com.landawn.abacus.util.function.ToDoubleFunction;
import com.landawn.abacus.util.function.ToFloatFunction;
import com.landawn.abacus.util.function.ToIntFunction;
import com.landawn.abacus.util.function.ToLongFunction;
/**
*
*/
final class ParallelIteratorLongStream extends IteratorLongStream {
private final int maxThreadNum;
private final Splitor splitor;
private final AsyncExecutor asyncExecutor;
private volatile IteratorLongStream sequential;
private volatile Stream boxed;
ParallelIteratorLongStream(final LongIterator values, final boolean sorted, final int maxThreadNum, final Splitor splitor, final AsyncExecutor asyncExector,
final Collection closeHandlers) {
super(values, sorted, closeHandlers);
this.maxThreadNum = checkMaxThreadNum(maxThreadNum);
this.splitor = splitor == null ? DEFAULT_SPLITOR : splitor;
this.asyncExecutor = asyncExector == null ? DEFAULT_ASYNC_EXECUTOR : asyncExector;
}
ParallelIteratorLongStream(final LongStream stream, final boolean sorted, final int maxThreadNum, final Splitor splitor, final AsyncExecutor asyncExector,
final Deque closeHandlers) {
this(stream.iteratorEx(), sorted, maxThreadNum, splitor, asyncExector, mergeCloseHandlers(stream, closeHandlers));
}
ParallelIteratorLongStream(final Stream stream, final boolean sorted, final int maxThreadNum, final Splitor splitor, final AsyncExecutor asyncExector,
final Deque closeHandlers) {
this(longIterator(stream.iteratorEx()), sorted, maxThreadNum, splitor, asyncExector, mergeCloseHandlers(stream, closeHandlers));
}
@Override
public LongStream filter(final LongPredicate predicate) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.filter(predicate);
}
final Stream stream = boxed().filter(new Predicate() {
@Override
public boolean test(Long value) {
return predicate.test(value);
}
});
return new ParallelIteratorLongStream(stream, false, maxThreadNum, splitor, asyncExecutor, closeHandlers);
}
@Override
public LongStream takeWhile(final LongPredicate predicate) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.takeWhile(predicate);
}
final Stream stream = boxed().takeWhile(new Predicate() {
@Override
public boolean test(Long value) {
return predicate.test(value);
}
});
return new ParallelIteratorLongStream(stream, false, maxThreadNum, splitor, asyncExecutor, closeHandlers);
}
@Override
public LongStream dropWhile(final LongPredicate predicate) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.dropWhile(predicate);
}
final Stream stream = boxed().dropWhile(new Predicate() {
@Override
public boolean test(Long value) {
return predicate.test(value);
}
});
return new ParallelIteratorLongStream(stream, false, maxThreadNum, splitor, asyncExecutor, closeHandlers);
}
@Override
public LongStream map(final LongUnaryOperator mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.map(mapper);
}
final LongStream stream = boxed().mapToLong(new ToLongFunction() {
@Override
public long applyAsLong(Long value) {
return mapper.applyAsLong(value);
}
});
return new ParallelIteratorLongStream(stream, false, maxThreadNum, splitor, asyncExecutor, closeHandlers);
}
@Override
public IntStream mapToInt(final LongToIntFunction mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.mapToInt(mapper);
}
final IntStream stream = boxed().mapToInt(new ToIntFunction() {
@Override
public int applyAsInt(Long value) {
return mapper.applyAsInt(value);
}
});
return new ParallelIteratorIntStream(stream, false, maxThreadNum, splitor, asyncExecutor, closeHandlers);
}
@Override
public FloatStream mapToFloat(final LongToFloatFunction mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.mapToFloat(mapper);
}
final FloatStream stream = boxed().mapToFloat(new ToFloatFunction() {
@Override
public float applyAsFloat(Long value) {
return mapper.applyAsFloat(value);
}
});
return new ParallelIteratorFloatStream(stream, false, maxThreadNum, splitor, asyncExecutor, closeHandlers);
}
@Override
public DoubleStream mapToDouble(final LongToDoubleFunction mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.mapToDouble(mapper);
}
final DoubleStream stream = boxed().mapToDouble(new ToDoubleFunction() {
@Override
public double applyAsDouble(Long value) {
return mapper.applyAsDouble(value);
}
});
return new ParallelIteratorDoubleStream(stream, false, maxThreadNum, splitor, asyncExecutor, closeHandlers);
}
@Override
public Stream mapToObj(final LongFunction extends U> mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.mapToObj(mapper);
}
return boxed().map(new Function() {
@Override
public U apply(Long value) {
return mapper.apply(value);
}
});
}
@Override
public LongStream flatMap(final LongFunction extends LongStream> mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return new ParallelIteratorLongStream(sequential().flatMap(mapper), false, maxThreadNum, splitor, asyncExecutor, null);
}
final LongStream stream = boxed().flatMapToLong(new Function() {
@Override
public LongStream apply(Long value) {
return mapper.apply(value.longValue());
}
});
return new ParallelIteratorLongStream(stream, false, maxThreadNum, splitor, asyncExecutor, null);
}
@Override
public IntStream flatMapToInt(final LongFunction extends IntStream> mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return new ParallelIteratorIntStream(sequential().flatMapToInt(mapper), false, maxThreadNum, splitor, asyncExecutor, null);
}
final IntStream stream = boxed().flatMapToInt(new Function() {
@Override
public IntStream apply(Long value) {
return mapper.apply(value.longValue());
}
});
return new ParallelIteratorIntStream(stream, false, maxThreadNum, splitor, asyncExecutor, null);
}
@Override
public FloatStream flatMapToFloat(final LongFunction extends FloatStream> mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return new ParallelIteratorFloatStream(sequential().flatMapToFloat(mapper), false, maxThreadNum, splitor, asyncExecutor, null);
}
final FloatStream stream = boxed().flatMapToFloat(new Function() {
@Override
public FloatStream apply(Long value) {
return mapper.apply(value.longValue());
}
});
return new ParallelIteratorFloatStream(stream, false, maxThreadNum, splitor, asyncExecutor, null);
}
@Override
public DoubleStream flatMapToDouble(final LongFunction extends DoubleStream> mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return new ParallelIteratorDoubleStream(sequential().flatMapToDouble(mapper), false, maxThreadNum, splitor, asyncExecutor, null);
}
final DoubleStream stream = boxed().flatMapToDouble(new Function() {
@Override
public DoubleStream apply(Long value) {
return mapper.apply(value.longValue());
}
});
return new ParallelIteratorDoubleStream(stream, false, maxThreadNum, splitor, asyncExecutor, null);
}
@Override
public Stream flatMapToObj(final LongFunction extends Stream> mapper) {
assertNotClosed();
if (maxThreadNum <= 1) {
return new ParallelIteratorStream<>(sequential().flatMapToObj(mapper), false, null, maxThreadNum, splitor, asyncExecutor, null);
}
return boxed().flatMap(new Function>() {
@Override
public Stream apply(Long value) {
return mapper.apply(value.longValue());
}
});
}
@Override
public LongStream peek(final LongConsumer action) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.peek(action);
}
final LongStream stream = boxed().peek(new Consumer() {
@Override
public void accept(Long t) {
action.accept(t);
}
}).sequential().mapToLong(ToLongFunction.UNBOX);
return new ParallelIteratorLongStream(stream, false, maxThreadNum, splitor, asyncExecutor, closeHandlers);
}
@Override
public void forEach(final Try.LongConsumer action) throws E {
assertNotClosed();
if (maxThreadNum <= 1) {
super.forEach(action);
return;
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
for (int i = 0; i < maxThreadNum; i++) {
futureList.add(asyncExecutor.execute(new Try.Runnable() {
@Override
public void run() {
long next = 0;
try {
while (eHolder.value() == null) {
synchronized (elements) {
if (elements.hasNext()) {
next = elements.nextLong();
} else {
break;
}
}
action.accept(next);
}
} catch (Exception e) {
setError(eHolder, e);
}
}
}));
}
try {
complette(futureList, eHolder, (E) null);
} finally {
close();
}
}
@Override
public > M toMap(final LongFunction extends K> keyMapper, final LongFunction extends V> valueMapper,
final BinaryOperator mergeFunction, final Supplier extends M> mapFactory) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.toMap(keyMapper, valueMapper, mergeFunction, mapFactory);
}
final Function super Long, ? extends K> keyMapper2 = new Function() {
@Override
public K apply(Long value) {
return keyMapper.apply(value);
}
};
final Function super Long, ? extends V> valueMapper2 = new Function() {
@Override
public V apply(Long value) {
return valueMapper.apply(value);
}
};
return boxed().toMap(keyMapper2, valueMapper2, mergeFunction, mapFactory);
}
@Override
public > M toMap(final LongFunction extends K> keyMapper, final Collector downstream,
final Supplier extends M> mapFactory) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.toMap(keyMapper, downstream, mapFactory);
}
final Function super Long, ? extends K> keyMapper2 = new Function() {
@Override
public K apply(Long value) {
return keyMapper.apply(value);
}
};
return boxed().toMap(keyMapper2, downstream, mapFactory);
}
@Override
public long reduce(final long identity, final LongBinaryOperator op) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.reduce(identity, op);
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
for (int i = 0; i < maxThreadNum; i++) {
futureList.add(asyncExecutor.execute(new Callable() {
@Override
public Long call() {
long result = identity;
long next = 0;
try {
while (eHolder.value() == null) {
synchronized (elements) {
if (elements.hasNext()) {
next = elements.nextLong();
} else {
break;
}
}
result = op.applyAsLong(result, next);
}
} catch (Exception e) {
setError(eHolder, e);
}
return result;
}
}));
}
if (eHolder.value() != null) {
close();
throw N.toRuntimeException(eHolder.value());
}
Long result = null;
try {
for (ContinuableFuture future : futureList) {
if (result == null) {
result = future.get();
} else {
result = op.applyAsLong(result, future.get());
}
}
} catch (InterruptedException | ExecutionException e) {
throw N.toRuntimeException(e);
} finally {
close();
}
return result == null ? identity : result;
}
@Override
public OptionalLong reduce(final LongBinaryOperator accumulator) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.reduce(accumulator);
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
for (int i = 0; i < maxThreadNum; i++) {
futureList.add(asyncExecutor.execute(new Callable() {
@Override
public Long call() {
long result = 0;
synchronized (elements) {
if (elements.hasNext()) {
result = elements.nextLong();
} else {
return null;
}
}
long next = 0;
try {
while (eHolder.value() == null) {
synchronized (elements) {
if (elements.hasNext()) {
next = elements.nextLong();
} else {
break;
}
}
result = accumulator.applyAsLong(result, next);
}
} catch (Exception e) {
setError(eHolder, e);
}
return result;
}
}));
}
if (eHolder.value() != null) {
close();
throw N.toRuntimeException(eHolder.value());
}
Long result = null;
try {
for (ContinuableFuture future : futureList) {
final Long tmp = future.get();
if (tmp == null) {
continue;
} else if (result == null) {
result = tmp;
} else {
result = accumulator.applyAsLong(result, tmp);
}
}
} catch (InterruptedException | ExecutionException e) {
throw N.toRuntimeException(e);
} finally {
close();
}
return result == null ? OptionalLong.empty() : OptionalLong.of(result);
}
@Override
public R collect(final Supplier supplier, final ObjLongConsumer super R> accumulator, final BiConsumer combiner) {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.collect(supplier, accumulator, combiner);
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
for (int i = 0; i < maxThreadNum; i++) {
futureList.add(asyncExecutor.execute(new Callable() {
@Override
public R call() {
final R container = supplier.get();
long next = 0;
try {
while (eHolder.value() == null) {
synchronized (elements) {
if (elements.hasNext()) {
next = elements.nextLong();
} else {
break;
}
}
accumulator.accept(container, next);
}
} catch (Exception e) {
setError(eHolder, e);
}
return container;
}
}));
}
if (eHolder.value() != null) {
close();
throw N.toRuntimeException(eHolder.value());
}
R container = (R) NONE;
try {
for (ContinuableFuture future : futureList) {
if (container == NONE) {
container = future.get();
} else {
combiner.accept(container, future.get());
}
}
} catch (InterruptedException | ExecutionException e) {
throw N.toRuntimeException(e);
} finally {
close();
}
return container == NONE ? supplier.get() : container;
}
@Override
public boolean anyMatch(final Try.LongPredicate predicate) throws E {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.anyMatch(predicate);
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
final MutableBoolean result = MutableBoolean.of(false);
for (int i = 0; i < maxThreadNum; i++) {
futureList.add(asyncExecutor.execute(new Try.Runnable() {
@Override
public void run() {
long next = 0;
try {
while (result.isFalse() && eHolder.value() == null) {
synchronized (elements) {
if (elements.hasNext()) {
next = elements.nextLong();
} else {
break;
}
}
if (predicate.test(next)) {
result.setTrue();
break;
}
}
} catch (Exception e) {
setError(eHolder, e);
}
}
}));
}
try {
complette(futureList, eHolder, (E) null);
} finally {
close();
}
return result.value();
}
@Override
public boolean allMatch(final Try.LongPredicate predicate) throws E {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.allMatch(predicate);
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
final MutableBoolean result = MutableBoolean.of(true);
for (int i = 0; i < maxThreadNum; i++) {
futureList.add(asyncExecutor.execute(new Try.Runnable() {
@Override
public void run() {
long next = 0;
try {
while (result.isTrue() && eHolder.value() == null) {
synchronized (elements) {
if (elements.hasNext()) {
next = elements.nextLong();
} else {
break;
}
}
if (predicate.test(next) == false) {
result.setFalse();
break;
}
}
} catch (Exception e) {
setError(eHolder, e);
}
}
}));
}
try {
complette(futureList, eHolder, (E) null);
} finally {
close();
}
return result.value();
}
@Override
public boolean noneMatch(final Try.LongPredicate predicate) throws E {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.noneMatch(predicate);
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
final MutableBoolean result = MutableBoolean.of(true);
for (int i = 0; i < maxThreadNum; i++) {
futureList.add(asyncExecutor.execute(new Try.Runnable() {
@Override
public void run() {
long next = 0;
try {
while (result.isTrue() && eHolder.value() == null) {
synchronized (elements) {
if (elements.hasNext()) {
next = elements.nextLong();
} else {
break;
}
}
if (predicate.test(next)) {
result.setFalse();
break;
}
}
} catch (Exception e) {
setError(eHolder, e);
}
}
}));
}
try {
complette(futureList, eHolder, (E) null);
} finally {
close();
}
return result.value();
}
@Override
public OptionalLong findFirst(final Try.LongPredicate predicate) throws E {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.findFirst(predicate);
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
final Holder> resultHolder = new Holder<>();
final MutableLong index = MutableLong.of(0);
for (int i = 0; i < maxThreadNum; i++) {
futureList.add(asyncExecutor.execute(new Try.Runnable() {
@Override
public void run() {
final Pair pair = new Pair<>();
try {
while (resultHolder.value() == null && eHolder.value() == null) {
synchronized (elements) {
if (elements.hasNext()) {
pair.left = index.getAndIncrement();
pair.right = elements.nextLong();
} else {
break;
}
}
if (predicate.test(pair.right)) {
synchronized (resultHolder) {
if (resultHolder.value() == null || pair.left < resultHolder.value().left) {
resultHolder.setValue(pair.copy());
}
}
break;
}
}
} catch (Exception e) {
setError(eHolder, e);
}
}
}));
}
try {
complette(futureList, eHolder, (E) null);
} finally {
close();
}
return resultHolder.value() == null ? OptionalLong.empty() : OptionalLong.of(resultHolder.value().right);
}
@Override
public OptionalLong findLast(final Try.LongPredicate predicate) throws E {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.findLast(predicate);
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
final Holder> resultHolder = new Holder<>();
final MutableLong index = MutableLong.of(0);
for (int i = 0; i < maxThreadNum; i++) {
futureList.add(asyncExecutor.execute(new Try.Runnable() {
@Override
public void run() {
final Pair pair = new Pair<>();
try {
while (eHolder.value() == null) {
synchronized (elements) {
if (elements.hasNext()) {
pair.left = index.getAndIncrement();
pair.right = elements.nextLong();
} else {
break;
}
}
if (predicate.test(pair.right)) {
synchronized (resultHolder) {
if (resultHolder.value() == null || pair.left > resultHolder.value().left) {
resultHolder.setValue(pair.copy());
}
}
}
}
} catch (Exception e) {
setError(eHolder, e);
}
}
}));
}
try {
complette(futureList, eHolder, (E) null);
} finally {
close();
}
return resultHolder.value() == null ? OptionalLong.empty() : OptionalLong.of(resultHolder.value().right);
}
@Override
public OptionalLong findAny(final Try.LongPredicate predicate) throws E {
assertNotClosed();
if (maxThreadNum <= 1) {
return super.findAny(predicate);
}
final List> futureList = new ArrayList<>(maxThreadNum);
final Holder eHolder = new Holder<>();
final Holder