java - 尝试在Web应用程序中使用代理进行字节码操作

Question

我不太擅长 Java，但我的 web 应用程序在 Wildfly 上运行。 我有 3 个线程，它们只调用一个插入日志的函数，该函数将日志保存到数据库中，之后每个线程都会发送一个时间来完成此操作。 它们将数据发送到我编写的另一个程序，该程序有 3 个线程来调用 3 个服务器线程之一。

所以现在我尝试做一些字节码操作，服务器上的每个线程都会保存日期时间调用，日志函数等待 1 秒，然后返回他们需要的时间。

1 线程在等待 1 秒之前或之后在日志文件中写入内容。 但是他们等待一秒钟并调用日志函数的这一部分我希望通过字节码操作将其注入(inject)到每 3 个线程中。

public class MyTransformer implements ClassFileTransformer {


@Override
public byte[] transform(ClassLoader loader, String className, Class redefiningClass, ProtectionDomain protectionDomain, byte[] bytes) throws IllegalClassFormatException {

    return transformClass(redefiningClass, bytes);
}

private byte[] transformClass(Class classToTransform, byte[] b) {
    ClassPool pool = ClassPool.getDefault();
    CtClass cl = null;
    try {
        cl = pool.get("de.soptim.ws.MyApplication");
    } catch (javassist.NotFoundException e) {
        e.printStackTrace();
    }
    try {

        assert cl != null;
        CtMethod[] methods = cl.getMethods();
        for (int i = 0; i < methods.length; i++) {
            if (methods[i].isEmpty() == false) {
                changeMethod(methods[i]);
            }
        }
        b = cl.toBytecode();
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        if (cl != null) {
            cl.detach();
        }
    }
    return b;
}

private void changeMethod(CtMethod method) throws NotFoundException, CannotCompileException {

    if (method.hasAnnotation(Loggable.class)) {
        method.insertBefore("threadLogger.info(\"ADDED THIS FROM BYTECODE !!!\");");
        method.insertAfter("threadLogger.info(\"ADDED THIS FROM BYTECODE !!!\");");
    }
}}

这是我的变压器类，它应该增加我的方法需要的代码，它检查什么方法有 @Loggable 注释，然后将代码添加到其中(“目前它只是一些用于检查它是否有效的日志语句”)

我现在最大的问题是我不知道如何调用我的代理...我在谷歌上搜索了如何在运行时使用 agentmain() 调用代理，但我想我并没有真正理解它是如何工作的。

代理类

public class LogAgent  {

public static void agentmain(String agentArgs, Instrumentation inst) {
    System.out.println("Starting the agent");
    inst.addTransformer(new MyTransformer());
}}

希望您理解我的问题:)如果您回答，请尽量保持菜鸟友好:D。

Answer 1

您没有显式调用代理，您应该为 JVM 指定其他参数:

java -javaagent:jarpath[=选项]

其中 jarpath 是包含代理的 jar 的路径。 JVM会在java程序的main方法之前调用premain方法。

并且 transform 方法将在 JVM 类加载之前调用(您无需显式调用它)。

最后一句话:您应该实现 premain 方法，而不是 agentmain。 agentmain 在附加到正在运行的 vm 期间使用，而 premain 在使用 -javaagent 方法启动 JVM 时使用。

并确保您的 jar 具有有效的 list ，如下所述:https://docs.oracle.com/javase/8/docs/api/java/lang/instrument/package-summary.html

我没有使用过 javaassit，所以我不能说你的代码是有效的，但是像 Wildfly 这样的 web 应用程序服务器比普通的 java 应用程序要困难得多(主要是由于类加载器的可见性和层次结构)。

另请参阅:

http://www.tomsquest.com/blog/2014/01/intro-java-agent-and-bytecode-manipulation/ Tutorials about javaagents