如何找到Java8方法引用的目标?

时间:2015-07-02 07:26:16

标签: java reflection java-8

我想捕获对模拟对象的调用

public interface Service {
    public String stringify(Object o);
}
service = mockery.mock(Service.class);
mockery.allowing(service::stringify).with(42).will(() -> "42");

所以在allowing内我有一个Function<Object, String>

是否有任何reflecto-magic可以让我从方法参考创建的函数中找到服务?

public WithClause allowing(Function<T,R> f) {
    Object myServiceBackAgain = findTargetOf(function);
    ....
}

我知道函数将始终来自这些方法引用,所以我很乐意尽可能地向下投射。

这与相关Is it possible to convert method reference to MethodHandle?的问题不同,因为一开始它不是同一个问题,只是在相关领域。即使我可以获得MethodHandle,我也无法从中获取目标。

2 个答案:

答案 0 :(得分:6)

使用this SO post中的技巧,您可以找到目标。以下重要方法是findTarget。事实证明,lambdas确实捕获了他们的目标,你可以从SerializedLambda访问它们。

然而,这是一个非常讨厌的反思黑客,它可能会在未来的版本中破解。我不宽恕它的使用。

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

public class FindMethodReferenceTarget {
  public static void main(String[] args) {
    String s = "123";
    Optional<Object> target = findTarget(s::charAt);
    System.out.println(target.get().equals(s));

    Object o = new FindMethodReferenceTarget();
    target = findTarget(o::equals);
    System.out.println(target.get().equals(o));
  }

  private static <T, R> Optional<Object> findTarget(
      DebuggableFunction<T, R> methodReference) {
    return getLambda(methodReference).map(l -> l.getCapturedArg(0));
  }

  private static Optional<SerializedLambda> getLambda(Serializable lambda) {
    for (Class<?> cl = lambda.getClass(); cl != null; cl = cl.getSuperclass()) {
      try {
        Method m = cl.getDeclaredMethod("writeReplace");
        m.setAccessible(true);
        Object replacement = m.invoke(lambda);
        if (!(replacement instanceof SerializedLambda)) {
          break; // custom interface implementation
        }
        SerializedLambda l = (SerializedLambda) replacement;
        return Optional.of(l);
      } catch (NoSuchMethodException e) {
        // do nothing
      } catch (IllegalAccessException | InvocationTargetException e) {
        break;
      }
    }

    return Optional.empty();
  }

  @FunctionalInterface
  private static interface DebuggableFunction<T, R> extends
      Serializable,
      Function<T, R> {}
}

答案 1 :(得分:0)

没有直接的方法来查找目标,因为方法引用只是被转换为lambdas(根据定义,匿名)。所以你需要使用一种解决方法。

大概你熟悉Java 7的代理,因为你已经设法实现了mock工厂方法。

然后解决方法是,当有人调用您的allowing方法时,您设置某种全局标志以警告您要记录下一个调用的所有模拟,然后调用您给出的lambda。通过查看哪个模拟记录了调用,您现在已经找到了方法引用的目标,并且您可以取消设置全局标志并继续使用其余的模拟框架。

这很难看,我知道。