Java8:流映射同一个流中的两个属性

时间:2018-03-29 10:31:59

标签: java java-8 java-stream

我有一个具有以下签名的班级Model

class Model {
        private String stringA;
        private String stringB;

        public Model(String stringA, String stringB) {
            this.stringA = stringA;
            this.stringB = stringB;
        }

        public String getStringA() {
            return stringA;
        }

        public String getStringB() {
            return stringB;
        }
}

我想将List<Model>映射到包含stringA和stringB的List<String>

List<String> strings = models.stream()
                .mapFirst(Model::getStringA)
                .thenMap(Model::getStringB)
                .collect(Collectors.toList());

或:

List<String> strings = models.stream()
                .map(Mapping.and(Model::getStringA,Model::getStringB))
                .collect(Collectors.toList());

当然,没有人会编译,但你明白了。

有可能吗?

修改

示例:

Model ab = new Model("A","B");
Model cd = new Model("C","D");
List<String> stringsFromModels = {"A","B","C","D"};

2 个答案:

答案 0 :(得分:13)

您可以一个接一个地列出所有值,如下所示:

List<String> resultSet =
               modelList.stream()
                        .flatMap(e -> Stream.of(e.getStringA(), e.getStringB()))
                        .collect(Collectors.toList());

传递给flatMap的函数将应用于每个元素,产生Stream<String>。此处必须使用flatMap将所有嵌套Stream<Stream<String>>折叠为Stream<String>,从而使我们能够将元素收集到List<String>

答案 1 :(得分:5)

flatMap是你的朋友:

models
    .stream()
    .flatMap(model -> Stream.of(model.getStringA(),model.getStringB()))
    .collect(Collectors.toList());

flatMap采用类型R并期望返回新类型RR的Stream(来自列表,集合,数组)。对于每个1模型,您会获得n个新元素(在本例中为StringAStringB):

{model_1[String_A1,String_B1] , model_2[String_A2,String_B2] , model_3[String_A3,String_B3]}

所有n元素 { [String_A1,String_B1], [String_A2,String_B2],[String_A3,String_B3]}

然后展平,放入具有结构的新流

{String_A1,String_B1,String_A2,String_B2,String_A3,String_B3}

类型字符串。这就是你有一个新的流。

您可以使用flatMap,例如,当您有许多需要合并为一个元素的集合/流时。有关更简单的说明,请查看此answer