将 “for” 循环中的 if-else 替换为 Java-8 Streams

2022-09-03 13:10:52

我有以下简单的代码,我试图将其转换为功能样式

for(String str: list){
    if(someCondition(str)){
       list2.add(doSomeThing(str));
    }
    else{
        list2.add(doSomethingElse(str));
    }
}

是否可以很容易地将此循环替换为流?我看到的唯一选择是使用两个不同的筛选条件对流进行两次迭代。


答案 1

这听起来像是你可以只用一个条件:map

List<String> list2 = list
    .stream()
    .map(str -> someCondition(str) ? doSomething(str) : doSomethingElse(str))
    .collect(Collectors.toList());

将短字符串映射到小写字母,将长字符串映射到大写的短字符串示例:

import java.util.*;
import java.util.stream.*;

public class Test {

    public static void main(String[] args) {
        List<String> list = Arrays.asList("abC", "Long Mixed", "SHORT");
        List<String> list2 = list
            .stream()
            .map(str -> str.length() > 5 ? str.toUpperCase() : str.toLowerCase())
            .collect(Collectors.toList());
        for (String result : list2) {
            System.out.println(result); // abc, LONG MIXED, short
        }
    }
}

答案 2