シンプルな Java プロファイラーを作成し、javaagent を使用してメモリ使用量を監視しようとしています。これは私のjavaagentの実現の一部です:
import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException;
import java.lang.instrument.Instrumentation;
import java.rmi.RemoteException;
import java.rmi.registry.LocateRegistry;
import java.rmi.registry.Registry;
import java.security.ProtectionDomain;
import java.util.ArrayList;
import java.util.Timer;
import java.lang.management.ManagementFactory;
import java.lang.management.MemoryPoolMXBean;
import java.lang.management.MemoryUsage;
import java.util.Iterator;
import java.util.List;
import java.util.TimerTask;
public class Transformer implements ClassFileTransformer {
private Notifier notifier;
private Instrumentation inst;
private ArrayList<Class> loadedClasses;
public Transformer(Instrumentation inst) {
this.inst = inst;
loadedClasses = new ArrayList<Class>();
initNotifier();
Timer time = new Timer();
MemoryMonitor mm = new MemoryMonitor(notifier);
time.schedule(mm, 0, 1000);
}
public static void premain(String args, Instrumentation inst) {
inst.addTransformer(new Transformer(inst));
}
private void initNotifier() {
if (notifier != null) return;
try {
Registry registry = LocateRegistry.getRegistry(Const.registryPort);
notifier = (Notifier) registry.lookup(Const.stubName);
} catch (Exception e) {
e.printStackTrace();
System.exit(1);
}
}
}
public class MemoryMonitor extends TimerTask {
private Notifier notifier;
public MemoryMonitor(Notifier notifier) {
this.notifier = notifier;
}
@Override
public void run() {
try {
notifier.memoryUsed(monitorMemory());
} catch (Exception e) {
e.printStackTrace();
}
}
private double monitorMemory() {
List memBeans = ManagementFactory.getMemoryPoolMXBeans();
double used = 0;
for (Iterator i = memBeans.iterator(); i.hasNext(); ) {
MemoryPoolMXBean mpool = (MemoryPoolMXBean) i.next();
MemoryUsage usage = mpool.getUsage();
used = usage.getUsed() / 1000;
}
return used;
}
}
このコードは適切に機能し、メモリ使用量に関するデータをプロファイリング ツールに送信します。ただし、アプリケーションの実行が終了しても javaagent は停止しません。
アプリケーションの実行終了後に javaagent を停止する方法を知っている人はいますか?