java 动态注入代码

Java 动态注入代码可以使用反射或者动态生成字节码的方式实现。

1. 反射方式

反射是 Java 中的一种机制，可以在运行时动态地获取类的信息，并对类的属性、方法等进行操作。通过反射可以动态地加载、创建、调用类和对象，从而实现动态注入代码的功能。

以注入一个方法为例：

// 获取 Class 对象
Class<?> clazz = Class.forName("com.example.TestClass");

// 获取方法对象
Method method = clazz.getDeclaredMethod("testMethod", String.class);

// 设置方法为可访问
method.setAccessible(true);

// 执行方法
method.invoke(clazz.newInstance(), "hello");

2. 字节码生成方式

字节码生成是指在运行时通过程序生成 Java 字节码，然后将其加载到 JVM 中执行。通过字节码生成可以实现动态生成类、方法等功能，从而实现动态注入代码的功能。

以使用 ASM 生成一个类为例：

// 生成 ClassWriter 对象
ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS | ClassWriter.COMPUTE_FRAMES);

// 定义类
cw.visit(Opcodes.V1_8, Opcodes.ACC_PUBLIC, "com/example/TestClass", null, "java/lang/Object", null);

// 定义方法
MethodVisitor mv = cw.visitMethod(Opcodes.ACC_PUBLIC, "testMethod", "(Ljava/lang/String;)V", null, null);
mv.visitCode();
mv.visitFieldInsn(Opcodes.GETSTATIC, "java/lang/System", "out", "Ljava/io/PrintStream;");
mv.visitVarInsn(Opcodes.ALOAD, 1);
mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/io/PrintStream", "println", "(Ljava/lang/String;)V", false);
mv.visitInsn(Opcodes.RETURN);
mv.visitMaxs(1, 1);
mv.visitEnd();

// 生成字节数组
byte[] code = cw.toByteArray();

// 加载类
ClassLoader cl = new ClassLoader() {
    @Override
    protected Class<?> findClass(String name) throws ClassNotFoundException {
        if ("com.example.TestClass".equals(name)) {
            return defineClass(name, code, 0, code.length);
        }
        return super.findClass(name);
    }
};
Class<?> clazz = cl.loadClass("com.example.TestClass");

// 创建对象并调用方法
Object obj = clazz.newInstance();
Method method = clazz.getDeclaredMethod("testMethod", String.class);
method.invoke(obj, "hello");

以上是两种实现动态注入代码的方式，具体选择哪种方式取决于应用场景和需求。需要注意的是，动态注入代码有可能会影响系统的安全和稳定性，因此应该谨慎使用。