private void deleteUserIncludeTrace(final Trace trace) {
if (isDebug) {
logger.debug("Delete user include trace={}, sampled={}", trace, trace.canSampled());
}
traceContext.removeTraceObject();
trace.close();
}
@Override
public void before(Object target, Object[] args) {
if (isDebug) {
logger.beforeInterceptor(target, args);
}
Trace trace = traceContext.currentRawTraceObject();
if (trace == null) {
// create user include trace for standalone entry point.
trace = createUserIncludeTrace();
if (trace == null) {
return;
}
}
// check user include trace.
if (!isUserIncludeTrace(trace)) {
return;
}
// entry scope(default & disable trace).
entryUserIncludeTraceScope(trace);
// check sampled.
if (!trace.canSampled()) {
// skip
return;
}
trace.traceBlockBegin();
}
@Override
public void after(Object target, Object[] args, Object result, Throwable throwable) {
if (isDebug) {
logger.afterInterceptor(target, args, result, throwable);
}
// Unset server markers
if (args[0] instanceof AsyncFrameBuffer) {
AsyncFrameBuffer frameBuffer = (AsyncFrameBuffer) args[0];
attachMarkersToInputProtocol(frameBuffer.getInputProtocol(), false);
}
final Trace trace = this.traceContext.currentRawTraceObject();
if (trace == null) {
return;
}
this.traceContext.removeTraceObject();
if (trace.canSampled()) {
try {
processTraceObject(trace, target, args, throwable);
} catch (Throwable t) {
logger.warn("Error processing trace object. Cause:{}", t.getMessage(), t);
} finally {
trace.close();
}
}
}
@Override
public void after(Object target, Object[] args, Object result, Throwable throwable) {
if (isDebug) {
logger.afterInterceptor(target, args);
}
final Trace trace = traceContext.currentRawTraceObject();
if (trace == null) {
return;
}
// check user include trace.
if (!isUserIncludeTrace(trace)) {
return;
}
// leave scope(default & disable trace).
if (!leaveUserIncludeTraceScope(trace)) {
logger.warn(
"Failed to leave scope of user include trace. trace={}, sampled={}",
trace,
trace.canSampled());
// delete unstable trace.
deleteUserIncludeTrace(trace);
return;
}
// check sampled.
if (!trace.canSampled()) {
if (isUserIncludeTraceDestination(trace)) {
deleteUserIncludeTrace(trace);
}
return;
}
try {
final SpanEventRecorder recorder = trace.currentSpanEventRecorder();
recorder.recordApi(descriptor);
recorder.recordServiceType(ServiceType.INTERNAL_METHOD);
recorder.recordException(throwable);
} finally {
trace.traceBlockEnd();
if (isUserIncludeTraceDestination(trace)) {
deleteUserIncludeTrace(trace);
}
}
}
@Override
public void before(Object target, Object[] args) {
if (isDebug) {
logger.beforeInterceptor(target, args);
}
if (!validate(target, args)) {
return;
}
final Trace trace = this.traceContext.currentRawTraceObject();
if (trace == null) {
return;
}
try {
ThriftRequestProperty parentTraceInfo = new ThriftRequestProperty();
final boolean shouldSample = trace.canSampled();
if (!shouldSample) {
if (isDebug) {
logger.debug("set Sampling flag=false");
}
parentTraceInfo.setShouldSample(shouldSample);
} else {
SpanEventRecorder recorder = trace.traceBlockBegin();
Object asyncMethodCallObj = args[0];
// inject async trace info to AsyncMethodCall object
final AsyncTraceId asyncTraceId = injectAsyncTraceId(asyncMethodCallObj, trace);
recorder.recordServiceType(THRIFT_CLIENT_INTERNAL);
// retrieve connection information
String remoteAddress = getRemoteAddress(asyncMethodCallObj);
final TraceId nextId = asyncTraceId.getNextTraceId();
// Inject nextSpanId as the actual sending of data will be handled asynchronously.
final long nextSpanId = nextId.getSpanId();
parentTraceInfo.setSpanId(nextSpanId);
parentTraceInfo.setTraceId(nextId.getTransactionId());
parentTraceInfo.setParentSpanId(nextId.getParentSpanId());
parentTraceInfo.setFlags(nextId.getFlags());
parentTraceInfo.setParentApplicationName(this.traceContext.getApplicationName());
parentTraceInfo.setParentApplicationType(this.traceContext.getServerTypeCode());
parentTraceInfo.setAcceptorHost(remoteAddress);
this.asyncCallRemoteAddressAccessor.set(asyncMethodCallObj, remoteAddress);
this.asyncNextSpanIdAccessor.set(asyncMethodCallObj, nextSpanId);
}
InterceptorGroupInvocation currentTransaction = this.group.getCurrentInvocation();
currentTransaction.setAttachment(parentTraceInfo);
} catch (Throwable t) {
logger.warn("before error. Caused:{}", t.getMessage(), t);
}
}
private Trace createUserIncludeTrace() {
final Trace trace = traceContext.newTraceObject();
if (isDebug) {
logger.debug("New user include trace {} and sampled {}", trace, trace.canSampled());
}
// add user scope.
TraceScope oldScope = trace.addScope(SCOPE_NAME);
if (oldScope != null) {
// delete corrupted trace.
logger.warn("Duplicated user include trace scope={}.", oldScope.getName());
deleteUserIncludeTrace(trace);
return null;
}
if (trace.canSampled()) {
// record root span.
final SpanRecorder recorder = trace.getSpanRecorder();
recorder.recordServiceType(ServiceType.STAND_ALONE);
recorder.recordApi(USER_INCLUDE_METHOD_DESCRIPTOR);
}
return trace;
}
@Override
public void before(Object target, Object[] args) {
if (isDebug) {
logger.beforeInterceptor(target, args);
}
final Trace trace = traceContext.currentRawTraceObject();
if (trace == null) {
return;
}
if (!validate(target)) {
return;
}
if (!trace.canSampled()) {
return;
}
SpanEventRecorder recorder = trace.traceBlockBegin();
try {
final TraceId nextId = trace.getTraceId().getNextTraceId();
recorder.recordNextSpanId(nextId.getSpanId());
recorder.recordServiceType(OK_HTTP_CLIENT);
InterceptorGroupInvocation invocation = interceptorGroup.getCurrentInvocation();
if (invocation != null) {
invocation.getOrCreateAttachment(
new AttachmentFactory() {
@Override
public Object createAttachment() {
return nextId;
}
});
}
} catch (Throwable t) {
logger.warn("Failed to BEFORE process. {}", t.getMessage(), t);
}
}
@Override
public void after(Object target, Object[] args, Object result, Throwable throwable) {
final Trace trace = this.traceContext.currentRawTraceObject();
if (trace == null) {
return;
}
// logging here as some Thrift servers depend on TTransportException being thrown for normal
// operations.
// log only when current transaction is being traced.
if (isDebug) {
logger.afterInterceptor(target, args, result, throwable);
}
this.traceContext.removeTraceObject();
if (trace.canSampled()) {
try {
processTraceObject(trace, target, args, throwable);
} catch (Throwable t) {
logger.warn("Error processing trace object. Cause:{}", t.getMessage(), t);
} finally {
trace.close();
}
}
}