当前位置 : 主页 > 编程语言 > 其它开发 >

Lambda表达式工具类-日常搬砖

来源:互联网 收集:自由互联 发布时间:2022-05-30
众所周知,jdk8提供的Lambda表达式能帮我极大的精简代码,但是就在我们在实际使用过程中会有一些重复性的代码,比如对于stream来源的空判断处理,今天我就给大家整理了一些工具类,

众所周知,jdk8提供的Lambda表达式能帮我极大的精简代码,但是就在我们在实际使用过程中会有一些重复性的代码,比如对于stream来源的空判断处理,今天我就给大家整理了一些工具类,帮助大家更好的使用流
public final class EnhanceStream {

public static <T> List<T> singletonList(T o) {
    return Collections.singletonList(o);
}

/**
 * 生成list
 * 
 * @param elements
 *            元素
 * @param <T>
 *            泛型
 * @return List
 */
@SafeVarargs
public static <T> List<T> toList(T... elements) {
    List<T> list = new ArrayList<>();
    if (Objects.nonNull(elements)) {
        for (T element : elements) {
            if (Objects.nonNull(element)) {
                list.add(element);
            }
        }
    }
    return list;
}

public static <T> void addAll(Collection<T> collection, Collection<T> add) {
    if (isNotEmpty(add)) {
        collection.addAll(add);
    }
}

/**
 * 两个Map进行合并
 *
 * @param map
 *            原始map
 * @param add
 *            要合并的map
 * @param <K>
 *            key泛型
 * @param <V>
 *            值泛型
 */
public static <K, V> void addAll(Map<K, V> map, Map<K, V> add) {
    if (isNotEmpty(add) && Objects.nonNull(map)) {
        for (Map.Entry<K, V> kvEntry : add.entrySet()) {
            map.put(kvEntry.getKey(), kvEntry.getValue());
        }
    }
}

public static <T> List<T> setToList(Set<T> set) {
    if (EnhanceStream.isNotEmpty(set)) {
        return new ArrayList<>(set);
    } else {
        return EnhanceStream.emptyList();
    }
}

public static boolean isEmpty(final Collection<?> coll) {
    return coll == null || coll.isEmpty();
}

/**
 * Null-safe check if the specified collection is not empty.
 * <p>
 * Null returns false.
 *
 * @param coll
 *            the collection to check, may be null
 * @return true if non-null and non-empty
 * @since 3.2
 */
public static boolean isNotEmpty(final Collection<?> coll) {
    return !isEmpty(coll);
}

public static boolean isEmpty(final Map<?, ?> map) {
    return map == null || map.isEmpty();
}

public static boolean isNotEmpty(final Map<?, ?> map) {
    return !isEmpty(map);
}

/**
 * 返回空集合
 *
 * @param <T>
 *            泛型
 * @return List
 */
public static <T> List<T> emptyList() {
    return Collections.emptyList();
}

public static final <K, V> Map<K, V> emptyMap() {
    return Collections.emptyMap();
}

/**
 * 返回空集合
 *
 * @param <T>
 *            泛型
 * @return List
 */
public static <T> Set<T> emptySet() {
    return Collections.emptySet();
}

/**
 * 去重
 *
 * @param dataSources
 *            数据
 * @param <T>
 *            泛型
 * @return List<T>
 */
public static <T> List<T> distinct(List<T> dataSources) {
    return newStream(dataSources).filter(Objects::nonNull).distinct().collect(Collectors.toList());
}

/**
 * 去重
 *
 * @param dataSources
 *            数据
 * @param <T>
 *            泛型
 * @return List<T>
 */
public static <T, R> List<R> distinct(List<T> dataSources, Function<T, R> mapper) {
    List<R> rs = mapToList(dataSources, mapper);
    return newStream(rs).filter(Objects::nonNull).distinct().collect(Collectors.toList());
}

/**
 * 去重
 *
 * @param dataSources
 *            数据
 * @param <T>
 *            泛型
 * @return List<T>
 */
public static <T, T1, R> List<R> distinct(List<T> dataSources, Function<T, T1> mapper1, Function<T1, R> mapper2) {
    List<T1> rs = mapToList(dataSources, mapper1);
    return newStream(rs).filter(Objects::nonNull).map(mapper2).distinct().collect(Collectors.toList());
}

/**
 * 去重
 *
 * @param dataSources
 *            数据
 * @param <T>
 *            泛型
 * @return List<T>
 */
public static <T, R> List<T> distinctByKey(List<T> dataSources, Function<T, R> mapper) {
    Map<R, T> dismantling = dismantlingFirst(dataSources, mapper);
    return getValues(dismantling);
}

/**
 * 从流中获取Inter类型数据,非空处理后,求和
 *
 * @param dataStream
 *            原始数据流
 * @param mapper
 *            指定数字类型字段
 * @param <T>
 *            泛型
 * @return Integer
 */
public static <T> Integer mapToNonNullIntSum(Stream<T> dataStream, Function<T, Integer> mapper) {
    List<Integer> integers = mapToList(dataStream, mapper);
    int total = 0;
    for (Integer integer : integers) {
        if (Objects.nonNull(integer)) {
            total += integer;
        }
    }
    return total;
}

/**
 * 从流中获取Inter类型数据,非空处理后,求和(如果元素为空,则返回空)
 *
 * @param dataStream
 *            原始数据流
 * @param mapper
 *            指定数字类型字段
 * @param <T>
 *            泛型
 * @return Integer
 */
public static <T> Integer mapToNullAbleIntSum(Stream<T> dataStream, Function<T, Integer> mapper) {
    List<Integer> integers = mapToList(dataStream, mapper);
    int total = 0;
    boolean haveNonNull = false;
    for (Integer integer : integers) {
        if (Objects.nonNull(integer)) {
            haveNonNull = true;
            total += integer;
        }
    }
    return haveNonNull ? total : null;
}

/**
 * 从流中获取Inter类型数据,非空处理后,求和
 *
 * @param dataStream
 *            原始数据流
 * @param mapper
 *            指定数字类型字段
 * @param <T>
 *            泛型
 * @return Integer
 */
public static <T> Long mapToNonNullLongSum(Stream<T> dataStream, Function<T, Long> mapper) {
    List<Long> longs = mapToList(dataStream, mapper);
    long total = 0;
    for (Long nextLong : longs) {
        if (Objects.nonNull(nextLong)) {
            total += nextLong;
        }
    }
    return total;
}

/**
 * 从流中获取Inter类型数据,非空处理后,求和
 *
 * @param dataSources
 *            原始数据源
 * @param mapper
 *            指定数字类型字段
 * @param <T>
 *            泛型
 * @return Integer
 */
public static <T> Integer mapToNonNullIntSum(List<T> dataSources, Function<T, Integer> mapper) {
    return mapToNonNullIntSum(newStream(dataSources), mapper);
}

/**
 * 从流中获取Inter类型数据,非空处理后,求和(如果元素为空,则返回空)
 *
 * @param dataSources
 *            原始数据源
 * @param mapper
 *            指定数字类型字段
 * @param <T>
 *            泛型
 * @return Integer
 */
public static <T> Integer mapToNullAbleIntSum(List<T> dataSources, Function<T, Integer> mapper) {
    return mapToNullAbleIntSum(newStream(dataSources), mapper);
}

/**
 * 从流中获取Inter类型数据,非空处理后,求和
 *
 * @param dataSources
 *            原始数据源
 * @param mapper
 *            指定数字类型字段
 * @param <T>
 *            泛型
 * @return Integer
 */
public static <T> Long mapToNonNullLongSum(List<T> dataSources, Function<T, Long> mapper) {
    return mapToNonNullLongSum(newStream(dataSources), mapper);
}

/**
 * 从map中获取所有的值的list
 *
 * @param map
 *            Map数据
 * @param <K>
 *            key泛型
 * @param <V>
 *            value泛型
 * @return List
 */
public static <K, V> List<V> getValues(Map<K, V> map) {
    if (CollectionUtils.isEmpty(map)) {
        return Collections.emptyList();
    }
    Collection<V> values = map.values();
    return new ArrayList<>(values);
}

/**
 * 从map中获取所有的值的list
 *
 * @param map
 *            Map数据
 * @param <K>
 *            key泛型
 * @param <V>
 *            value泛型
 * @return List
 */
public static <K, V> Optional<V> getValuesFirst(Map<K, V> map) {
    List<V> values = getValues(map);
    return values.stream().findFirst();
}

/**
 * 如果是数据分组格式key是一个,value是list的情况,只去value中的第一个。
 *
 * @param map
 *            Map数据
 * @param <K>
 *            key泛型
 * @param <V>
 *            value泛型
 * @return List
 */
public static <K, V> List<V> eachGetValueFirst(Map<K, List<V>> map) {
    if (CollectionUtils.isEmpty(map)) {
        return Collections.emptyList();
    }
    List<V> result = new ArrayList<>(map.size());
    for (Map.Entry<K, List<V>> entry : map.entrySet()) {
        if (!CollectionUtils.isEmpty(entry.getValue())) {
            result.add(entry.getValue().get(0));
        }
    }
    return result;
}

/**
 * 从map中获取所有的key的list
 *
 * @param map
 *            Map数据
 * @param <K>
 *            key泛型
 * @param <V>
 *            value泛型
 * @return List
 */
public static <K, V> List<K> getKeys(Map<K, V> map) {
    if (CollectionUtils.isEmpty(map)) {
        return Collections.emptyList();
    }
    Set<K> ks = map.keySet();
    return new ArrayList<>(ks);
}

/**
 * 从流中建立新的类型
 *
 * @param dataStream
 *            数据流
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R> List<R> mapToList(Stream<T> dataStream, Function<T, R> mapping) {
    return newStream(dataStream).map(mapping).filter(Objects::nonNull).collect(Collectors.toList());
}

/**
 * 从流中建立新的类型
 *
 * @param dataSources
 *            数据源
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R> List<R> mapToList(List<T> dataSources, Function<T, R> mapping) {
    return newStream(dataSources).map(mapping).filter(Objects::nonNull).collect(Collectors.toList());
}

/**
 * 从流中建立新的类型
 *
 * @param dataSources
 *            数据源
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R> List<R> mapToUniqueList(List<T> dataSources, Function<T, R> mapping) {
    return newStream(dataSources).map(mapping).filter(Objects::nonNull).distinct().collect(Collectors.toList());
}

/**
 * 从流中建立新的类型
 *
 * @param dataSources
 *            数据源
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R1, R> List<R> mapToUniqueList(List<T> dataSources, Function<T, R1> mapping,
    Function<R1, R> mapping1) {
    return newStream(dataSources).map(mapping).filter(Objects::nonNull).map(mapping1).filter(Objects::nonNull)
        .distinct().collect(Collectors.toList());
}

/**
 * 从流中建立新的类型
 *
 * @param dataSources
 *            数据源
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R, V> List<V> supperMapToList(List<T> dataSources, Function<T, R> mapping,
    Function<R, V> mapping2) {
    return newStream(dataSources).map(mapping).filter(Objects::nonNull).map(mapping2).filter(Objects::nonNull)
        .distinct().collect(Collectors.toList());
}

/**
 * 将多个list集合合并成一个 如果要保证添加顺序,请按照顺序添加
 *
 * @param dataSourceList
 *            数据源
 * @return List
 */
@SafeVarargs
public static <T> List<T> mergeList(List<T>... dataSourceList) {
    if (Objects.isNull(dataSourceList)) {
        return emptyList();
    }
    List<T> result = new ArrayList<>();
    for (List<T> ts : dataSourceList) {
        if (isNotEmpty(ts)) {
            result.addAll(ts);
        }
    }
    return result;
}

/**
 * 从流中建立新的List类型,并将新的list
 *
 * @param dataSources
 *            数据源
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R> List<R> mapToMergeList(List<T> dataSources, Function<T, List<R>> mapping) {
    List<List<R>> collect =
        newStream(dataSources).map(mapping).filter(Objects::nonNull).collect(Collectors.toList());
    List<R> allList = new ArrayList<>();
    for (List<R> rs : collect) {
        allList.addAll(rs);
    }
    return allList;
}

/**
 * 从流中建立新的类型
 *
 * @param dataSource
 *            数据
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R> Set<R> mapToSet(List<T> dataSource, Function<T, R> mapping) {
    return newStream(dataSource).map(mapping).filter(Objects::nonNull).collect(Collectors.toSet());
}

/**
 * 从流中建立新的类型
 *
 * @param dataStream
 *            数据流
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R> Set<R> mapToSet(Stream<T> dataStream, Function<T, R> mapping) {
    return newStream(dataStream).map(mapping).filter(Objects::nonNull).collect(Collectors.toSet());
}

/**
 * 从流中建立新的类型并去重
 *
 * @param dataSources
 *            数据流
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R> List<R> distinctMapToList(List<T> dataSources, Function<T, R> mapping) {
    return newStream(newStream(dataSources)).map(mapping).filter(Objects::nonNull).distinct()
        .collect(Collectors.toList());
}

/**
 * 从流中建立新的类型并去重
 *
 * @param dataStream
 *            数据流
 * @param mapping
 *            转换函数
 * @param <T>
 *            原数据类型
 * @param <R>
 *            目标数据类型
 * @return List
 */
public static <T, R> List<R> distinctMapToList(Stream<T> dataStream, Function<T, R> mapping) {
    return newStream(dataStream).map(mapping).filter(Objects::nonNull).distinct().collect(Collectors.toList());
}

/**
 * 对数据流进行过滤
 *
 * @param dataStream
 *            数据流
 * @param predicate
 *            过滤条件
 * @param exceptionSupplier
 *            异常生成
 * @param <T>
 *            数据泛型
 * @param <X>
 *            异常泛型
 * @return List 过滤后的数据条件
 * @throws X
 *             Throwable
 */
public static <T, X extends Throwable> T filterSingle(Stream<T> dataStream, Predicate<? super T> predicate,
    Supplier<? extends X> exceptionSupplier) throws X {
    return StreamFilter.filterSingle(newStream(dataStream), predicate, exceptionSupplier);
}

/**
 * 过滤
 *
 * @param dataSource
 *            数据源
 * @param predicate
 *            过滤条件
 * @param <T>
 *            数据泛型
 * @return List
 */
public static <T, R> List<T> superFilter(List<T> dataSource, Function<T, R> function,
    Predicate<? super R> predicate) {
    return superFilter(newStream(dataSource), function, predicate);
}

/**
 * 根据条件生成List集合
 *
 * @param dataStream
 *            数据流
 * @param predicate
 *            过滤条件
 * @param <T>
 *            数据泛型
 * @return List
 */
public static <T, R> List<T> superFilter(Stream<T> dataStream, Function<T, R> function,
    Predicate<? super R> predicate) {
    List<T> result = new ArrayList<>();
    dataStream.filter(Objects::nonNull).forEach(t -> {
        R apply = function.apply(t);
        if (predicate.test(apply)) {
            result.add(t);
        }
    });
    return result;
}

/**
 * 对数据流进行过滤
 *
 * @param dataSource
 *            数据源头
 * @param predicate
 *            过滤条件
 * @param <T>
 *            数据泛型
 * @return List 过滤后的数据条件
 */
public static <T> List<T> filter(List<T> dataSource, Predicate<? super T> predicate) {
    return StreamFilter.filter(newStream(dataSource), predicate);
}

/**
 * 对数据流进行过滤
 *
 * @param dataStream
 *            数据流
 * @param predicate
 *            过滤条件
 * @param <T>
 *            数据泛型
 * @return List 过滤后的数据条件
 */
public static <T> List<T> filter(Stream<T> dataStream, Predicate<? super T> predicate) {
    return StreamFilter.filter(newStream(dataStream), predicate);
}

public static <T, X extends Throwable> T findFirst(Stream<T> dataStream, Supplier<? extends X> exceptionSupplier)
    throws X {
    return newStream(dataStream).findFirst().orElseThrow(exceptionSupplier);
}

public static <T> Optional<T> findFirst(Stream<T> dataStream) {
    return newStream(dataStream).findFirst();
}

public static <T> Optional<T> findFirst(List<T> dataList) {
    return newStream(dataList).findFirst();
}

public static <T> Optional<T> findAny(Stream<T> dataStream) {
    return newStream(dataStream).findAny();
}

/**
 * 从流中获取任意数据
 *
 * @param dataList
 *            数据集合
 * @param <T>
 *            泛型
 * @return Optional
 */
public static <T> Optional<T> findAny(List<T> dataList) {
    return newStream(dataList).findAny();
}

/**
 * 从流中获取任意数据
 *
 * @param dataList
 *            数据集合
 * @param <T>
 *            泛型
 * @return Optional
 */
public static <T> T findRequiredAny(List<T> dataList) {
    Optional<T> any = newStream(dataList).findAny();
    if (!any.isPresent()) {
        throw new RuntimeException("EnhanceStream.findRequiredAny:"+dataList);
    }
    return any.get();
}

/**
 * 对数据流进行过滤,并对过滤后的数据进行类型转换
 *
 * @param dataStream
 *            数据流
 * @param predicate
 *            过滤条件
 * @param applyMapping
 *            数据转换器
 * @param <T>
 *            数据泛型
 * @param <V>
 *            转换后的数据类型
 * @return List 过滤后的数据条件
 */
public static <T, V> List<V> filter(Stream<T> dataStream, Predicate<? super T> predicate,
    Function<? super T, ? extends V> applyMapping) {
    return StreamFilter.filter(newStream(dataStream), predicate, applyMapping);
}

/**
 * 对数据流进行分组
 *
 * @param dataStream
 *            数据流
 * @param keyApply
 *            分组项
 * @param <T>
 *            数据泛型
 * @param <K>
 *            分组项泛型
 * @return Map 分组后数据
 */
public static <T, K> Map<K, List<T>> group(Stream<T> dataStream, Function<? super T, ? extends K> keyApply) {
    return StreamBinder.group(newStream(dataStream), keyApply);
}

/**
 * 对数据流进行分组,并对分组后的数据进行类型转换
 *
 * @param dataStream
 *            数据流
 * @param keyApply
 *            分组项
 * @param valueApply
 *            数据转换器
 * @param <T>
 *            数据泛型
 * @param <K>
 *            分组项泛型
 * @param <V>
 *            转换后的数据类型
 * @return Map 分组后数据
 */
public static <T, K, V> Map<K, List<V>> group(Stream<T> dataStream, Function<? super T, ? extends K> keyApply,
    Function<? super T, ? extends V> valueApply) {
    return StreamBinder.group(newStream(dataStream), keyApply, valueApply);
}

/**
 * @param dataSource
 *            商品配置
 * @param keyApply
 *            key生成
 * @param <K>
 *            key泛型
 * @param <T>
 *            数据源泛型
 * @return Map<K, V>
 */
public static <K, T> Map<K, T> dismantlingFirst(List<T> dataSource, Function<? super T, ? extends K> keyApply) {
    return StreamBinder.dismantlingFirst(newStream(dataSource), keyApply);
}

/**
 * @param dataSource
 *            商品配置
 * @param keyApply
 *            key生成
 * @param <K>
 *            key泛型
 * @param <T>
 *            数据源泛型
 * @return Map<K, V>
 */
public static <K, V,T> Map<K, V> dismantlingFirst(List<T> dataSource, Function<? super T, ? extends K> keyApply,Function<? super T, ? extends V> valueApply) {
    return StreamBinder.dismantlingFirst(newStream(dataSource), keyApply,valueApply);
}

/**
 * @param dataSourceStream
 *            商品配置
 * @param keyApply
 *            key生成
 * @param <K>
 *            key泛型
 * @param <T>
 *            数据源泛型
 * @return Map<K, V>
 */
public static <K, T> Map<K, T> dismantlingFirst(Stream<T> dataSourceStream,
    Function<? super T, ? extends K> keyApply) {
    return StreamBinder.dismantlingFirst(dataSourceStream, keyApply);
}

/**
 * 数据拆分「注意: 拆分后的数据,如果Key一样会产生异常,请谨慎使用
 *
 * @param dataStream
 *            数据流
 * @param keyApply
 *            Key转换器
 * @param <T>
 *            原始数据泛型
 * @param <K>
 *            Key泛型
 * @return Map 拆分后数据
 */
public static <T, K> Map<K, T> dismantling(List<T> dataStream, Function<? super T, ? extends K> keyApply) {
    return StreamBinder.dismantling(newStream(dataStream), keyApply);
}

/**
 * 数据拆分「注意: 拆分后的数据,如果Key一样会产生异常,请谨慎使用
 *
 * @param dataStream
 *            数据流
 * @param keyApply
 *            Key转换器
 * @param <T>
 *            原始数据泛型
 * @param <K>
 *            Key泛型
 * @return Map 拆分后数据
 */
public static <T, K> Map<K, T> dismantling(Stream<T> dataStream, Function<? super T, ? extends K> keyApply) {
    return StreamBinder.dismantling(newStream(dataStream), keyApply);
}

/**
 * 数据拆分「注意: 拆分后的数据,如果Key一样会产生异常,请谨慎使用
 *
 * @param dataSource
 *            数据源
 * @param keyApply
 *            Key转换器
 * @param <T>
 *            原始数据泛型
 * @param <K>
 *            Key泛型
 * @return Map 拆分后数据
 */
public static <T, K, X extends Throwable> Map<K, T> dismantling(List<T> dataSource,
    Function<? super T, ? extends K> keyApply, Supplier<? extends X> exceptionSupplier) throws X {
    return StreamBinder.dismantling(newStream(dataSource), keyApply, exceptionSupplier);
}

/**
 * 数据拆分「注意: 拆分后的数据,如果Key一样会产生异常,请谨慎使用
 *
 * @param dataStream
 *            数据流
 * @param keyApply
 *            Key转换器
 * @param <T>
 *            原始数据泛型
 * @param <K>
 *            Key泛型
 * @return Map 拆分后数据
 */
public static <T, K, X extends Throwable> Map<K, T> dismantling(Stream<T> dataStream,
    Function<? super T, ? extends K> keyApply, Supplier<? extends X> exceptionSupplier) throws X {
    return StreamBinder.dismantling(newStream(dataStream), keyApply, exceptionSupplier);
}

/**
 * 数据拆分「注意: 拆分后的数据,如果Key一样会产生异常,请谨慎使用
 *
 * @param dataStream
 *            数据流
 * @param keyApply
 *            Key转换器
 * @param valueApply
 *            Value转换器
 * @param <T>
 *            原始数据泛型
 * @param <K>
 *            Key泛型
 * @param <V>
 *            Value泛型
 * @return Map 拆分后数据
 */
public static <T, K, V> Map<K, V> dismantling(Stream<T> dataStream, Function<? super T, ? extends K> keyApply,
    Function<? super T, ? extends V> valueApply) {
    return StreamBinder.dismantling(newStream(dataStream), keyApply, valueApply);
}

/**
 * 数据拆分「注意: 拆分后的数据,如果Key一样会产生异常,请谨慎使用
 *
 * @param dataStream
 *            数据流
 * @param keyApply
 *            Key转换器
 * @param valueApply
 *            Value转换器
 * @param <T>
 *            原始数据泛型
 * @param <K>
 *            Key泛型
 * @param <V>
 *            Value泛型
 * @param <X>
 *            数据重复产生的异常描述
 * @return Map 拆分后数据
 */
public static <T, K, V, X extends Throwable> Map<K, V> dismantling(Stream<T> dataStream,
    Function<? super T, ? extends K> keyApply, Function<? super T, ? extends V> valueApply,
    Supplier<? extends X> exceptionSupplier) throws X {
    return StreamBinder.dismantling(newStream(dataStream), keyApply, valueApply, exceptionSupplier);
}

private static <T> Stream<T> newStream(Stream<T> dataSourcesStream) {
    if (dataSourcesStream == null) {
        return Stream.empty();
    }
    return dataSourcesStream;
}

public static <T> Stream<T> newStream(List<T> dataSources) {
    if (dataSources == null) {
        return Stream.empty();
    }
    return dataSources.stream();
}

/**
 * 对数据流进行过滤,并对过滤后的数据进行类型转换
 *
 * @param dataSources
 *            原始数据流
 * @param predicate
 *            过滤条件
 * @param applyMapping
 *            数据转换器
 * @param <V>
 *            转换后的数据类型
 * @param <T>
 *            原始数据类型
 * @return List 过滤后的数据条件
 */
public static <T, V> List<V> filter(List<T> dataSources, Predicate<? super T> predicate,
    Function<? super T, ? extends V> applyMapping) {
    return StreamFilter.filter(newStream(dataSources), predicate, applyMapping);
}

/**
 * 数据拆分「注意: 拆分后的数据,如果Key一样会产生异常,请谨慎使用
 *
 * @param dataSources
 *            原始数据
 * @param keyApply
 *            Key转换器
 * @param valueApply
 *            Value转换器
 * @param <K>
 *            Key泛型
 * @param <V>
 *            Value泛型
 * @param <T>
 *            原始数据类型
 * @return Map 拆分后数据
 */
public static <T, K, V> Map<K, V> dismantling(List<T> dataSources, Function<? super T, ? extends K> keyApply,
    Function<? super T, ? extends V> valueApply) {
    return StreamBinder.dismantling(newStream(dataSources), keyApply, valueApply);
}

public static <T> void removeIf(List<T> dataSources, Predicate<T> predicate) {
    if (Objects.nonNull(dataSources) && !dataSources.isEmpty()) {
        dataSources.removeIf(predicate);
    }
}

public static <T> void removeNotIf(List<T> dataSources, Predicate<T> predicate) {
    if (Objects.nonNull(dataSources) && !dataSources.isEmpty()) {
        dataSources.removeIf(t -> !predicate.test(t));
    }
}

/**
 * 对数据流进行分组
 *
 * @param dataSources
 *            原始数据流
 * @param keyApply
 *            分组项
 * @param <K>
 *            分组项泛型
 * @return Map 分组后数据
 */
public static <K, T> Map<K, List<T>> group(List<T> dataSources, Function<? super T, ? extends K> keyApply) {
    return StreamBinder.group(newStream(dataSources), keyApply);
}

/**
 * 对数据流进行分组,并对分组后的数据进行类型转换
 *
 * @param dataSources
 *            原始数据流
 * @param keyApply
 *            分组项
 * @param valueApply
 *            数据转换器
 * @param <K>
 *            分组项泛型
 * @param <V>
 *            转换后的数据类型
 * @return Map 分组后数据
 */
public static <K, T, V> Map<K, List<V>> group(List<T> dataSources, Function<? super T, ? extends K> keyApply,
    Function<? super T, ? extends V> valueApply) {
    return StreamBinder.group(newStream(dataSources), keyApply, valueApply);
}

/**
 * 升序
 *
 * @param dataSources
 *            数据集合
 * @param compareFunction
 *            排序字段
 * @param <T>
 *            集合类型泛型类
 * @return List<T>
 */
public static <T, U extends Comparable<? super U>> List<T> sortedAsc(List<T> dataSources,
    Function<T, U> compareFunction) {
    return sorted(newStream(dataSources), compareFunction, SortStrategy.UP, SortNullStrategy.NULL_IGNORE);
}

/**
 * 升序
 *
 * @param dataStream
 *            数据集合
 * @param compareFunction
 *            排序字段
 * @param <T>
 *            集合类型泛型类
 * @return List<T>
 */
public static <T, U extends Comparable<? super U>> List<T> sortedAsc(Stream<T> dataStream,
    Function<T, U> compareFunction) {
    return sorted(dataStream, compareFunction, SortStrategy.UP, SortNullStrategy.NULL_IGNORE);
}

/**
 * 升序
 *
 * @param dataSources
 *            数据集合
 * @param compareFunction
 *            排序字段
 * @param sortNullStrategy
 *            排序控制策略
 * @param <T>
 *            集合类型泛型类
 * @return List<T>
 */
public static <T, U extends Comparable<? super U>> List<T> sortedAsc(List<T> dataSources,
    Function<T, U> compareFunction, SortNullStrategy sortNullStrategy) {
    return sorted(newStream(dataSources), compareFunction, SortStrategy.UP, sortNullStrategy);
}

/**
 * 升序
 *
 * @param dataStream
 *            数据集合
 * @param compareFunction
 *            排序字段
 * @param <T>
 *            集合类型泛型类
 * @return List<T>
 */
public static <T, U extends Comparable<? super U>> List<T> sortedAsc(Stream<T> dataStream,
    Function<T, U> compareFunction, SortNullStrategy sortNullStrategy) {
    return sorted(dataStream, compareFunction, SortStrategy.UP, sortNullStrategy);
}

/**
 * 降序
 *
 * @param dataSources
 *            数据集合
 * @param compareFunction
 *            排序字段
 * @param <T>
 *            集合类型泛型类
 * @return List<T>
 */
public static <T, U extends Comparable<? super U>> List<T> sortedDesc(List<T> dataSources,
    Function<T, U> compareFunction) {
    return sorted(newStream(dataSources), compareFunction, SortStrategy.DOWN, SortNullStrategy.NULL_IGNORE);
}

/**
 * 降序
 *
 * @param dataStream
 *            数据集合
 * @param compareFunction
 *            排序字段
 * @param <T>
 *            集合类型泛型类
 * @return List<T>
 */
public static <T, U extends Comparable<? super U>> List<T> sortedDesc(Stream<T> dataStream,
    Function<T, U> compareFunction) {
    return sorted(dataStream, compareFunction, SortStrategy.DOWN, SortNullStrategy.NULL_IGNORE);
}

/**
 * 降序
 *
 * @param dataSources
 *            数据集合
 * @param compareFunction
 *            排序字段
 * @param <T>
 *            集合类型泛型类
 * @return List<T>
 */
public static <T, U extends Comparable<? super U>> List<T> sortedDesc(List<T> dataSources,
    Function<T, U> compareFunction, SortNullStrategy sortNullStrategy) {
    return sorted(newStream(dataSources), compareFunction, SortStrategy.DOWN, sortNullStrategy);
}

/**
 * 降序
 *
 * @param dataStream
 *            数据集合
 * @param compareFunction
 *            排序字段
 * @param <T>
 *            集合类型泛型类
 * @return List<T>
 */
public static <T, U extends Comparable<? super U>> List<T> sortedDesc(Stream<T> dataStream,
    Function<T, U> compareFunction, SortNullStrategy sortNullStrategy) {
    return sorted(dataStream, compareFunction, SortStrategy.DOWN, sortNullStrategy);
}

private enum SortStrategy {
    /**
     * 升序
     */
    UP,
    /**
     * 降序
     */
    DOWN;
}

public enum SortNullStrategy {

    /**
     * 如果值为空就忽略
     */
    NULL_IGNORE,

    /**
     * 值为空就放最后
     */
    NULL_LAST,

    /**
     * 值为空就放前面
     */
    NULL_FIRST;
}

/**
 * @param dataStream
 *            数据集合
 * @param compareFunction
 *            排序字段
 * @param sortStrategy
 *            排序策略
 * @param sortStrategy
 *            true报异常
 * @param <T>
 *            集合类型泛型类
 * @return List<T>
 */
private static <T, U extends Comparable<? super U>> List<T> sorted(Stream<T> dataStream,
    Function<T, U> compareFunction, SortStrategy sortStrategy, SortNullStrategy sortNullStrategy) {
    Comparator<T> comparing;
    if (Objects.isNull(sortStrategy) || SortStrategy.UP == sortStrategy) {
        if (sortNullStrategy == SortNullStrategy.NULL_LAST) {
            comparing = Comparator.comparing(compareFunction, Comparator.nullsLast(U::compareTo));
        } else if (sortNullStrategy == SortNullStrategy.NULL_FIRST) {
            comparing = Comparator.comparing(compareFunction, Comparator.nullsFirst(U::compareTo));
        } else {
            comparing = Comparator.comparing(compareFunction);
        }
    } else {
        if (sortNullStrategy == SortNullStrategy.NULL_LAST) {
            comparing = Comparator.comparing(compareFunction, Comparator.nullsLast(U::compareTo).reversed());
        } else if (sortNullStrategy == SortNullStrategy.NULL_FIRST) {
            comparing = Comparator.comparing(compareFunction, Comparator.nullsFirst(U::compareTo).reversed());
        } else {
            comparing = Comparator.comparing(compareFunction, Comparator.reverseOrder());
        }
    }
    return dataStream.sorted(comparing).collect(Collectors.toList());
}

public static <T> boolean contains(T[] dataSources, T... elements) {
    return contains(Stream.of(dataSources), elements);
}

public static <T> boolean containsAny(T[] dataSources, T... elements) {
    return containsAny(Stream.of(dataSources), elements);
}

public static <T> boolean contains(List<T> dataSource, T... elements) {
    return contains(newStream(dataSource), elements);
}

public static <T> boolean containsAny(List<T> dataSource, T... elements) {
    return containsAny(newStream(dataSource), elements);
}

public static <T> boolean contains(Stream<T> dataStream, T... elements) {
    List<T> ts = mapToList(newStream(dataStream), Function.identity());
    if (Objects.isNull(elements)) {
        return false;
    }
    for (T element : elements) {
        if (!ts.contains(element)) {
            return false;
        }
    }
    return true;
}

public static <T> boolean containsAny(Stream<T> dataStream, T... elements) {
    List<T> ts = mapToList(dataStream, Function.identity());
    if (Objects.isNull(elements)) {
        return false;
    }
    for (T element : elements) {
        if (ts.contains(element)) {
            return true;
        }
    }
    return false;
}

@SafeVarargs
public static <T, R> List<R> merge(Function<T, R> mapper, List<T>... lists) {
    List<R> result = new ArrayList<>();
    for (R r : distinct(lists[0], mapper)) {
        boolean temp = true;
        for (int i = 1; i < lists.length; i++) {
            List<R> rs = EnhanceStream.distinct(lists[i], mapper);
            Map<R, R> reMapping = EnhanceStream.dismantling(rs, Function.identity());
            if (Objects.isNull(reMapping.get(r))) {
                temp = false;
            }
        }
        if (temp) {
            result.add(r);
        }
    }
    return EnhanceStream.distinct(result);
}

public static <T> T[] listToArr(List<T> dataList) {
    return (T[])dataList.toArray(new Object[0]);
}

public static <T> List<T> arrToList(T[] dataList) {
    return Stream.of(dataList).collect(Collectors.toList());
}

}

有兴趣可以一起交流工具类,请关注我得公众号

往事随风、无忧无恼
上一篇:Kafka 怎么顺序消费?面试必备。。。
下一篇:没有了
网友评论