流API。收集所有值​将模型类的字符串字段合并到一个集合中，然后打印

Question

//一个任务：有一个Model类，它有一个字段类型为String。有两个容器类。A类包含一个特定类型Model的对象列表和一个B类的字段。B类还包含一个对象列表Model。创建一个对象A的集合，填入它们的内部状态（列表，对象B），类似于B。将Model类的String字段的所有值收集到一个集合中，打印。这样写的：怎么写的更短？

public class Task_2 {

public static void main(String[] args) {

    List<Model> modelsForB = new ArrayList<>();
    modelsForB.add(new Model("B1 "));
    modelsForB.add(new Model("B2 "));
    B b = new B(modelsForB);

    List<Model> modelsForA = new ArrayList<>();
    modelsForA.add(new Model("A1 "));
    modelsForA.add(new Model("A2 "));
    A a = new A(modelsForA, b);

    final List<String> collected = new ArrayList<>();

    Stream.of(a)
            .flatMap(item -> Stream.of(item.b))
            .flatMap(itemB -> itemB.models.stream())
            .map(model -> model.string)
            .forEach(collected::add);

    Stream.of(a)
            .flatMap(itemA -> itemA.models.stream())
            .map(model -> model.string)
            .forEach(collected::add);
    collected.forEach(System.out::println);
}

static class Model {

    String string;

    public Model(String string) {
        this.string = string;
    }
}

static class A {

    private List<Model> models;
    B b;

    public A(List<Model> models, B b) {
        this.models = models;
        this.b = b;
    }
}

static class B {
    private List<Model> models;

    public B(List<Model> models) {
        this.models = models;
    }
}

怎么写短一点？

Answer 1

根据你的问题，你想“创建一个对象集合”，所以我会确保你的答案提供了这个，而不仅仅是一个A。

我还使用了数组。asList（用于添加的条目），只是为了简洁

public static void main(String[] args) {
    List<Model> modelsForB = new ArrayList<>();
    modelsForB.add(new Model("B1 "));
    modelsForB.add(new Model("B2 "));
    B b = new B(modelsForB);
    B b1 = new B(Arrays.asList( new Model("B1.1" ), new Model("B1.2" )));
    List<Model> modelsForA = new ArrayList<>();
    modelsForA.add(new Model("A1 "));
    modelsForA.add(new Model("A2 "));
    A a = new A(modelsForA, b);
    A a1 = new A(Arrays.asList( new Model("A1.1" ), new Model( "A1.2" )), b1 );
    List<A> listOfA = Arrays.asList( a, a1 );

    // If you need the collection, per the requirement, otherwise just use forEach to print instead of collect
    final List<String> collected = listOfA.stream()
            .flatMap( x -> Stream.concat( x.models.stream(), x.b.models.stream() ))
            .map(m -> m.string )
            .collect(Collectors.toList());
    collected.forEach(System.out::println);
}

Answer 2

静态concat方法可用于该任务：

    List<String> collected = Stream
            .concat(a.models.stream(), a.b.models.stream())
            .map(model -> model.string)
            .collect(Collectors.toList());