ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-05-09 23:51:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

!770 对EasyStream中补充peekIdx()方法,修改mapMulti()方法适配于更高版本JDK

Browse Source 
Merge pull request !770 from ZVerify/v6-dev
This commit is contained in:
Looly 2022-08-26 01:23:19 +00:00 committed by Gitee
commit 7d3c567961
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 47 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
hutool-core/src/main/java/cn/hutool/core/stream/EasyStream.java
View File

@ -25,6 +25,7 @@ import java.util.Spliterator;
import java.util.Spliterators;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.BiConsumer;
import java.util.function.BiFunction;
import java.util.function.BiPredicate;
@ -526,7 +527,7 @@ public class EasyStream<T> implements Stream<T>, Iterable<T> {
* @param <R> 拆分后流的元素类型
* @return 返回叠加拆分操作后的流
*/
public <R> EasyStream<R> mapMulti(final BiConsumer<? super T, ? super FastStreamBuilder<R>> mapper) {
public <R> EasyStream<R> mapMulti(final BiConsumer<? super T, ? super Consumer<R>> mapper) {
Objects.requireNonNull(mapper);
return flatMap(e -> {
final FastStreamBuilder<R> buffer = EasyStream.builder();
@ -630,6 +631,31 @@ public class EasyStream<T> implements Stream<T>, Iterable<T> {
return new EasyStream<>(stream.peek(action));
}
/**
* 返回与指定函数将元素作为参数执行后组成的流。操作带下标，并行流时下标永远为-1
* 这是一个无状态中间操作
* @param action 指定的函数
* @return 返回叠加操作后的FastStream
* @apiNote 该方法存在的意义主要是用来调试
* 当你需要查看经过操作管道某处的元素和下标，可以执行以下操作:
* <pre>{@code
* .of("one", "two", "three", "four")
* .filter(e -> e.length() > 3)
* .peekIdx((e,i) -> System.out.println("Filtered value: " + e + " Filtered idx:" + i))
* .map(String::toUpperCase)
* .peekIdx((e,i) -> System.out.println("Mapped value: " + e + " Mapped idx:" + i))
* .collect(Collectors.toList());
* }</pre>
*/
public EasyStream<T> peekIdx(BiConsumer<? super T, Integer> action) {
Objects.requireNonNull(action);
if (isParallel()) {
return peek(e -> action.accept(e, NOT_FOUND_INDEX));
} else {
AtomicInteger index = new AtomicInteger(NOT_FOUND_INDEX);
return peek(e -> action.accept(e, index.incrementAndGet()));
}
}
/**
* 返回叠加调用{@link Console#log(Object)}打印出结果的流
*

20
hutool-core/src/test/java/cn/hutool/core/stream/EasyStreamTest.java
View File

@ -208,6 +208,26 @@ public class EasyStreamTest {
Assert.assertEquals(Arrays.asList(-1, -1, -1), EasyStream.of(1, 2, 3).parallel().flatMapIdx((e, i) -> EasyStream.of(i)).toList());
}
@Test
public void testPeek(){
EasyStream.of("one", "two", "three", "four")
.filter(e -> e.length() == 4)
.peek(e -> Assert.assertEquals("four", e))
.map(String::toUpperCase)
.peek(e -> Assert.assertEquals("FOUR", e))
.collect(Collectors.toList());
}
@Test
public void testPeekIdx(){
EasyStream.of("one", "two", "three", "four")
.filter(e -> e.length() == 4)
.peekIdx((e,i) -> Assert.assertEquals("four:0", e + ":" + i))
.map(String::toUpperCase)
.peekIdx((e,i) -> Assert.assertEquals("FOUR:0", e + ":" + i))
.collect(Collectors.toList());
}
@Test
public void testFlat() {
final List<Integer> list = Arrays.asList(1, 2, 3);