Files
pinpoint/src/main/java/com/profiler/TomcatProfiler.java
T

121 lines
4.1 KiB
Java

package com.profiler;
import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.Instrumentation;
import java.util.Properties;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;
import com.profiler.common.ServiceType;
import com.profiler.config.ProfilerConfig;
import com.profiler.logging.LoggerBinder;
import com.profiler.logging.LoggerFactory;
import com.profiler.interceptor.bci.ByteCodeInstrumentor;
import com.profiler.interceptor.bci.JavaAssistByteCodeInstrumentor;
import com.profiler.modifier.ModifierRegistry;
public class TomcatProfiler {
private static final Logger logger = Logger.getLogger(TomcatProfiler.class.getName());
private boolean isFine = logger.isLoggable(Level.FINE);
private String agentArgString = "";
private Instrumentation instrumentation;
private final ByteCodeInstrumentor byteCodeInstrumentor;
private final ModifierRegistry modifierRepository;
private final ProfilerConfig profilerConfig;
private final Agent agent;
public void boot(Instrumentation instrumentation, String agentArgs) {
if (agentArgs != null) {
logger.info("HIPPO agentArgs:" + agentArgs);
}
try {
LoggerBinder loggerBinder = null;
loggerBinder.getLogger("LoggerFactory initialize start");
LoggerFactory.initialize(loggerBinder);
com.profiler.logging.Logger tomcatLogger = LoggerFactory.getLogger(TomcatProfiler.class);
tomcatLogger.info("LoggerFactory initialize end");
} catch (Exception e) {
logger.log(Level.INFO, "boot class not found", e);
}
try {
ProfilerConfig profilerConfig = null;
if (!profilerConfig.isProfileEnable()) {
logger.warning("Profiler Agent not started. profile.enable=" + profilerConfig.isProfileEnable());
return;
}
Agent agent = new DefaultAgent(agentArgs, instrumentation, profilerConfig);
// new TomcatProfiler(agentArgs, instrumentation, agent, profilerConfig);
} catch (Exception e) {
logger.log(Level.SEVERE, "Profiler Agent start fail. Cause:" + e.getMessage(), e);
}
}
private static void dumpSystemProperties() {
if (logger.isLoggable(Level.FINE)) {
Properties properties = System.getProperties();
Set<String> strings = properties.stringPropertyNames();
for (String key : strings) {
logger.fine("SystemProperties " + key + "=" + properties.get(key));
}
}
}
public TomcatProfiler(String agentArgs, Instrumentation instrumentation, DefaultAgent agent, ProfilerConfig profilerConfig) {
this.agentArgString = agentArgs;
this.profilerConfig = profilerConfig;
this.agent = agent;
String[] paths = getTomcatlibPath();
this.byteCodeInstrumentor = new JavaAssistByteCodeInstrumentor(paths, agent);
this.modifierRepository = null;
ClassFileTransformer classFileTransformerDispatcher = new ClassFileTransformerDispatcher(agent);
this.instrumentation.addTransformer(classFileTransformerDispatcher);
}
public static String findHippoConfigFormSystemProeprty() {
String hippoConfigFileName = System.getProperty("hippo.config");
if (hippoConfigFileName == null) {
return null;
}
logger.info("hippo.config property found. " + hippoConfigFileName);
return hippoConfigFileName;
}
private String[] getTomcatlibPath() {
String catalinaHome = System.getProperty("catalina.home");
if (catalinaHome == null) {
logger.info("CATALINA_HOME is null");
return null;
}
if (logger.isLoggable(Level.INFO)) {
logger.info("CATALINA_HOME=" + catalinaHome);
}
if (profilerConfig.getServiceType() == ServiceType.BLOC) {
return new String[] { catalinaHome + "/server/lib/catalina.jar", catalinaHome + "/common/lib/servlet-api.jar" };
} else {
return new String[] { catalinaHome + "/lib/servlet-api.jar", catalinaHome + "/lib/catalina.jar" };
}
}
}