Java如何传递不同子类的方法引用实现目标调用效果

你想要的调用语法是可行的，问题主要出在泛型擦除特性、Map取值逻辑、类型安全转换这几个关键点上。

现有代码的核心问题 #

1. Map取值逻辑错误：你的Map中存储的key是ADerived.class、BDerived.class这类子类的Class对象，而现有代码中用Base.class作为key去查询，必然返回null，这是运行结果不符合预期的直接原因。
2. 泛型擦除限制：Java泛型在运行时会被擦除，你直接用Function<R, String>作为参数，运行时无法获取到泛型参数R的实际类型，没法确定需要从Map中取出哪个子类的实例。
3. 类型转换不安全：无校验的强制类型转换(R) map.get(...)很容易触发ClassCastException，没有类型安全保障。

解决方案 #

方案一：显式传入类型令牌（推荐，稳定无兼容问题） #

仅需修改getValue方法的参数，额外传入对应子类的Class作为类型令牌，逻辑清晰且类型安全：
首先修改User类的getValue实现：

public class User {
    public Map<Class<? extends Base>, Base> map;

    public User() {
        map = new HashMap<>();
        // 初始化时以子类Class为key存储实例
        map.put(ADerived.class, new ADerived());
        map.put(BDerived.class, new BDerived());
    }

    public <T extends Base> String getValue(Class<T> clazz, Function<T, String> func) {
        // 安全类型转换
        T instance = clazz.cast(map.get(clazz));
        if (instance == null) {
            throw new IllegalArgumentException("不存在对应类的实例：" + clazz.getName());
        }
        return func.apply(instance);
    }
}

调用方式只需额外传入子类Class参数即可：

User user = new User();
user.getValue(ADerived.class, ADerived::getPropertyFromA); // 返回Test A
user.getValue(BDerived.class, BDerived::getPropertyFromB); // 返回Test B

方案二：保持原有调用语法（tricky实现，依赖JDK内部特性） #

如果你完全不想修改原有调用语法，可以通过反射解析序列化的方法引用获取目标类，需要自定义可序列化的Function接口：

import java.io.Serializable;
import java.lang.invoke.SerializedLambda;
import java.lang.reflect.Method;
import java.util.function.Function;

@FunctionalInterface
public interface SerializableFunction<T, R> extends Function<T, R>, Serializable {
    // 解析方法引用对应的目标类
    default Class<?> getTargetClass() {
        try {
            Method writeReplace = this.getClass().getDeclaredMethod("writeReplace");
            writeReplace.setAccessible(true);
            SerializedLambda lambda = (SerializedLambda) writeReplace.invoke(this);
            return Class.forName(lambda.getImplClass().replace('/', '.'));
        } catch (Exception e) {
            throw new RuntimeException("解析方法引用目标类失败", e);
        }
    }
}

修改User的getValue方法：

@SuppressWarnings("unchecked")
public <T extends Base> String getValue(SerializableFunction<T, String> func) {
    Class<T> clazz = (Class<T>) func.getTargetClass();
    T instance = clazz.cast(map.get(clazz));
    if (instance == null) {
        throw new IllegalArgumentException("不存在对应类的实例：" + clazz.getName());
    }
    return func.apply(instance);
}

此时你可以完全保持原有调用语法：

User user = new User();
user.getValue(ADerived::getPropertyFromA); // 返回Test A
user.getValue(BDerived::getPropertyFromB); // 返回Test B

该方案的局限性：

• 仅支持直接的方法引用，不支持普通lambda表达式、包装后的函数实例
• 依赖JDK内部的SerializedLambda实现，跨JDK版本可能存在兼容问题
• 反射操作有额外性能开销，不适合高性能场景

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