如何使用可变参数调用 MethodHandle

Question

我正在尝试用 MethodHandle 替换反射调用，但 varargs 似乎无法处理。

我的反射调用程序当前如下所示：

public class Invoker {

    private final Method delegate;

    public Invoker(Method delegate) {
        this.delegate = delegate;
    }

    public Object execute(Object target, Object[] args) {
        return delegate.invoke(target, args);
    }
}

我目前重写它的尝试看起来像这样（Invoker 暴露的接口必须保持不变）：

public class Invoker {

    private final Method delegate;
    private final MethodHandle handle;

    public Invoker(Method delegate) {
        this.delegate = delegate;
        this.handle = MethodHandles.lookup().unreflect(delegate);
    }

    public Object execute(Object target, Object[] args) throws InvocationTargetException, IllegalAccessException {
        Object[] allArgs = Stream.concat(Stream.of(target), Stream.of(args)).toArray(Object[]::new);
        return handle.invokeWithArguments(allArgs);
    }
}

这在大多数情况下都可以正常工作。但是可变参数打破了一切。 例如。有这样的方法：

public String test(int i, String... args) {
    return ...;
}

还有如下参数：

Object[] args = new Object[] {10, new String[] {"aaa", "bbb"}};

上面实现的execute 将失败。我尝试了asSpreader()、MethodHandles.explicitCastArguments()、invoke 而不是invokeWithArguments 等的各种组合，但没有成功。

我可以调用可变参数方法的唯一方法是提供内联参数而不是数组。例如

handle.invokeWithArguments(10, "aaa", "bbb")

但我不能这样做并保持它当前拥有的Invoker 的通用性质。

这真的不可能按照我尝试的方式进行吗？

更新： 在对各种场景进行基准测试后，我决定坚持使用反射，因为invokeWithArguments 在所有测试用例中的表现要差得多。

Answer 1

似乎您只需要一次调用.asFixedArity，因为默认情况下java将使用asVarargsCollector创建方法句柄

public class Main {
    public static String test(int i, String... args) { return "works!"; }

    public static void main(String[] args) throws Throwable {
        Method method = Main.class.getMethod("test", int.class, String[].class);
        System.out.println(new Invoker(method).execute(null, new Object[]{1, new String[] {"foo", "bar"} }));
    }

    public static class Invoker {
        private final MethodHandle handle;

        public Invoker(final Method delegate) throws Exception {
            MethodHandle handle = MethodHandles.lookup().unreflect(delegate);
            if (Modifier.isStatic(delegate.getModifiers())) { // for easy static methods support
                handle = MethodHandles.dropArguments(handle, 0, Object.class);
            }
            this.handle = handle.asFixedArity();
        }

        public Object execute(Object target, Object[] args) throws Throwable {
            Object[] allArgs = new Object[args.length + 1];
            allArgs[0] = target;
            System.arraycopy(args, 0, allArgs, 1, args.length);
            return handle.invokeWithArguments(allArgs);
        }
    }
}

还有许多其他可能的解决方案，例如您可以向 Invoker 构造函数添加更多逻辑（静态工厂可能是个好主意）并使用 asType 方法准备您想要的签名，然后您应该可以使用.invokeExact 获得小的性能提升。

你也可以继续使用Method ;)

Answer 2

我完成了您的代码以重现您的问题，但它可以使用 invokeWithArguments。也许我错过了什么？

public class Main {

    public String test(int i, String... args) {
        return i + Stream.of(args).collect(Collectors.joining());
    }


    public static void main(String[] args) throws Throwable {
        Main main = new Main();
        Method method = Main.class.getMethod(
            "test",
            int.class,
            String[].class)
        Invoker invoker = new Invoker(method);
        assertEquals("1foobar", invoker.execute(main, new Object[]{1, "foo", "bar"})); // Success
    }


    public static class Invoker {

        private final MethodHandle handle;

        public Invoker(final Method delegate) throws Exception {
            this.handle = MethodHandles.lookup().unreflect(delegate);
        }

        public Object execute(Object target, Object[] args) throws Throwable {
            // Add the target and all arguments in a new array
            Object[] allArgs = Stream.concat(Stream.of(new Object[]{target}), Stream.of(args))
                .toArray(Object[]::new);
            return handle.invokeWithArguments(allArgs);
        }
    }
}