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



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

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>

有没有什么反射魔法可以让我从方法引用创建的函数中找到服务?

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

我知道Function总是来自这些方法引用,所以我很乐意根据需要进行降频转换。

这与相关的问题不同。是否可以将方法引用转换为MethodHandle?因为,首先,这不是同一个问题,只是在一个相关的领域。即使我能得到一个MethodHandle,我也无法从中得到目标

使用这篇SO文章中的技巧,您可以找到目标。下面的重要方法是findTarget。事实证明,Lambda确实捕获了它们的目标,并且您可以从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> {}
}

没有直接的方法可以找到目标,因为方法引用只是被翻译成lambdas(根据定义,lambdas是匿名的)。因此,您需要使用一种变通方法。

假设您熟悉Java7的代理,因为您已经成功地实现了mock工厂方法。

解决方法是,当有人调用你的allowing方法时,你设置某种全局标志来提醒所有mock你想记录下一次调用,然后你调用你得到的lambda。通过查看哪个mock记录了调用,您现在已经找到了方法引用的目标,您可以取消设置全局标志,继续执行mocking框架的其余部分。

我知道这很难看。

相关内容

  • 没有找到相关文章