cn.hutool.core.stream.StreamUtil Maven / Gradle / Ivy
package cn.hutool.core.stream;
import cn.hutool.core.io.IORuntimeException;
import cn.hutool.core.lang.Assert;
import cn.hutool.core.util.CharsetUtil;
import java.io.File;
import java.io.IOException;
import java.nio.charset.Charset;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Collection;
import java.util.Iterator;
import java.util.Spliterators;
import java.util.function.Function;
import java.util.function.UnaryOperator;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
/**
* {@link Stream} 工具类
*
* @author looly
* @since 5.6.7
*/
public class StreamUtil {
@SafeVarargs
public static Stream of(T... array) {
Assert.notNull(array, "Array must be not null!");
return Stream.of(array);
}
/**
* {@link Iterable}转换为{@link Stream}，默认非并行
*
* @param iterable 集合
* @param 集合元素类型
* @return {@link Stream}
*/
public static Stream of(Iterable iterable) {
return of(iterable, false);
}
/**
* {@link Iterable}转换为{@link Stream}
*
* @param iterable 集合
* @param parallel 是否并行
* @param 集合元素类型
* @return {@link Stream}
*/
public static Stream of(Iterable iterable, boolean parallel) {
Assert.notNull(iterable, "Iterable must be not null!");
return iterable instanceof Collection ?
parallel ? ((Collection) iterable).parallelStream() : ((Collection) iterable).stream() :
StreamSupport.stream(iterable.spliterator(), parallel);
}
/**
* {@link Iterator} 转换为 {@link Stream}
* @param iterator 迭代器
* @param 集合元素类型
* @return {@link Stream}
* @throws IllegalArgumentException 如果iterator为null，抛出该异常
*/
public static Stream of(Iterator iterator) {
return of(iterator, false);
}
/**
* {@link Iterator} 转换为 {@link Stream}
* @param iterator 迭代器
* @param parallel 是否并行
* @param 集合元素类型
* @return {@link Stream}
* @throws IllegalArgumentException 如果iterator为null，抛出该异常
*/
public static Stream of(Iterator iterator, boolean parallel) {
Assert.notNull(iterator, "iterator must not be null!");
return StreamSupport.stream(Spliterators.spliteratorUnknownSize(iterator, 0), parallel);
}
/**
* 按行读取文件为{@link Stream}
*
* @param file 文件
* @return {@link Stream}
*/
public static Stream of(File file) {
return of(file, CharsetUtil.CHARSET_UTF_8);
}
/**
* 按行读取文件为{@link Stream}
*
* @param path 路径
* @return {@link Stream}
*/
public static Stream of(Path path) {
return of(path, CharsetUtil.CHARSET_UTF_8);
}
/**
* 按行读取文件为{@link Stream}
*
* @param file 文件
* @param charset 编码
* @return {@link Stream}
*/
public static Stream of(File file, Charset charset) {
Assert.notNull(file, "File must be not null!");
return of(file.toPath(), charset);
}
/**
* 按行读取文件为{@link Stream}
*
* @param path 路径
* @param charset 编码
* @return {@link Stream}
*/
public static Stream of(Path path, Charset charset) {
try {
return Files.lines(path, charset);
} catch (IOException e) {
throw new IORuntimeException(e);
}
}
/**
* 通过函数创建Stream
*
* @param seed 初始值
* @param elementCreator 递进函数，每次调用此函数获取下一个值
* @param limit 限制个数
* @param 创建元素类型
* @return {@link Stream}
*/
public static Stream of(T seed, UnaryOperator elementCreator, int limit) {
return Stream.iterate(seed, elementCreator).limit(limit);
}
/**
* 将Stream中所有元素以指定分隔符，合并为一个字符串，对象默认调用toString方法
*
* @param stream {@link Stream}
* @param delimiter 分隔符
* @param 元素类型
* @return 字符串
*/
public static String join(Stream stream, CharSequence delimiter) {
return stream.collect(CollectorUtil.joining(delimiter));
}
/**
* 将Stream中所有元素以指定分隔符，合并为一个字符串
*
* @param stream {@link Stream}
* @param delimiter 分隔符
* @param toStringFunc 元素转换为字符串的函数
* @param 元素类型
* @return 字符串
*/
public static String join(Stream stream, CharSequence delimiter,
Function toStringFunc) {
return stream.collect(CollectorUtil.joining(delimiter, toStringFunc));
}
}