如何识别作为BiFunction的Java非静态方法引用类型

原Java类代码 #

class Abc {
    private String extraInfo;
    public Abc(String extraInfo) {
        this.extraInfo = extraInfo;
    }

    // 假设Input1、Input2继承自同一父类Input
    public Input2 method1(Input1 input) {
        // 基于输入执行操作
        return null;
    }

    public Input1 method2(Input2 input) {
        return null;
    }
}

需求说明 #

希望编写一个通用方法，接收上述类中方法的引用（以BiFunction形式传入），判断该引用对应method1还是method2，示例代码如下：

void testSuccess(BiFunction<Abc, I, O> func) {
    Abc obj = new Abc("text");
    func.apply(obj, decideInput(func));
}

需要解决的问题 #

1. 实现decideInput方法：根据传入的方法引用，对应method1则返回Input1对象，对应method2则返回Input2对象；
2. 若Input1、Input2未继承自同一父类Input，实现decideInput是否会存在问题。

注：已知func可能是lambda表达式，这会增加识别实际方法引用的难度。

解决方案 #

问题1：实现decideInput方法 #

Java的lambda和方法引用在运行时会被封装为匿名类，常规方式无法直接获取目标方法信息，仅当传入的是直接方法引用（非lambda）时，可通过反射解析实现：

import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.function.BiFunction;

@SuppressWarnings("unchecked")
private static <I> I decideInput(BiFunction<Abc, I, ?> func) {
    try {
        // 方法引用类会包含target字段，用于存储目标方法的元数据
        Field targetField = func.getClass().getDeclaredField("target");
        targetField.setAccessible(true);
        Object target = targetField.get(func);

        // 获取方法引用对应的实际Method对象
        Method invokeMethod = func.getClass().getDeclaredMethod("invoke", Object.class, Object.class);
        Method actualMethod = (Method) invokeMethod.invoke(target);

        // 根据方法名和参数类型判断目标方法
        if ("method1".equals(actualMethod.getName()) 
            && actualMethod.getParameterTypes()[0] == Input1.class) {
            return (I) new Input1();
        } else if ("method2".equals(actualMethod.getName()) 
                   && actualMethod.getParameterTypes()[0] == Input2.class) {
            return (I) new Input2();
        }
    } catch (NoSuchFieldException | IllegalAccessException | NoSuchMethodException | InvocationTargetException e) {
        // 处理lambda或非直接方法引用的情况
        throw new IllegalArgumentException("无法识别目标方法引用", e);
    }
    throw new IllegalArgumentException("不支持的方法引用类型");
}

注意：该实现依赖JVM对方法引用的内部实现细节，不同JDK版本可能存在兼容性问题，生产环境需谨慎使用。如果传入的是lambda表达式，反射会直接失败，因为lambda没有对应的target字段和方法元数据。

问题2：Input1、Input2无共同父类时的问题 #

如果Input1和Input2没有共同父类或接口，会出现以下问题：

1. 编译错误：testSuccess方法的泛型I无法同时兼容Input1和Input2两种类型，编译器会提示类型不匹配；
2. 运行时风险：即使通过强制类型转换绕过编译错误，decideInput返回的对象无法安全转换为泛型I，可能触发ClassCastException。

解决办法是为Input1和Input2定义一个共同的标记接口（如InputMarker），让两者都实现该接口，将testSuccess的泛型I指定为InputMarker，既解决编译问题，也保证类型转换的安全性。

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