获取列表中具有特定泛型类型的元素

Question

我目前正在做一个火车项目，我有以下问题：

我将所有机车车辆保存在一个列表中：为了更好地理解我的类层次结构，这里是一个简化的继承概述：

RollingStock
    Engine
        SteamEngine
        DieselEngine
        ...
    Coach
        FreightCoach
        PassengerCoach
        ...
    TrainSet

在我的寄存器中，我想将所有机车车辆保存在一个列表private List<RollingStock> rollingStock; . 到目前为止，我已经为每种机车车辆类型（发动机、客车、火车组）创建了一个列表。 但是，我需要仅使用其 ID 删除某个机车车辆，因此将所有内容保存在一个列表中会更容易。

之前，我创建了一个这样的引擎：

    public void createEngine(Engine engine) {
        this.engines.add(engine);
    }

现在，只有一个列表，我这样做：

    public void createEngine(Engine engine) {
        this.rollingStocks.add(engine);
    }

这工作得很好。 对于 returnEngines() 方法，我似乎没有找到解决方案：

每个机车车辆类型都有一个列表，就这么简单：

    public List<Engine> returnEngines() {
        return engines;
    }

现在，我必须从机车车辆列表中过滤掉所有引擎：

    public List<Engine> returnEngines() {
        ...
        return rollingStock.???;

    }

我可以添加方法public String getType()并检查它的类型。

我无法想象没有更好的解决方案。

如何才能做到这一点？

Answer 1

流式传输列表，过滤Engine实例； 将实例从RollingStock映射到Engine （使用演员表），将结果收集到一个新的List 。 喜欢，

public List<Engine> returnEngines() {
    return rollingStocks.stream().filter(x -> x instanceof Engine)
            .map(x -> (Engine) x).collect(Collectors.toList());
}

Answer 2

Elliott Frisch 的答案是完全有效的，这是一个通用的解决方案，以防您还需要一种方法来过滤您的寄存器以适应任何类型的层次结构：

public <T> List<T> returnClazz(Class<T> clazz) {
    return rollingStocks.stream()
            .filter(clazz::isInstance)
            .map(clazz::cast)
            .collect(Collectors.toList());
}

然后可以由几种辅助方法使用，例如

public List<TrainSet> returnTrainset() {
    return returnClazz(TrainSet.class);
}

public List<Engines> returnEngines() {
    return returnClazz(Engine.class);
}
...

或者可以直接调用。