获取调用方方法(java.lang.reflect.Method)

获取调用方方法(java.lang.reflect.Method),第1张

获取调用方方法(java.lang.reflect.Method)

如果仅用于测试,则可能有效。它假定可以通过调用类的类访问类文件

ClassLoader
,并且类文件已使用调试符号进行编译(我希望它们用于测试!)。此代码依赖于ASM字节码库。

public static Method getMethod(final StackTraceElement stackTraceElement) throws Exception {    final String stackTraceClassName = stackTraceElement.getClassName();    final String stackTraceMethodName = stackTraceElement.getMethodName();    final int stackTraceLineNumber = stackTraceElement.getLineNumber();    Class<?> stackTraceClass = Class.forName(stackTraceClassName);    // I am only using AtomicReference as a container to dump a String into, feel free to ignore it for now    final AtomicReference<String> methodDescriptorReference = new AtomicReference<String>();    String classFileResourceName = "/" + stackTraceClassName.replaceAll("\.", "/") + ".class";    InputStream classFileStream = stackTraceClass.getResourceAsStream(classFileResourceName);    if (classFileStream == null) {        throw new RuntimeException("Could not acquire the class file containing for the calling class");    }    try {        ClassReader classReader = new ClassReader(classFileStream);        classReader.accept(     new EmptyVisitor() {         @Override         public MethodVisitor visitMethod(int access, final String name, final String desc, String signature, String[] exceptions) {  if (!name.equals(stackTraceMethodName)) {      return null;  }  return new EmptyVisitor() {      @Override      public void visitLineNumber(int line, Label start) {          if (line == stackTraceLineNumber) {   methodDescriptorReference.set(desc);          }      }  };         }     },     0 );    } finally {        classFileStream.close();    }    String methodDescriptor = methodDescriptorReference.get();    if (methodDescriptor == null) {        throw new RuntimeException("Could not find line " + stackTraceLineNumber);    }    for (Method method : stackTraceClass.getMethods()) {        if (stackTraceMethodName.equals(method.getName()) && methodDescriptor.equals(Type.getMethodDescriptor(method))) { return method;        }    }    throw new RuntimeException("Could not find the calling method");}


欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/zaji/5488837.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-12
下一篇 2022-12-12

发表评论

登录后才能评论

评论列表(0条)

保存