org.glowroot.agent.it.harness.impl.JavaagentContainer 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.it.harness.impl;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.management.ManagementFactory;
import java.lang.management.RuntimeMXBean;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.jar.JarEntry;
import java.util.jar.JarInputStream;
import org.glowroot.agent.it.harness.shaded.com.google.common.base.Joiner;
import org.glowroot.agent.it.harness.shaded.com.google.common.base.Splitter;
import org.glowroot.agent.it.harness.shaded.com.google.common.base.StandardSystemProperty;
import org.glowroot.agent.it.harness.shaded.com.google.common.base.Stopwatch;
import org.glowroot.agent.it.harness.shaded.com.google.common.base.Strings;
import org.glowroot.agent.it.harness.shaded.com.google.common.collect.ImmutableList;
import org.glowroot.agent.it.harness.shaded.com.google.common.collect.Lists;
import org.glowroot.agent.it.harness.shaded.com.google.common.io.ByteStreams;
import org.glowroot.agent.it.harness.shaded.com.google.common.util.concurrent.ThreadFactoryBuilder;
import org.glowroot.agent.it.harness.shaded.io.grpc.ManagedChannel;
import org.glowroot.agent.it.harness.shaded.io.grpc.netty.NegotiationType;
import org.glowroot.agent.it.harness.shaded.io.grpc.netty.NettyChannelBuilder;
import org.glowroot.agent.it.harness.shaded.io.netty.channel.EventLoopGroup;
import org.checkerframework.checker.nullness.qual.Nullable;
import org.glowroot.agent.shaded.org.slf4j.Logger;
import org.glowroot.agent.shaded.org.slf4j.LoggerFactory;
import org.glowroot.agent.AgentPremain;
import org.glowroot.agent.it.harness.AppUnderTest;
import org.glowroot.agent.it.harness.ConfigService;
import org.glowroot.agent.it.harness.Container;
import org.glowroot.agent.it.harness.TempDirs;
import org.glowroot.agent.it.harness.grpc.JavaagentServiceGrpc;
import org.glowroot.agent.it.harness.grpc.JavaagentServiceGrpc.JavaagentServiceBlockingStub;
import org.glowroot.agent.it.harness.grpc.JavaagentServiceOuterClass.AppUnderTestClassName;
import org.glowroot.agent.it.harness.grpc.JavaagentServiceOuterClass.Void;
import org.glowroot.wire.api.model.TraceOuterClass.Trace;
import static 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.SECONDS;
public class JavaagentContainer implements Container {
private static final boolean XDEBUG = Boolean.getBoolean("glowroot.test.xdebug");
private static final Logger logger = LoggerFactory.getLogger(JavaagentContainer.class);
private final File testDir;
private final boolean deleteTestDirOnClose;
private final ServerSocket heartbeatListenerSocket;
private final ExecutorService heartbeatListenerExecutor;
private final @Nullable GrpcServerWrapper server;
private final EventLoopGroup eventLoopGroup;
private final ExecutorService executor;
private final ManagedChannel channel;
private final @Nullable TraceCollector traceCollector;
private final JavaagentServiceBlockingStub javaagentService;
private final ExecutorService consolePipeExecutor;
private final Future consolePipeFuture;
private final Process process;
private final ConsoleOutputPipe consoleOutputPipe;
private final @Nullable ConfigServiceImpl configService;
private final Thread shutdownHook;
public static JavaagentContainer create() throws Exception {
return new JavaagentContainer(null, false, ImmutableList.of());
}
public static JavaagentContainer create(File testDir) throws Exception {
return new JavaagentContainer(testDir, false, ImmutableList.of());
}
public static JavaagentContainer createWithExtraJvmArgs(List extraJvmArgs)
throws Exception {
return new JavaagentContainer(null, false, extraJvmArgs);
}
public JavaagentContainer(@Nullable File testDir, boolean embedded, List extraJvmArgs)
throws Exception {
if (testDir == null) {
this.testDir = TempDirs.createTempDir("glowroot-test-dir");
deleteTestDirOnClose = true;
} else {
this.testDir = testDir;
deleteTestDirOnClose = false;
}
// need to start heartbeat socket listener before spawning process
heartbeatListenerSocket = new ServerSocket(0);
heartbeatListenerExecutor = Executors.newSingleThreadExecutor();
heartbeatListenerExecutor.execute(new Runnable() {
@Override
public void run() {
try {
// TODO report checker framework issue that occurs without checkNotNull
Socket socket = checkNotNull(heartbeatListenerSocket).accept();
InputStream socketIn = socket.getInputStream();
ByteStreams.exhaust(socketIn);
} catch (IOException e) {
logger.error(e.getMessage(), e);
}
}
});
boolean pointingToCentral = false;
for (String extraJvmArg : extraJvmArgs) {
if (extraJvmArg.startsWith("-Dglowroot.collector.address=")) {
pointingToCentral = true;
break;
}
}
int collectorPort;
if (embedded || pointingToCentral) {
collectorPort = 0;
traceCollector = null;
server = null;
} else {
collectorPort = LocalContainer.getAvailablePort();
traceCollector = new TraceCollector();
server = new GrpcServerWrapper(traceCollector, collectorPort);
}
int javaagentServicePort = LocalContainer.getAvailablePort();
List command = buildCommand(heartbeatListenerSocket.getLocalPort(), collectorPort,
javaagentServicePort, this.testDir, extraJvmArgs);
ProcessBuilder processBuilder = new ProcessBuilder(command);
processBuilder.redirectErrorStream(true);
Process process = processBuilder.start();
consolePipeExecutor = Executors.newSingleThreadExecutor();
InputStream in = process.getInputStream();
// process.getInputStream() only returns null if ProcessBuilder.redirectOutput() is used
// to redirect output to a file
checkNotNull(in);
consoleOutputPipe = new ConsoleOutputPipe(in, System.out);
consolePipeFuture = consolePipeExecutor.submit(consoleOutputPipe);
this.process = process;
eventLoopGroup = EventLoopGroups.create("Glowroot-IT-Harness*-GRPC-Worker-ELG");
executor = Executors.newCachedThreadPool(
new ThreadFactoryBuilder()
.setDaemon(true)
.setNameFormat("Glowroot-IT-Harness*-GRPC-Executor-%d")
.build());
channel = NettyChannelBuilder.forAddress("localhost", javaagentServicePort)
.eventLoopGroup(eventLoopGroup)
.executor(executor)
.negotiationType(NegotiationType.PLAINTEXT)
.build();
Stopwatch stopwatch = Stopwatch.createStarted();
// this can take a while on slow travis ci build machines
while (stopwatch.elapsed(SECONDS) < 30) {
try {
JavaagentServiceBlockingStub javaagentService =
JavaagentServiceGrpc.newBlockingStub(channel)
.withCompression("gzip");
javaagentService.ping(Void.getDefaultInstance());
break;
} catch (Exception e) {
logger.debug(e.getMessage(), e);
}
MILLISECONDS.sleep(100);
}
javaagentService = JavaagentServiceGrpc.newBlockingStub(channel)
.withCompression("gzip");
if (server == null) {
configService = null;
javaagentService.setSlowThresholdToZero(Void.getDefaultInstance());
} else {
configService = new ConfigServiceImpl(server, true);
// need to set through config service so config service can keep track of changes,
// otherwise it will clobber slow threshold value on next update through config service
configService.setSlowThresholdToZero();
}
shutdownHook = new ShutdownHookThread(javaagentService);
// unfortunately, ctrl-c during maven test will kill the maven process, but won't kill the
// forked surefire jvm where the tests are being run
// (http://jira.codehaus.org/browse/SUREFIRE-413), and so this hook won't get triggered by
// ctrl-c while running tests under maven
Runtime.getRuntime().addShutdownHook(shutdownHook);
}
@Override
public ConfigService getConfigService() {
checkNotNull(configService);
return configService;
}
@Override
public void addExpectedLogMessage(String loggerName, String partialMessage) throws Exception {
checkNotNull(traceCollector);
traceCollector.addExpectedLogMessage(loggerName, partialMessage);
}
@Override
public Trace execute(Class appClass) throws Exception {
return executeInternal(appClass, null, null);
}
@Override
public Trace execute(Class appClass, String transactionType)
throws Exception {
return executeInternal(appClass, transactionType, null);
}
@Override
public Trace execute(Class appClass, String transactionType,
String transactionName) throws Exception {
return executeInternal(appClass, transactionType, transactionName);
}
@Override
public void executeNoExpectedTrace(Class appClass) throws Exception {
executeInternal(appClass);
// give a short time to see if trace gets collected
MILLISECONDS.sleep(10);
if (traceCollector != null && traceCollector.hasTrace()) {
throw new IllegalStateException("Trace was collected when none was expected");
}
}
@Override
public void interruptAppUnderTest() throws Exception {
javaagentService.interruptApp(Void.getDefaultInstance());
}
@Override
public Trace getCollectedPartialTrace() throws InterruptedException {
checkNotNull(traceCollector);
return traceCollector.getPartialTrace(10, SECONDS);
}
@Override
public void checkAndReset() throws Exception {
if (configService == null) {
javaagentService.resetConfig(Void.getDefaultInstance());
} else {
// need to reset through config service so config service can keep track of changes,
// otherwise it will clobber the reset config on next update through config service
configService.resetConfig();
}
if (traceCollector != null) {
traceCollector.checkAndResetLogMessages();
}
}
@Override
public void close() throws Exception {
javaagentService.shutdown(Void.getDefaultInstance());
javaagentService.kill(Void.getDefaultInstance());
channel.shutdown();
if (!channel.awaitTermination(10, SECONDS)) {
throw new IllegalStateException("Could not terminate channel");
}
executor.shutdown();
if (!executor.awaitTermination(10, SECONDS)) {
throw new IllegalStateException("Could not terminate executor");
}
if (!eventLoopGroup.shutdownGracefully(0, 0, SECONDS).await(10, SECONDS)) {
throw new IllegalStateException("Could not terminate event loop group");
}
if (server != null) {
server.close();
}
process.waitFor();
consolePipeFuture.get();
consolePipeExecutor.shutdown();
if (!consolePipeExecutor.awaitTermination(10, SECONDS)) {
throw new IllegalStateException("Could not terminate executor");
}
heartbeatListenerExecutor.shutdown();
if (!heartbeatListenerExecutor.awaitTermination(10, SECONDS)) {
throw new IllegalStateException("Could not terminate executor");
}
heartbeatListenerSocket.close();
Runtime.getRuntime().removeShutdownHook(shutdownHook);
if (deleteTestDirOnClose) {
TempDirs.deleteRecursively(testDir);
}
}
private Trace executeInternal(Class appClass,
@Nullable String transactionType, @Nullable String transactionName) throws Exception {
checkNotNull(traceCollector);
executeInternal(appClass);
// extra long wait time is needed for StackOverflowOOMIT on slow travis ci machines since it
// can sometimes take a long time for that large trace to be serialized and transferred
Trace trace =
traceCollector.getCompletedTrace(transactionType, transactionName, 20, SECONDS);
traceCollector.clearTrace();
return trace;
}
private void executeInternal(Class appUnderTestClass) {
javaagentService.executeApp(AppUnderTestClassName.newBuilder()
.setValue(appUnderTestClass.getName())
.build());
}
private static List buildCommand(int heartbeatPort, int collectorPort,
int javaagentServicePort, File testDir, List extraJvmArgs) throws Exception {
List command = Lists.newArrayList();
String javaExecutable = StandardSystemProperty.JAVA_HOME.value() + File.separator + "bin"
+ File.separator + "java";
command.add(javaExecutable);
boolean hasXmx = false;
for (String extraJvmArg : extraJvmArgs) {
command.add(extraJvmArg);
if (extraJvmArg.startsWith("-Xmx")) {
hasXmx = true;
}
}
// it is important for jacoco javaagent to be prior to glowroot javaagent so that jacoco
// will use original class bytes to form its class id at runtime which will then match up
// with the class id at analysis time
command.addAll(getJacocoArgsFromCurrentJvm());
String classpath = Strings.nullToEmpty(StandardSystemProperty.JAVA_CLASS_PATH.value());
List bootPaths = Lists.newArrayList();
List paths = Lists.newArrayList();
List maybeBootPaths = Lists.newArrayList();
File javaagentJarFile = null;
for (String path : Splitter.on(File.pathSeparatorChar).split(classpath)) {
File file = new File(path);
String name = file.getName();
String targetClasses = File.separator + "target" + File.separator + "classes";
if (name.matches("glowroot-agent-core(-unshaded)?-[0-9.]+(-SNAPSHOT)?.jar")
|| name.matches("glowroot-agent-it-harness-[0-9.]+(-SNAPSHOT)?.jar")) {
javaagentJarFile = file;
} else if (name.matches("glowroot-common-[0-9.]+(-SNAPSHOT)?.jar")
|| name.matches("glowroot-wire-api-[0-9.]+(-SNAPSHOT)?.jar")
|| name.matches("glowroot-agent-api-[0-9.]+(-SNAPSHOT)?.jar")
|| name.matches("glowroot-agent-plugin-api-[0-9.]+(-SNAPSHOT)?.jar")
|| name.matches("glowroot-agent-bytecode-api-[0-9.]+(-SNAPSHOT)?.jar")
|| name.matches("glowroot-build-error-prone-jdk6-[0-9.]+(-SNAPSHOT)?.jar")) {
// these are glowroot-agent-core-unshaded transitive dependencies
maybeBootPaths.add(path);
} else if (file.getAbsolutePath().endsWith(File.separator + "common" + targetClasses)
|| file.getAbsolutePath().endsWith(File.separator + "wire-api" + targetClasses)
|| file.getAbsolutePath().endsWith(File.separator + "api" + targetClasses)
|| file.getAbsolutePath()
.endsWith(File.separator + "plugin-api" + targetClasses)
|| file.getAbsolutePath()
.endsWith(File.separator + "bytecode-api" + targetClasses)
|| file.getAbsolutePath()
.endsWith(File.separator + "error-prone-jdk6" + targetClasses)) {
// these are glowroot-agent-core-unshaded transitive dependencies
maybeBootPaths.add(path);
} else if (name.matches("asm-.*\\.jar")
|| name.matches("grpc-.*\\.jar")
|| name.matches("opencensus-.*\\.jar")
|| name.matches("guava-.*\\.jar")
|| name.matches("HdrHistogram-.*\\.jar")
|| name.matches("instrumentation-api-.*\\.jar")
|| name.matches("jackson-.*\\.jar")
|| name.matches("logback-.*\\.jar")
// javax.servlet-api is needed because logback-classic has
// META-INF/services/javax.servlet.ServletContainerInitializer
|| name.matches("javax.servlet-api-.*\\.jar")
|| name.matches("netty-buffer-.*\\.jar")
|| name.matches("netty-codec-.*\\.jar")
|| name.matches("netty-codec-http2-.*\\.jar")
|| name.matches("netty-codec-http-.*\\.jar")
|| name.matches("netty-codec-socks-.*\\.jar")
|| name.matches("netty-common-.*\\.jar")
|| name.matches("netty-handler-.*\\.jar")
|| name.matches("netty-handler-proxy-.*\\.jar")
|| name.matches("netty-resolver-.*\\.jar")
|| name.matches("netty-transport-.*\\.jar")
// optional netty dependency that is required by HttpContentCompressor, need to
// include in bootstrap class loader since netty is
|| name.matches("jzlib-.*\\.jar")
|| name.matches("protobuf-java-.*\\.jar")
|| name.matches("slf4j-api-.*\\.jar")
|| name.matches("value-.*\\.jar")
|| name.matches("error_prone_annotations-.*\\.jar")
|| name.matches("jsr305-.*\\.jar")) {
// these are glowroot-agent-core-unshaded transitive dependencies
maybeBootPaths.add(path);
} else if (name.matches("glowroot-common2-[0-9.]+(-SNAPSHOT)?.jar")
|| name.matches("glowroot-ui-[0-9.]+(-SNAPSHOT)?.jar")
|| name.matches(
"glowroot-agent-embedded(-unshaded)?-[0-9.]+(-SNAPSHOT)?.jar")) {
// these are glowroot-agent-embedded-unshaded transitive dependencies
paths.add(path);
} else if (file.getAbsolutePath().endsWith(File.separator + "common2" + targetClasses)
|| file.getAbsolutePath().endsWith(File.separator + "ui" + targetClasses)
|| file.getAbsolutePath()
.endsWith(File.separator + "embedded" + targetClasses)) {
// these are glowroot-agent-embedded-unshaded transitive dependencies
paths.add(path);
} else if (name.matches("compress-.*\\.jar")
|| name.matches("h2-.*\\.jar")
|| name.matches("mailapi-.*\\.jar")
|| name.matches("smtp-.*\\.jar")) {
// these are glowroot-agent-embedded-unshaded transitive dependencies
paths.add(path);
} else if (name.matches("glowroot-agent-it-harness-unshaded-[0-9.]+(-SNAPSHOT)?.jar")) {
// this is integration test harness, needs to be in bootstrap class loader when it
// it is shaded (because then it contains glowroot-agent-core), and for consistency
// putting it in bootstrap class loader at other times as well
bootPaths.add(path);
} else if (file.getAbsolutePath()
.endsWith(File.separator + "it-harness" + targetClasses)) {
// this is integration test harness, needs to be in bootstrap class loader when it
// it is shaded (because then it contains glowroot-agent-core), and for consistency
// putting it in bootstrap class loader at other times as well
bootPaths.add(path);
} else if (name.endsWith(".jar") && file.getAbsolutePath()
.endsWith(File.separator + "target" + File.separator + name)) {
// this is the plugin under test
bootPaths.add(path);
} else if (name.matches("glowroot-agent-[a-z-]+-plugin-[0-9.]+(-SNAPSHOT)?.jar")) {
// this another (core) plugin that it depends on, e.g. the executor plugin
bootPaths.add(path);
} else if (file.getAbsolutePath().endsWith(targetClasses)) {
// this is the plugin under test
bootPaths.add(path);
} else if (file.getAbsolutePath()
.endsWith(File.separator + "target" + File.separator + "test-classes")) {
// this is the plugin test classes
paths.add(path);
} else {
// these are plugin test dependencies
paths.add(path);
}
}
if (javaagentJarFile == null) {
bootPaths.addAll(maybeBootPaths);
} else {
boolean shaded = false;
JarInputStream jarIn = new JarInputStream(new FileInputStream(javaagentJarFile));
try {
JarEntry jarEntry;
while ((jarEntry = jarIn.getNextJarEntry()) != null) {
if (jarEntry.getName().startsWith("org/glowroot/agent/shaded/")) {
shaded = true;
break;
}
}
} finally {
jarIn.close();
}
if (shaded) {
paths.addAll(maybeBootPaths);
} else {
bootPaths.addAll(maybeBootPaths);
}
}
command.add("-Xbootclasspath/a:" + Joiner.on(File.pathSeparatorChar).join(bootPaths));
command.add("-classpath");
command.add(Joiner.on(File.pathSeparatorChar).join(paths));
if (XDEBUG) {
// the -agentlib arg needs to come before the -javaagent arg
command.add("-Xdebug");
command.add("-agentlib:jdwp=transport=dt_socket,address=8000,server=y,suspend=y");
}
if (javaagentJarFile == null) {
// create jar file in test dir since that gets cleaned up at end of test already
javaagentJarFile = DelegatingJavaagent.createDelegatingJavaagentJarFile(testDir);
command.add("-javaagent:" + javaagentJarFile);
command.add("-DdelegateJavaagent=" + AgentPremain.class.getName());
} else {
command.add("-javaagent:" + javaagentJarFile);
}
command.add("-Dglowroot.test.dir=" + testDir.getAbsolutePath());
if (collectorPort != 0) {
command.add("-Dglowroot.collector.address=localhost:" + collectorPort);
}
command.add("-Dglowroot.debug.preCheckLoadedClasses=true");
// this is used inside low-entropy docker containers
String sourceOfRandomness = System.getProperty("java.security.egd");
if (sourceOfRandomness != null) {
command.add("-Djava.security.egd=" + sourceOfRandomness);
}
if (!hasXmx) {
command.add("-Xmx" + Runtime.getRuntime().maxMemory());
}
// leave as much memory as possible to old gen
command.add("-XX:NewRatio=20");
for (Map.Entry entry : System.getProperties().entrySet()) {
Object keyObject = entry.getKey();
if (!(keyObject instanceof String)) {
continue;
}
String key = (String) keyObject;
if (key.startsWith("glowroot.internal.") || key.startsWith("glowroot.test.")) {
command.add("-D" + key + "=" + entry.getValue());
}
}
command.add(JavaagentMain.class.getName());
command.add(Integer.toString(heartbeatPort));
command.add(Integer.toString(javaagentServicePort));
return command;
}
private static List getJacocoArgsFromCurrentJvm() {
RuntimeMXBean runtimeMXBean = ManagementFactory.getRuntimeMXBean();
List arguments = runtimeMXBean.getInputArguments();
List jacocoArgs = Lists.newArrayList();
for (String argument : arguments) {
if (argument.startsWith("-javaagent:") && argument.contains("jacoco")) {
jacocoArgs.add(argument + ",inclbootstrapclasses=true,includes=org.glowroot.*");
break;
}
}
return jacocoArgs;
}
private static class ShutdownHookThread extends Thread {
private final JavaagentServiceBlockingStub javaagentService;
private ShutdownHookThread(JavaagentServiceBlockingStub javaagentService) {
this.javaagentService = javaagentService;
}
@Override
public void run() {
try {
javaagentService.kill(Void.getDefaultInstance());
} catch (Exception e) {
logger.error(e.getMessage(), e);
}
}
}
private static class ConsoleOutputPipe implements Runnable {
private final InputStream in;
private final OutputStream out;
private ConsoleOutputPipe(InputStream in, OutputStream out) {
this.in = in;
this.out = out;
}
@Override
public void run() {
try {
ByteStreams.copy(in, out);
} catch (IOException e) {
logger.error(e.getMessage(), e);
}
}
}
}