Java Stream 有可能吗

Question

我只是要求兴趣。是否可以通过使用 Java 的 Stream 来制作这样的东西？ （应该是 Toys.stream()...）

Set<String> colors = new HashSet<>();

toys.forEach((toy) -> {
    if (toy.getType() == Toys.BIKE) {
        colors.add(((Bike) toy.getData()).getFrameColor());
    } else {
        colors.add(((Skateboard) toy.getData()).getColor());
    }
});

我应该提到，我不可能将颜色存储在超类中。

Answer 1

这不是利用继承的正确方法。更好的方法是有类似的东西

abstract class Toy {
  abstract Color getActualColor();
}

class Bike {
  @Override Color getActualColor() { return getFrameColor(); }
}

class Skateboard {
  @Override Color getActualColor() { return getColor(); }
}

...
toys.stream().map(Toy::getActualColor).collect(Collectors.toSet());

这利用了多态性，因此您不必在运行时检查实例的类型，这在 99% 的情况下都是不好的。请注意，您的设计似乎在Toy 中包含按组合排列的数据，有点

class Toy {
  ToyData data;
}

但设计没有改变，你只有两次映射，例如stream().map(Toy::getData).map(ToyData::getActualColor)。

根据您的编辑，解决方案必须使用一组接口来标记各种功能，例如：

class Toy
{
    final ToyData data;

    public Toy(ToyData data) { this.data = data; }

    public ToyData getData() { return data; }
}

interface ToyData { } 

interface Colored
{
    Color getColor();
}

class Bike implements ToyData, Colored
{
    @Override public Color getColor() { return Color.WHITE; }
}

class Skateboard implements ToyData, Colored
{
    @Override public Color getColor() { return Color.BLACK; }
}

class Ship implements ToyData { }


public static void main (String[] args) throws java.lang.Exception
{
    List<Toy> toys = new ArrayList<>();

    toys.add(new Skateboard());
    toys.add(new Bike());
    toys.add(new Ship());

    Set<Color> = toys.stream()
                     .map(Toy::getData)
                     .filter(t -> t instanceof Colored)
                     .map(t -> ((Colored)t).getColor())
                     .collect(Collectors.toSet());
}