javassist之取上级调用方法

private static Method getInvokingMethod(int index) {
    try {
        StackTraceElement stack[] = Thread.currentThread().getStackTrace();
        StackTraceElement parent = stack[index];
        Class cls = Class.forName(parent.getClassName());
        String methodName = parent.getMethodName();
        ClassPool pool = ClassPool.getDefault();
        ClassClassPath classPath = new ClassClassPath(cls);
        pool.insertClassPath(classPath);
        CtClass cc = pool.get(cls.getName());
        Class[] paramsTypes = null;
        for (CtMethod ctMethod : cc.getMethods()) {
            MethodInfo methodInfo = ctMethod.getMethodInfo();
            if (!methodName.equals(methodInfo.getName())) {
                continue;
            }
            //int ctMethodLine = methodInfo.getLineNumber(0);
            /* 取得方法起始行和结束行,然后对比执行的代码是否在此方法区间内 */
            int ctMethodMaxLine = methodInfo.getLineNumber(Integer.MAX_VALUE);
            int ctMethodMinLine = methodInfo.getLineNumber(0);
            int methodLine = parent.getLineNumber();
            /* 探测StackTraceElement中所带的行数是否超出了方法体外*/
            if(!(methodLine <= ctMethodMaxLine && methodLine >= ctMethodMinLine)){
                continue;
            }
            CtClass[] ctClass = ctMethod.getParameterTypes();
            paramsTypes = new Class[ctMethod.getParameterTypes().length];
            for (int i = 0; i < ctClass.length; i++) {
                CtClass paramsType = ctClass[i];
                //CtPrimitiveType

                Class type = null;
                String clzzName = null;

                if(paramsType.isPrimitive()){
                    type = ClassUtils.forName(paramsType.getSimpleName(), Thread.currentThread().getContextClassLoader());
                }else{
                    type = ClassUtils.forName((paramsType.getPackageName()+"."+paramsType.getSimpleName()), Thread.currentThread().getContextClassLoader());
                }
                paramsTypes[i] = type;
            }
            break;
        }
        return ReflectionUtils.findMethod(cls,methodName,paramsTypes);
    } catch (Exception e) {
        e.printStackTrace();
    }
    return null;
}
/**
 * 取上层调用方法
 * @return
 */
public static Method getParentInvokingMethod() {
    return getInvokingMethod(4);
}

/**
 * 取得调用该方法的上层方法
 * @return
 */
public static Method getParentInvokingThisMethod() {
    return getInvokingMethod(3);
}

导包

import javassist.*;
import javassist.bytecode.CodeAttribute;
import javassist.bytecode.LocalVariableAttribute;
import javassist.bytecode.MethodInfo;
import org.springframework.util.*;

import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;
Javassist 是一个用于在运行时编辑字节码的 Java 库,它提供了一些工具类和 API,使得我们能够动态地修改类的方法、字段和构造方法。其中,ExprEditor 类是一个 Javassist 中的工具类,它提供了一些方法来编辑表达式。 ExprEditor 中的主要方法如下: 1. public void edit(MethodCall m) 该方法用于编辑方法调用表达式。我们可以通过这个方法来修改方法调用的参数、方法名或者方法所在的类名,从而实现对方法调用的修改。 2. public void edit(NewExpr e) 该方法用于编辑构造方法调用表达式。我们可以通过这个方法来修改构造方法调用的参数、构造方法所在的类名,从而实现对构造方法调用的修改。 3. public void edit(FieldAccess f) 该方法用于编辑字段访问表达式。我们可以通过这个方法来修改字段的名称、类型或者所在的类名,从而实现对字段访问的修改。 4. public void edit(MethodCall m, String classname, String methodname) 该方法用于编辑指定类中指定方法方法调用表达式。我们可以通过这个方法来修改方法调用的参数、方法名或者方法所在的类名,从而实现对指定方法调用的修改。 5. public void edit(NewExpr e, String classname) 该方法用于编辑指定类的构造方法调用表达式。我们可以通过这个方法来修改构造方法调用的参数、构造方法所在的类名,从而实现对指定类的构造方法调用的修改。 6. public void edit(FieldAccess f, String classname, String fieldname) 该方法用于编辑指定类中指定字段的字段访问表达式。我们可以通过这个方法来修改字段的名称、类型或者所在的类名,从而实现对指定类中指定字段的访问的修改。 以上就是 ExprEditor 中的主要方法。我们可以通过这些方法来实现对字节码中的表达式的动态修改,以达到一些特定的目的。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值