org.glowroot.agent.init.AgentModule Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of glowroot-agent-it-harness Show documentation
Show all versions of glowroot-agent-it-harness Show documentation
Glowroot Agent Integration Test Harness
/*
* Copyright 2011-2018 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.glowroot.agent.init;
import java.io.File;
import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.Instrumentation;
import java.lang.instrument.UnmodifiableClassException;
import java.lang.management.ManagementFactory;
import java.util.List;
import java.util.Map;
import java.util.Random;
import java.util.concurrent.ScheduledExecutorService;
import java.util.jar.JarFile;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.base.Joiner;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.base.Ticker;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.collect.ImmutableList;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.collect.Lists;
import org.glowroot.agent.shaded.org.checkerframework.checker.nullness.qual.MonotonicNonNull;
import org.glowroot.agent.shaded.org.checkerframework.checker.nullness.qual.Nullable;
import org.glowroot.agent.shaded.org.glowroot.agent.shaded.org.slf4j.Logger;
import org.glowroot.agent.shaded.org.glowroot.agent.shaded.org.slf4j.LoggerFactory;
import org.glowroot.agent.api.internal.GlowrootService;
import org.glowroot.agent.api.internal.GlowrootServiceHolder;
import org.glowroot.agent.bytecode.api.BytecodeServiceHolder;
import org.glowroot.agent.collector.Collector;
import org.glowroot.agent.config.ConfigService;
import org.glowroot.agent.config.PluginCache;
import org.glowroot.agent.config.PluginDescriptor;
import org.glowroot.agent.impl.Aggregator;
import org.glowroot.agent.impl.BytecodeServiceImpl;
import org.glowroot.agent.impl.BytecodeServiceImpl.OnEnteringMain;
import org.glowroot.agent.impl.ConfigServiceImpl;
import org.glowroot.agent.impl.GlowrootServiceImpl;
import org.glowroot.agent.impl.PluginServiceImpl;
import org.glowroot.agent.impl.PluginServiceImpl.ConfigServiceFactory;
import org.glowroot.agent.impl.StackTraceCollector;
import org.glowroot.agent.impl.TimerNameCache;
import org.glowroot.agent.impl.TransactionCollector;
import org.glowroot.agent.impl.TransactionRegistry;
import org.glowroot.agent.impl.TransactionService;
import org.glowroot.agent.impl.UserProfileScheduler;
import org.glowroot.agent.init.PreCheckLoadedClasses.PreCheckClassFileTransformer;
import org.glowroot.agent.live.LiveAggregateRepositoryImpl;
import org.glowroot.agent.live.LiveJvmServiceImpl;
import org.glowroot.agent.live.LiveTraceRepositoryImpl;
import org.glowroot.agent.live.LiveWeavingServiceImpl;
import org.glowroot.agent.plugin.api.internal.PluginService;
import org.glowroot.agent.plugin.api.internal.PluginServiceHolder;
import org.glowroot.agent.util.JavaVersion;
import org.glowroot.agent.util.LazyPlatformMBeanServer;
import org.glowroot.agent.util.OptionalService;
import org.glowroot.agent.util.ThreadAllocatedBytes;
import org.glowroot.agent.util.Tickers;
import org.glowroot.agent.weaving.AdviceCache;
import org.glowroot.agent.weaving.AnalyzedWorld;
import org.glowroot.agent.weaving.IsolatedWeavingClassLoader;
import org.glowroot.agent.weaving.PointcutClassFileTransformer;
import org.glowroot.agent.weaving.PreInitializeWeavingClasses;
import org.glowroot.agent.weaving.Weaver;
import org.glowroot.agent.weaving.WeavingClassFileTransformer;
import org.glowroot.agent.shaded.org.glowroot.common.util.Clock;
import org.glowroot.agent.shaded.org.glowroot.common.util.OnlyUsedByTests;
import org.glowroot.agent.shaded.org.glowroot.common.util.ScheduledRunnable;
import static org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.base.Preconditions.checkNotNull;
import static java.util.concurrent.TimeUnit.MILLISECONDS;
import static java.util.concurrent.TimeUnit.MINUTES;
import static java.util.concurrent.TimeUnit.SECONDS;
public class AgentModule {
private static final Logger logger = LoggerFactory.getLogger(AgentModule.class);
// log startup messages using logger name "org.glowroot"
private static final Logger startupLogger = LoggerFactory.getLogger("org.glowroot");
// 1 minute
private static final long ROLLUP_0_INTERVAL_MILLIS =
Long.getLong("glowroot.internal.rollup.0.intervalMillis", MINUTES.toMillis(1));
private final Clock clock;
private final Ticker ticker;
private final ConfigService configService;
private final TransactionRegistry transactionRegistry;
private final AdviceCache adviceCache;
private final AnalyzedWorld analyzedWorld;
private final Weaver weaver;
private final Random random;
private final UserProfileScheduler userProfileScheduler;
private final TransactionService transactionService;
private final BytecodeServiceImpl bytecodeService;
private volatile @MonotonicNonNull DeadlockedActiveWeavingRunnable deadlockedActiveWeavingRunnable;
private volatile @MonotonicNonNull Aggregator aggregator;
private volatile @MonotonicNonNull TransactionCollector transactionCollector;
private volatile @MonotonicNonNull LazyPlatformMBeanServer lazyPlatformMBeanServer;
private volatile @MonotonicNonNull GaugeCollector gaugeCollector;
private volatile @MonotonicNonNull StackTraceCollector stackTraceCollector;
private volatile @MonotonicNonNull ImmediateTraceStoreWatcher immedateTraceStoreWatcher;
private final boolean jvmRetransformClassesSupported;
private volatile @MonotonicNonNull LiveTraceRepositoryImpl liveTraceRepository;
private volatile @MonotonicNonNull LiveAggregateRepositoryImpl liveAggregateRepository;
private volatile @MonotonicNonNull LiveWeavingServiceImpl liveWeavingService;
private volatile @MonotonicNonNull LiveJvmServiceImpl liveJvmService;
// accepts @Nullable Ticker to deal with shading issues when called from GlowrootModule
public AgentModule(Clock clock, @Nullable Ticker nullableTicker, final PluginCache pluginCache,
final ConfigService configService, @Nullable Instrumentation instrumentation,
@Nullable File glowrootJarFile, File tmpDir,
@Nullable PreCheckClassFileTransformer preCheckClassFileTransformer) throws Exception {
this.clock = clock;
this.ticker = nullableTicker == null ? Tickers.getTicker() : nullableTicker;
this.configService = configService;
transactionRegistry = new TransactionRegistry();
ClassFileTransformer pointcutClassFileTransformer = null;
if (instrumentation != null) {
for (File pluginJar : pluginCache.pluginJars()) {
instrumentation.appendToBootstrapClassLoaderSearch(new JarFile(pluginJar));
}
pointcutClassFileTransformer = new PointcutClassFileTransformer();
instrumentation.addTransformer(pointcutClassFileTransformer);
}
adviceCache = new AdviceCache(pluginCache.pluginDescriptors(),
configService.getInstrumentationConfigs(), instrumentation, tmpDir);
if (pointcutClassFileTransformer != null) {
checkNotNull(instrumentation).removeTransformer(pointcutClassFileTransformer);
}
analyzedWorld = new AnalyzedWorld(adviceCache.getAdvisorsSupplier(),
adviceCache.getShimTypes(), adviceCache.getMixinTypes());
TimerNameCache timerNameCache = new TimerNameCache();
weaver = new Weaver(adviceCache.getAdvisorsSupplier(), adviceCache.getShimTypes(),
adviceCache.getMixinTypes(), analyzedWorld, transactionRegistry, ticker,
timerNameCache, configService);
// need to initialize glowroot-agent-api, glowroot-agent-plugin-api and glowroot-weaving-api
// services before enabling instrumentation
GlowrootService glowrootService = new GlowrootServiceImpl(transactionRegistry);
GlowrootServiceHolder.set(glowrootService);
ConfigServiceFactory configServiceFactory = new ConfigServiceFactory() {
@Override
public org.glowroot.agent.plugin.api.config.ConfigService create(String pluginId) {
return ConfigServiceImpl.create(configService, pluginCache.pluginDescriptors(),
pluginId);
}
};
PluginService pluginService = new PluginServiceImpl(timerNameCache, configServiceFactory);
PluginServiceHolder.set(pluginService);
random = new Random();
userProfileScheduler = new UserProfileScheduler(configService, random);
transactionService = TransactionService.create(transactionRegistry, configService,
timerNameCache, userProfileScheduler, ticker, clock);
bytecodeService = new BytecodeServiceImpl(transactionRegistry, transactionService);
BytecodeServiceHolder.set(bytecodeService);
if (instrumentation == null) {
// instrumentation is null when debugging with LocalContainer
IsolatedWeavingClassLoader isolatedWeavingClassLoader =
(IsolatedWeavingClassLoader) Thread.currentThread().getContextClassLoader();
checkNotNull(isolatedWeavingClassLoader);
isolatedWeavingClassLoader.setWeaver(weaver);
jvmRetransformClassesSupported = false;
} else {
PreInitializeWeavingClasses.preInitializeClasses();
ClassFileTransformer transformer =
new WeavingClassFileTransformer(weaver, instrumentation);
if (instrumentation.isRetransformClassesSupported()) {
instrumentation.addTransformer(transformer, true);
jvmRetransformClassesSupported = true;
} else {
instrumentation.addTransformer(transformer);
jvmRetransformClassesSupported = false;
}
if (preCheckClassFileTransformer != null) {
for (Map.Entry entry : preCheckClassFileTransformer
.getImportantClassLoadingPoints().entrySet()) {
logger.warn("important class loaded before Glowroot instrumentation could be"
+ " applied to it: {}", entry.getKey(), entry.getValue());
}
instrumentation.removeTransformer(preCheckClassFileTransformer);
}
logJavaClassAlreadyLoadedWarningIfNeeded(instrumentation.getAllLoadedClasses(),
glowrootJarFile, false);
if (instrumentation.isRetransformClassesSupported()
&& instrumentation.isModifiableClass(Thread.class)) {
try {
instrumentation.retransformClasses(Thread.class);
} catch (UnmodifiableClassException e) {
// IBM JDK 6 throws UnmodifiableClassException even though isModifiableClass()
// above returns true
logger.debug(e.getMessage(), e);
}
}
}
ManagementFactory.getThreadMXBean().setThreadCpuTimeEnabled(true);
ManagementFactory.getThreadMXBean().setThreadContentionMonitoringEnabled(true);
// need to initialize some classes while still single threaded in order to prevent possible
// deadlock later on
try {
Class.forName("sun.net.www.protocol.ftp.Handler");
Class.forName("sun.net.www.protocol.ftp.FtpURLConnection");
} catch (ClassNotFoundException e) {
logger.debug(e.getMessage(), e);
}
// verify initialization of glowroot-agent-api, glowroot-agent-plugin-api and
// glowroot-weaving-api services
Exception getterCalledTooEarlyLocation =
GlowrootServiceHolder.getRetrievedTooEarlyLocation();
if (getterCalledTooEarlyLocation != null) {
logger.error("Glowroot Agent API was called too early", getterCalledTooEarlyLocation);
}
initPlugins(pluginCache.pluginDescriptors());
List pluginDescriptors = pluginCache.pluginDescriptors();
List pluginNames = Lists.newArrayList();
for (PluginDescriptor pluginDescriptor : pluginDescriptors) {
pluginNames.add(pluginDescriptor.name());
}
if (!pluginNames.isEmpty()) {
startupLogger.info("plugins loaded: {}", Joiner.on(", ").join(pluginNames));
}
}
public void setOnEnteringMain(OnEnteringMain onEnteringMain) {
bytecodeService.setOnEnteringMain(onEnteringMain);
}
public void onEnteringMain(ScheduledExecutorService backgroundExecutor, Collector collector,
@Nullable Instrumentation instrumentation, @Nullable File glowrootJarFile,
@Nullable String mainClass) throws Exception {
weaver.setNoLongerNeedToWeaveMainMethods();
deadlockedActiveWeavingRunnable = new DeadlockedActiveWeavingRunnable(weaver);
deadlockedActiveWeavingRunnable.scheduleWithFixedDelay(backgroundExecutor, 5, 5, SECONDS);
// complete initialization of glowroot-agent-api, glowroot-agent-plugin-api and
// glowroot-weaving-api services
userProfileScheduler.setBackgroundExecutor(backgroundExecutor);
OptionalService threadAllocatedBytes = ThreadAllocatedBytes.create();
transactionService.setThreadAllocatedBytes(threadAllocatedBytes.getService());
aggregator = new Aggregator(collector, configService, ROLLUP_0_INTERVAL_MILLIS, clock);
transactionCollector =
new TransactionCollector(configService, collector, aggregator, clock, ticker);
transactionService.setTransactionCollector(transactionCollector);
lazyPlatformMBeanServer = LazyPlatformMBeanServer.create(mainClass);
bytecodeService.setOnExitingGetPlatformMBeanServer(new Runnable() {
@Override
public void run() {
// TODO report checker framework issue that occurs without checkNotNull
checkNotNull(lazyPlatformMBeanServer);
lazyPlatformMBeanServer.setPlatformMBeanServerAvailable();
}
});
File[] roots = File.listRoots();
if (roots != null) {
for (File root : roots) {
String name = root.getCanonicalPath();
if (name.length() > 1 && (name.endsWith("/") || name.endsWith("\\"))) {
name = name.substring(0, name.length() - 1);
}
name = name.replaceAll(":", "");
lazyPlatformMBeanServer.lazyRegisterMBean(new FileSystem(root),
"org.glowroot:type=FileSystem,name=" + name);
}
}
gaugeCollector = new GaugeCollector(configService, collector, lazyPlatformMBeanServer,
instrumentation, clock, ticker);
// using fixed rate to keep gauge collections close to on the second mark
long gaugeCollectionIntervalMillis = configService.getGaugeCollectionIntervalMillis();
gaugeCollector.scheduleWithFixedDelay(gaugeCollectionIntervalMillis, MILLISECONDS);
stackTraceCollector = new StackTraceCollector(transactionRegistry, configService, random);
immedateTraceStoreWatcher = new ImmediateTraceStoreWatcher(backgroundExecutor,
transactionRegistry, transactionCollector, configService, ticker);
immedateTraceStoreWatcher.scheduleWithFixedDelay(backgroundExecutor,
ImmediateTraceStoreWatcher.PERIOD_MILLIS, MILLISECONDS);
liveTraceRepository = new LiveTraceRepositoryImpl(transactionRegistry, transactionCollector,
clock, ticker);
liveAggregateRepository = new LiveAggregateRepositoryImpl(aggregator);
liveWeavingService = new LiveWeavingServiceImpl(analyzedWorld, instrumentation,
configService, adviceCache, jvmRetransformClassesSupported);
liveJvmService = new LiveJvmServiceImpl(lazyPlatformMBeanServer, transactionRegistry,
transactionCollector, threadAllocatedBytes.getAvailability(), configService,
glowrootJarFile);
}
public ConfigService getConfigService() {
return configService;
}
public LazyPlatformMBeanServer getLazyPlatformMBeanServer() {
if (lazyPlatformMBeanServer == null) {
throw new IllegalStateException("onEnteringMain() was never called");
}
return lazyPlatformMBeanServer;
}
public LiveTraceRepositoryImpl getLiveTraceRepository() {
if (liveTraceRepository == null) {
throw new IllegalStateException("onEnteringMain() was never called");
}
return liveTraceRepository;
}
public LiveAggregateRepositoryImpl getLiveAggregateRepository() {
if (liveAggregateRepository == null) {
throw new IllegalStateException("onEnteringMain() was never called");
}
return liveAggregateRepository;
}
public LiveWeavingServiceImpl getLiveWeavingService() {
if (liveWeavingService == null) {
throw new IllegalStateException("onEnteringMain() was never called");
}
return liveWeavingService;
}
public LiveJvmServiceImpl getLiveJvmService() {
if (liveJvmService == null) {
throw new IllegalStateException("onEnteringMain() was never called");
}
return liveJvmService;
}
public static boolean logJavaClassAlreadyLoadedWarningIfNeeded(Class[] allLoadedClasses,
@Nullable File glowrootJarFile, boolean preCheck) {
List loadedImportantClassNames = Lists.newArrayList();
for (Class clazz : allLoadedClasses) {
String className = clazz.getName();
if (PreCheckLoadedClasses.isImportantClass(className, clazz)) {
loadedImportantClassNames.add(className);
}
}
if (loadedImportantClassNames.isEmpty()) {
return false;
} else {
logLoadedImportantClassWarning(loadedImportantClassNames, glowrootJarFile, preCheck);
return true;
}
}
private static void logLoadedImportantClassWarning(List loadedImportantClassNames,
@Nullable File glowrootJarFile, boolean preCheck) {
if (preCheck) {
// this is only logged with -Dglowroot.debug.preCheckLoadedClasses=true
startupLogger.warn("PRE-CHECK: one or more important classes were loaded before"
+ " Glowroot startup: {}", Joiner.on(", ").join(loadedImportantClassNames));
return;
}
List javaAgentArgsBeforeGlowroot = getJavaAgentArgsBeforeGlowroot(glowrootJarFile);
if (!javaAgentArgsBeforeGlowroot.isEmpty()) {
startupLogger.warn("one or more important classes were loaded before Glowroot"
+ " instrumentation could be applied to them: {}. This likely occurred because"
+ " one or more other javaagents are listed in the JVM args prior to the"
+ " Glowroot agent ({}) which gives them a higher loading precedence.",
Joiner.on(", ").join(loadedImportantClassNames),
Joiner.on(" ").join(javaAgentArgsBeforeGlowroot));
return;
}
List nativeAgentArgs = getNativeAgentArgs();
if (!nativeAgentArgs.isEmpty()) {
startupLogger.warn("one or more important classes were loaded before Glowroot"
+ " instrumentation could be applied to them: {}. This likely occurred because"
+ " there are one or more native agents listed in the JVM args ({}), and native"
+ " agents have higher loading precedence than java agents.",
Joiner.on(", ").join(loadedImportantClassNames),
Joiner.on(" ").join(nativeAgentArgs));
return;
}
startupLogger.warn("one or more important classes were loaded before Glowroot"
+ " instrumentation could be applied to them: {}",
Joiner.on(", ").join(loadedImportantClassNames));
}
private static List getNativeAgentArgs() {
List nativeAgentArgs = Lists.newArrayList();
for (String jvmArg : ManagementFactory.getRuntimeMXBean().getInputArguments()) {
if (jvmArg.startsWith("-agentpath:") || jvmArg.startsWith("-agentlib:")) {
nativeAgentArgs.add(jvmArg);
}
}
return nativeAgentArgs;
}
private static List getJavaAgentArgsBeforeGlowroot(@Nullable File glowrootJarFile) {
if (glowrootJarFile == null) {
return ImmutableList.of();
}
List javaAgentArgsBeforeGlowroot = Lists.newArrayList();
for (String jvmArg : ManagementFactory.getRuntimeMXBean().getInputArguments()) {
if (jvmArg.startsWith("-javaagent:") && jvmArg.endsWith(glowrootJarFile.getName())) {
break;
}
if (jvmArg.startsWith("-javaagent:") || isIbmHealthcenterArg(jvmArg)) {
javaAgentArgsBeforeGlowroot.add(jvmArg);
}
}
return javaAgentArgsBeforeGlowroot;
}
private static boolean isIbmHealthcenterArg(String jvmArg) {
return JavaVersion.isIbmJvm()
&& (jvmArg.equals("-Xhealthcenter") || jvmArg.startsWith("-Xhealthcenter:"));
}
// now init plugins to give them a chance to do something in their static initializer
// e.g. append their package to jboss.modules.system.pkgs
private static void initPlugins(List pluginDescriptors) {
for (PluginDescriptor pluginDescriptor : pluginDescriptors) {
for (String aspect : pluginDescriptor.aspects()) {
try {
Class.forName(aspect, true, AgentModule.class.getClassLoader());
} catch (ClassNotFoundException e) {
// this would have already been logged as a warning during advice construction
logger.debug(e.getMessage(), e);
}
}
}
}
@OnlyUsedByTests
public void close() throws Exception {
if (immedateTraceStoreWatcher != null) {
immedateTraceStoreWatcher.cancel();
}
if (stackTraceCollector != null) {
stackTraceCollector.close();
}
if (gaugeCollector != null) {
gaugeCollector.close();
}
if (lazyPlatformMBeanServer != null) {
lazyPlatformMBeanServer.close();
}
if (transactionCollector != null) {
transactionCollector.close();
}
if (aggregator != null) {
aggregator.close();
}
if (deadlockedActiveWeavingRunnable != null) {
deadlockedActiveWeavingRunnable.cancel();
}
}
private static class DeadlockedActiveWeavingRunnable extends ScheduledRunnable {
private final Weaver weaver;
private DeadlockedActiveWeavingRunnable(Weaver weaver) {
this.weaver = weaver;
}
@Override
public void runInternal() {
weaver.checkForDeadlockedActiveWeaving();
}
}
}