io.opentelemetry.javaagent.instrumentation.servlet.v3_0.Servlet3OutputStreamWriteBytesAdvice Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of opentelemetry-javaagent-servlet-3.0 Show documentation
Show all versions of opentelemetry-javaagent-servlet-3.0 Show documentation
Instrumentation of Java libraries using OpenTelemetry.
The newest version!
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
package io.opentelemetry.javaagent.instrumentation.servlet.v3_0;
import static io.opentelemetry.javaagent.instrumentation.servlet.v3_0.Servlet3Singletons.getSnippetInjectionHelper;
import io.opentelemetry.javaagent.bootstrap.servlet.InjectionState;
import io.opentelemetry.javaagent.instrumentation.servlet.v3_0.snippet.ServletOutputStreamInjectionState;
import java.io.IOException;
import javax.servlet.ServletOutputStream;
import net.bytebuddy.asm.Advice;
public class Servlet3OutputStreamWriteBytesAdvice {
@Advice.OnMethodEnter(skipOn = Advice.OnDefaultValue.class, suppress = Throwable.class)
public static boolean methodEnter(
@Advice.This ServletOutputStream servletOutputStream, @Advice.Argument(0) byte[] write)
throws IOException {
InjectionState state = ServletOutputStreamInjectionState.getInjectionState(servletOutputStream);
if (state == null) {
return true;
}
// if handleWrite returns true, then it means the original bytes + the snippet were written
// to the servletOutputStream, and so we no longer need to execute the original method
// call (see skipOn above)
// if it returns false, then it means nothing was written to the servletOutputStream and the
// original method call should be executed
return !getSnippetInjectionHelper()
.handleWrite(state, servletOutputStream, write, 0, write.length);
}
}