首页 > 解决方案 > 如何将其转换为流表达式?(使用原子引用)

问题描述

我需要更改以下代码:

protected void checkNoDuplicateLabels( List<CompileResult> compileResult ) {
    Set<Label> infos = new HashSet<>();
    for ( AbstractTypeInfo info : infoRepo.getList() ) {
        if ( info instanceof Label ) {
            Label label = (Label) info;
            if ( infos.contains( label ) ) {
                compileResult.add( new CompileResult( Severity.FATAL, MessageFormat.format( "Duplicate label found! \n Type: '{0}' \n Language: '{1}'", label.getType(), label.getLanguage() ) ) );
            }
            infos.add( label );
        }
    }
}

成一条溪流。我知道将 Sets 与流一起使用的一种方法是实现 AtomicReferences,它将方法的第一行替换为:

AtomicReference<Set<Label>> infos = new AtomicReference<>( new HashSet<Label>() );

我怎样才能实现循环现在对流执行的相同功能?

标签: javacollectionsjava-8java-stream

解决方案


你可以不这样做AtomicReference

BinaryOperator<Label> logDuplicate = (label1, label2) -> {
    // Log label2 as duplicate
    compileResult.add(new CompileResult(Severity.FATAL, MessageFormat.format("Duplicate label found! \n Type: '{0}' \n Language: '{1}'", label2.getType(), label2.getLanguage())));
    return label1;
};

Set<Label> infos = infoRepo.getList()
                           .stream()
                           .filter(Label.class::isInstance)
                           .map(Label.class::cast)
                           .collect(toMap(identity(), identity(), logDuplicate, HashMap::new))
                           .keySet();

更新:

import static java.util.stream.Collectors.toMap;
import static java.util.function.Function.identity;

推荐阅读