Быстрое преобразование только одного класса с помощью агента Java

Поскольку вы находитесь в premain, вы, вероятно, измеряете время загрузки и инициализации многих классов, которые раньше не использовались. Вполне возможно, что значительное количество этих классов в любом случае будет загружено и инициализировано позже, когда приложение использует их в первый раз, без измерения как «время запуска», поэтому это время будет перемещено в измеренное время запуска. время не может быть актуальной проблемой.

Обратите внимание, что вы используете лямбда-выражения в обоих вариантах, что вызывает инициализацию их серверной части, предоставляемой JRE. В случае OpenJDK он использует ASM под капотом, но, поскольку он был переупакован, чтобы избежать конфликтов с приложениями, использующими ASM, это не те классы, которые Byte-Buddy использует внутри, поэтому вы платите цену за инициализацию ASM дважды здесь .

Как было сказано, если эти классы будут использоваться в любом случае, т.е. если приложение будет использовать лямбда-выражения или ссылки на методы позже, вам не следует беспокоиться об этом, так как «оптимизация» только сдвинет инициализацию на более позднее время. Но если приложение не использует лямбда-выражения или ссылки на методы или вы хотите любой ценой удалить этот промежуток времени из измеренного времени запуска, вы можете прибегнуть к обычной реализации интерфейса, используя внутренний класс или позволив Agent реализовать интерфейс.

Чтобы еще больше сократить время запуска, вы можете использовать ASM напрямую, пропустив инициализацию классов Byte-Buddy, например.

import java.lang.instrument.*;
import java.lang.management.ManagementFactory;
import java.security.ProtectionDomain;
import org.objectweb.asm.*;
import org.objectweb.asm.commons.AdviceAdapter;

public class Agent extends ClassVisitor implements ClassFileTransformer {
    private static final String NETTY_CLASS = "io/netty/bootstrap/AbstractBootstrap";

    public static void premain(String arg, Instrumentation instrumentation) {
        instrumentation.addTransformer(new Agent());
    }

    public Agent() {
        super(Opcodes.ASM5);
    }
    public byte[] transform(ClassLoader loader, String className, Class<?> cl,
                            ProtectionDomain pd, byte[] classfileBuffer) {
        if(!NETTY_CLASS.equals(className)) return null;

        ClassReader cr = new ClassReader(classfileBuffer);
        ClassWriter cw = new ClassWriter(cr, 0);
        synchronized(this) {
            super.cv = cw;
            try { cr.accept(this, 0); }
            finally { super.cv = null; }
        }
        return cw.toByteArray();
    }

    @Override
    public MethodVisitor visitMethod(
        int access, String name, String desc, String signature, String[] exceptions) {

        MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions);
        if(name.equals("bind")
        && desc.equals("(Ljava/net/SocketAddress;)Lio/netty/channel/ChannelFuture;")) {
            return new AdviceAdapter(Opcodes.ASM5, mv, access, name, desc) {
                @Override
                protected void onMethodExit(int opcode) {
                    super.visitMethodInsn(Opcodes.INVOKESTATIC,
                        Agent.class.getName().replace('.', '/'),
                        "injectedMethod", "()V", false);
                    super.onMethodExit(opcode);
                }
            };
        }
        return mv;
    }
    public static void injectedMethod() {
        System.out.printf("Server started. Current Time (ms): %d",
                          System.currentTimeMillis());
        System.out.printf("Server started. Current Uptime (ms): %d",
                          ManagementFactory.getRuntimeMXBean().getUptime());
    }
}

(не испытано)

Очевидно, что этот код сложнее, чем код, использующий Byte-Buddy, поэтому вам придется решить, на какой компромисс пойти.

ASM уже очень легкий. Если пойти еще глубже, это будет означать выполнение преобразования файла класса с использованием только ByteBuffer и HashMap; это возможно, но, конечно, не тот путь, по которому вы хотите пойти…