// ==============================
// 文件：MemoryAgent.java
// 包名：com.debugger.memory
// 功能：Java Agent 入口，注冊字節(jié)碼增強
// ==============================
package com.debugger.memory;
 
import java.lang.instrument.Instrumentation;
 
public class MemoryAgent {
    public static void premain(String agentArgs, Instrumentation inst) {
        System.out.println("[MemoryDebugger] Agent start with args: " + agentArgs);
        // 注冊字節(jié)碼轉(zhuǎn)換器
        inst.addTransformer(new MemoryClassFileTransformer());
        // 初始化 GC 監(jiān)聽
        MemoryDebugger.initGCListener();
        // 初始化數(shù)據(jù)緩沖區(qū)和快照 MBean
        MemoryDebugger.initDataCollector();
    }
}
 
// ==============================
// 文件：MemoryClassFileTransformer.java
// 包名：com.debugger.memory
// 功能：對類字節(jié)碼進行增強，攔截對象分配
// ==============================
package com.debugger.memory;
 
import java.lang.instrument.ClassFileTransformer;
import java.security.ProtectionDomain;
import org.objectweb.asm.*;
 
public class MemoryClassFileTransformer implements ClassFileTransformer {
    @Override
    public byte[] transform(
        ClassLoader loader, String className, Class<?> classBeingRedefined,
        ProtectionDomain pd, byte[] classfileBuffer) {
        // 僅增強業(yè)務(wù)包下的類，排除 JVM、第三方庫
        if (className == null || !className.startsWith("com/myapp")) {
            return classfileBuffer;
        }
        ClassReader cr = new ClassReader(classfileBuffer);
        ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_MAXS);
        ClassVisitor cv = new ClassVisitor(Opcodes.ASM9, cw) {
            @Override
            public MethodVisitor visitMethod(
                int access, String name, String desc, String signature, String[] exceptions) {
                MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions);
                return new MemoryMethodVisitor(Opcodes.ASM9, mv);
            }
        };
        cr.accept(cv, 0);
        return cw.toByteArray();
    }
}
 
// ==============================
// 文件：MemoryMethodVisitor.java
// 包名：com.debugger.memory
// 功能：攔截 NEW 指令，插入監(jiān)控調(diào)用
// ==============================
package com.debugger.memory;
 
import org.objectweb.asm.*;
 
public class MemoryMethodVisitor extends MethodVisitor {
    public MemoryMethodVisitor(int api, MethodVisitor mv) {
        super(api, mv);
    }
 
    @Override
    public void visitTypeInsn(int opcode, String type) {
        // 對 NEW 操作插樁
        if (opcode == Opcodes.NEW) {
            mv.visitInsn(Opcodes.DUP);
            mv.visitTypeInsn(Opcodes.NEW, type);
            mv.visitInsn(Opcodes.DUP);
            // 調(diào)用監(jiān)控方法
            mv.visitMethodInsn(
                Opcodes.INVOKESTATIC,
                "com/debugger/memory/MemoryDebugger",
                "onObjectAllocate",
                "(Ljava/lang/Object;)V",
                false
            );
        } else {
            super.visitTypeInsn(opcode, type);
        }
    }
}
 
// ==============================
// 文件：MemoryDebugger.java
// 包名：com.debugger.memory
// 功能：核心監(jiān)控邏輯，包括分配記錄與 GC 回調(diào)
// ==============================
package com.debugger.memory;
 
import java.lang.management.*;
import java.util.*;
import javax.management.*;
import com.sun.management.GarbageCollectionNotificationInfo;
import java.util.concurrent.*;
 
public class MemoryDebugger {
    // 環(huán)形緩沖區(qū)存儲分配記錄
    private static final int BUFFER_SIZE = 1 << 20;
    private static final RingBuffer<Record> buffer = new RingBuffer<>(BUFFER_SIZE);
 
    public static void initGCListener() {
        // 注冊 GC 通知監(jiān)聽
        for (GarbageCollectorMXBean gcBean : ManagementFactory.getGarbageCollectorMXBeans()) {
            if (gcBean instanceof NotificationEmitter) {
                NotificationEmitter emitter = (NotificationEmitter) gcBean;
                emitter.addNotificationListener((notif, handback) -> {
                    // 處理 GC 開始/結(jié)束事件
                    MemoryDebugger.onGCEvent(notif);
                }, null, null);
            }
        }
    }
 
    public static void initDataCollector() {
        // 注冊 JMX MBean，用于觸發(fā)快照
        try {
            MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
            ObjectName name = new ObjectName("com.debugger.memory:type=DataCollector");
            mbs.registerMBean(new DataCollector(), name);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
 
    public static void onObjectAllocate(Object obj) {
        // 采樣：僅記錄指定包或大對象
        String className = obj.getClass().getName();
        long size = getObjectSize(obj);
        Record r = new Record(
            Thread.currentThread().getName(),
            className,
            System.nanoTime(),
            size,
            getStackTrace(5)
        );
        buffer.put(r);
    }
 
    private static long getObjectSize(Object obj) {
        // 通過 Instrumentation 獲取對象大小（需在 Agent 中傳入）
        return InstrumentationHolder.getInstrumentation().getObjectSize(obj);
    }
 
    private static String[] getStackTrace(int depth) {
        StackTraceElement[] st = Thread.currentThread().getStackTrace();
        String[] arr = new String[Math.min(depth, st.length - 2)];
        for (int i = 2; i < arr.length + 2; i++) {
            arr[i - 2] = st[i].toString();
        }
        return arr;
    }
 
    public static void onGCEvent(javax.management.Notification notif) {
        // GC 事件解析
        CompositeData cd = (CompositeData) notif.getUserData();
        GarbageCollectionNotificationInfo info = GarbageCollectionNotificationInfo.from(cd);
        // 解析 GC 信息并輸出報告或關(guān)聯(lián)分配記錄
        GCRecord rec = new GCRecord(info.getGcAction(), info.getGcName(),
            info.getGcInfo().getId(), info.getGcInfo().getDuration(),
            info.getGcInfo().getMemoryUsageBeforeGc(),
            info.getGcInfo().getMemoryUsageAfterGc(),
            System.nanoTime());
        buffer.put(rec);
    }
 
    // 省略其他輔助類與方法，如 Record、GCRecord、RingBuffer、DataCollector MBean
}
 
// ==============================
// 文件：DataCollector.java
// 包名：com.debugger.memory
// 功能：JMX MBean，用于觸發(fā)內(nèi)存快照
// ==============================
package com.debugger.memory;
 
public class DataCollector implements DataCollectorMBean {
    @Override
    public void takeSnapshot() {
        MemoryReport.generateReport();  // 導(dǎo)出 JSON/HTML 報告
    }
}
 
// ==============================
// 文件：MemoryReport.java
// 包名：com.debugger.memory
// 功能：生成內(nèi)存調(diào)試報告
// ==============================
package com.debugger.memory;
 
import java.io.*;
import java.util.*;
import freemarker.template.*;
 
public class MemoryReport {
    public static void generateReport() {
        List<Record> records = RingBuffer.drainAll();
        // 用 FreeMarker 渲染模板，生成 HTML 報告
        // 模板包含 TopN 分配類、存活時間分布直方圖、GC 關(guān)聯(lián)分析
        // 省略具體實現(xiàn)
    }
}