如何通过AntType枚举实现返回对应Ant子类泛型List的Java方法

问题背景 #

需要实现方法从List<Ant>中过滤出指定Ant子类的实例，返回对应子类的泛型List（如List<AntScout>），要求参数使用已有的AntType枚举（无法替换为Class<T>参数），同时解决枚举类中Raw Class的警告。

第一步：修复AntType枚举的Raw Class警告 #

原枚举的getClass方法存在Raw类型警告，直接让每个枚举实例绑定对应的子类Class对象即可解决，优化后的枚举类如下：

public enum AntType {
    QUEEN(AntQueen.class),
    WARRIOR(AntWarrior.class),
    GATHERER(AntGatherer.class),
    SCOUT(AntScout.class);

    private final Class<? extends Ant> antClass;

    AntType(Class<? extends Ant> antClass) {
        this.antClass = antClass;
    }

    public Class<? extends Ant> getAntClass() {
        return antClass;
    }
}

• 每个枚举实例在构造时传入对应子类的Class对象，避免了switch分支的冗余
• 使用Class<? extends Ant>替代raw类型的Class，彻底消除警告

第二步：实现返回泛型子类List的方法 #

通过泛型方法结合枚举的类型绑定，实现类型安全的过滤与转换：

@SuppressWarnings("unchecked")
public <T extends Ant> List<T> getAntsByType(AntType type) {
    return ants.stream()
            .filter(ant -> ant.getType() == type)
            .map(ant -> (T) ant)
            .toList();
}

说明：

1. 泛型声明<T extends Ant>表示方法返回List<T>，T是Ant的任意子类
2. @SuppressWarnings("unchecked")用于消除编译器的类型转换警告——由于我们通过ant.getType() == type确保了过滤后的实例必然是对应子类，因此强制转换是安全的
3. 调用示例：

// 编译器自动推断返回List<AntScout>
List<AntScout> scoutAnts = getAntsByType(AntType.SCOUT);
// 显式指定泛型类型（可选）
List<AntWarrior> warriorAnts = this.<AntWarrior>getAntsByType(AntType.WARRIOR);

内容的提问来源于stack exchange，提问作者Wewius