在'之内更换if-else循环使用Java-8 Streams

时间:2015-07-24 11:43:44

标签: java functional-programming java-8

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

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

是否可以轻松地用流替换此循环?我看到的唯一选择是使用两种不同的过滤条件迭代流两次。

1 个答案:

答案 0 :(得分:23)

听起来你可以使用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
        }
    }
}