如何将Java的mapHeaderIndex方法泛型化以适配多个枚举类？

你的泛型方法主要存在两个核心问题，映射方向完全颠倒，还包含多余的泛型参数，咱们一步步拆解说明：

1. 映射方向完全搞反 #

原来的mapHeaderIndex方法是把流中的String类型header转换为对应的枚举实例（比如SomeHeaderEnum），但你定义的泛型参数Function<? super T, ? extends K>是接收T类型输出K类型，这和实际需求正好相反——我们需要的是接收String输出T（目标枚举类型）的函数。

2. 冗余的泛型参数K #

你给方法定义了<T, K>两个泛型参数，但整个方法逻辑里根本没用到K，完全是多余的，只需要保留T来表示目标枚举类型即可。

修复后的泛型方法 #

调整泛型参数和函数类型后，正确的方法实现如下：

protected <T> Map<T, Integer> mapHeaderIndex(Function<String, T> headerToEnumMapper) {
    AtomicInteger index = new AtomicInteger();
    return HEADERS.stream()
            .collect(toMap(
                    headerToEnumMapper,
                    s -> index.getAndIncrement(),
                    (oldV, newV) -> newV
            ));
}

调用示例 #

现在你可以轻松处理两个枚举类了：

// 生成SomeHeaderEnum对应的索引Map
Map<SomeHeaderEnum, Integer> someHeaderMap = mapHeaderIndex(SomeHeaderEnum::getFieldByHeader);
// 生成AnotherHeaderEnum对应的索引Map
Map<AnotherHeaderEnum, Integer> anotherHeaderMap = mapHeaderIndex(AnotherHeaderEnum::getFieldByHeader);

可选进阶优化：统一枚举行为 #

如果后续还有更多类似的枚举类，你可以定义一个公共接口来规范行为，避免重复代码：

public interface HeaderMappable {
    String getHeader();

    static <T extends Enum<T> & HeaderMappable> T fromHeader(String header) {
        String normalizedHeader = header.trim().toUpperCase();
        return Arrays.stream(EnumSet.allOf((Class<T>) HeaderMappable.class))
                .filter(e -> e.getHeader().toUpperCase().equals(normalizedHeader))
                .findFirst()
                .orElseThrow(() -> new IllegalArgumentException("Header not found: " + header));
    }
}

让你的枚举实现这个接口：

public enum SomeHeaderEnum implements HeaderMappable {
    SOME_A("A header"), SOME_B("B header"), SOME_C("C header"), SOME_D("D header");

    private String header;

    private SomeHeaderEnum(String header) {
        this.header = header;
    }

    @Override
    public String getHeader() {
        return header;
    }
}

这样调用时会更统一：mapHeaderIndex(SomeHeaderEnum::fromHeader)。

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