stream流根据对象指定字段去重或者获取重复数据

代码

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Function;
import java.util.function.Predicate;

/**
 * 操作去重工具类
 *
 * @author chengxi
 * @date 2020/3/17 17:15
 */
public class DeduplicationUtil {

    /**
     * 自定义函数去重(采用 Predicate函数式判断,采用 Function获取比较key)
     * 内部维护一个 ConcurrentHashMap,并采用 putIfAbsent特性实现
     *
     * @param keyExtractor
     * @param 
     * @return
     */
    public static  Predicate distinctByKey(Function keyExtractor) {
        Map seen = new ConcurrentHashMap<>();
        return t -> seen.putIfAbsent(keyExtractor.apply(t), Boolean.TRUE) == null;
    }

    /**
     * 只获取重复的数据
     * 
     * @param keyExtractor
     * @param 
     * @return
     */
    public static  Predicate distinctNotByKey(Function keyExtractor) {
        Map seen = new ConcurrentHashMap<>();
        return t -> seen.putIfAbsent(keyExtractor.apply(t), Boolean.TRUE) != null;
    }

}

示例

// 注意需要关流之后,才会完成去重,如果继续操作流可能拿到未去重的数据
List logEchoVOList = traceLogEchoVOList.stream()
         .filter(DeduplicationUtil.distinctByKey(TraceLogEchoVO::getVisitorNo))
         .collect(Collectors.toList());

后语
注意需要关流之后,才会完成去重,如果继续操作流可能会拿到未去重的数据。
 

你可能感兴趣的:(java,java)