Как преобразовать несколько списков в один, используя Java потоков? - PullRequest
24 голосов
/ 25 мая 2020

У меня есть класс A с несколькими членами List.

class A {
    List<X> xList;
    List<Y> yList;
    List<Z> zList;
    
    // getters and setters
}

class X {
     String desc;
     String xtype;
    
     // getters and setters    
}

class Y {   
    String name;
    String ytype;
    
    //getters and setters
}

class Z {
    String description;
    String ztype;
    
    // getters and setters    
}

И класс B всего с двумя атрибутами:

class B {
    String name;
    String type;     
}

Мне нужно повторить через различные списки в классе A, создайте объект класса B и добавьте в список следующим образом:

public void convertList(A a) {  
   List<B> b = new ArrayList<>();
    
    if (!a.getXList().isEmpty()) {
      for (final X x : a.getXList()) {
           b.add(new B(x.getDesc(), x.getXType()));
      }
    }
    
    if (!a.getYList().isEmpty()) {
      for (final Y y : a.getYList()) {
           b.add(new B(y.getName(), y.getYType()));
      }
    }
    
    if (!a.getZList().isEmpty()) {
      for (final Z z : a.getZList()) {
           b.add(new B(z.getDescription(), z.getZType()));
      }
    }    
}

Поскольку здесь повторяются циклы if и for.

Как могу ли я добиться этого, используя потоки Java?

Примечание: нет никакой связи между классами X, Y и Z, и нет общего интерфейса.

Ответы [ 4 ]

19 голосов
/ 25 мая 2020

Поскольку ваши типы X, Y и Z не имеют общего супертипа, вам необходимо преобразовать их в какой-нибудь общий тип, например Map.Entry<String,String>.

You может создать Stream всех пар имен и типов, а затем сопоставить его с экземплярами B:

List<B> b =
    Stream.of(
        a.getXList().stream().map(x -> new SimpleEntry<>(x.getDesc(),x.getXType())),
        a.getYList().stream().map(y -> new SimpleEntry<>(y.getName(),y.getYType())),
        a.getZList().stream().map(z -> new SimpleEntry<>(z.getDescription(),z.getZType())))
          .flatMap(Function.identity())
          .map(e -> new B(e.getKey(), e.getValue()))
          .collect(Collectors.toList());

Или напрямую сгенерировать B экземпляров:

List<B> b =
    Stream.of(
        a.getXList().stream().map(x -> new B(x.getDesc(),x.getXType())),
        a.getYList().stream().map(y -> new B(y.getName(),y.getYType())),
        a.getZList().stream().map(z -> new B(z.getDescription(),z.getZType())))
          .flatMap(Function.identity())
          .collect(Collectors.toList());
14 голосов
/ 25 мая 2020

Вы можете использовать Stream.concat(), как показано ниже

public List<B> convertList (A a) {
    return Stream.concat(Stream.concat(a.getXList().stream().map(x -> new B(x.getDesc(), x.getXType()))
            , a.getYList().stream().map(y -> new B(y.getName(), y.getYType())))
            , a.getZList().stream().map(z -> new B(z.getDescription(), z.getZType()))).collect(Collectors.toList());
}
7 голосов
/ 25 мая 2020

Поскольку у вас нет общего интерфейса, вам придется использовать метод forEach для перебора каждого списка.

a.getXList().forEach(i -> b.add(new B(i.getDesc(), i.getXType())));
a.getYList().forEach(i -> b.add(new B(i.getName(), i.getYType())));
a.getZList().forEach(i -> b.add(new B(i.getDescription(), i.getZType())));
5 голосов
/ 25 мая 2020

Вы используете разные свойства для классов X, Y, Z и не имеете общего интерфейса, вы можете добавлять их по одному в список.

b.addAll(a.getXList().stream().map(x ->new B(x.getDesc(), x.getXType())).collect(Collectors.toList()));
b.addAll(a.getYList().stream().map(y ->new B(y.getName(), y.getYType())).collect(Collectors.toList()));
b.addAll(a.getZList().stream().map(z ->new B(z.getDescription(), z.getZType())).collect(Collectors.toList()));
...