mirror of
https://github.com/wahyd4/pinpoint.git
synced 2026-08-15 15:55:54 +10:00
Woonduk Kang
48c99d2ffd
git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@1345 84d0f5b1-2673-498c-a247-62c4ff18d310
382 lines
12 KiB
Java
382 lines
12 KiB
Java
package com.profiler.context;
|
|
|
|
import com.profiler.common.AnnotationKey;
|
|
import com.profiler.common.ServiceType;
|
|
import com.profiler.common.util.ParsingResult;
|
|
import com.profiler.interceptor.MethodDescriptor;
|
|
import com.profiler.logging.LoggingUtils;
|
|
import com.profiler.util.StringUtils;
|
|
|
|
import java.util.List;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
|
|
/**
|
|
* @author netspider
|
|
*/
|
|
public final class Trace {
|
|
|
|
private static final Logger logger = Logger.getLogger(Trace.class.getName());
|
|
private static final boolean isDebug = LoggingUtils.isDebug(logger);
|
|
|
|
public static final int NOCHECK_STACKID = -1;
|
|
public static final int ROOT_STACKID = 0;
|
|
|
|
private short sequence;
|
|
|
|
private boolean tracingEnabled = true;
|
|
|
|
private CallStack callStack;
|
|
|
|
private Storage storage;
|
|
|
|
private TraceContext traceContext;
|
|
|
|
// use for calculating depth of each Span.
|
|
private int latestStackIndex = -1;
|
|
|
|
public Trace() {
|
|
TraceID traceId = TraceID.newTraceId();
|
|
this.callStack = new CallStack(traceId);
|
|
latestStackIndex = this.callStack.push();
|
|
StackFrame stackFrame = createRootStackFrame(ROOT_STACKID, callStack.getSpan());
|
|
this.callStack.setStackFrame(stackFrame);
|
|
}
|
|
|
|
public Trace(TraceID continueRoot) {
|
|
// this.root = continueRoot;
|
|
this.callStack = new CallStack(continueRoot);
|
|
latestStackIndex = this.callStack.push();
|
|
StackFrame stackFrame = createRootStackFrame(ROOT_STACKID, callStack.getSpan());
|
|
this.callStack.setStackFrame(stackFrame);
|
|
}
|
|
|
|
public CallStack getCallStack() {
|
|
return callStack;
|
|
}
|
|
|
|
|
|
public Storage getStorage() {
|
|
return storage;
|
|
}
|
|
|
|
public void setStorage(Storage storage) {
|
|
this.storage = storage;
|
|
}
|
|
|
|
public short getSequence() {
|
|
return sequence++;
|
|
}
|
|
|
|
public AsyncTrace createAsyncTrace() {
|
|
// 경우에 따라 별도 timeout 처리가 있어야 될수도 있음.
|
|
SpanEvent spanEvent = new SpanEvent(callStack.getSpan());
|
|
spanEvent.setSequence(getSequence());
|
|
AsyncTrace asyncTrace = new AsyncTrace(spanEvent);
|
|
// asyncTrace.setDataSender(this.getDataSender());
|
|
asyncTrace.setStorage(this.storage);
|
|
return asyncTrace;
|
|
}
|
|
|
|
private StackFrame createSubStackFrame(int stackId) {
|
|
SpanEvent spanEvent = new SpanEvent(callStack.getSpan());
|
|
SubStackFrame stackFrame = new SubStackFrame(spanEvent);
|
|
stackFrame.setStackFrameId(stackId);
|
|
stackFrame.setSequence(getSequence());
|
|
return stackFrame;
|
|
}
|
|
|
|
private StackFrame createRootStackFrame(int stackId, Span span) {
|
|
RootStackFrame stackFrame = new RootStackFrame(span);
|
|
stackFrame.setStackFrameId(stackId);
|
|
stackFrame.setSpan(span);
|
|
stackFrame.setStackFrameId(ROOT_STACKID);
|
|
return stackFrame;
|
|
}
|
|
|
|
public void traceBlockBegin() {
|
|
traceBlockBegin(NOCHECK_STACKID);
|
|
}
|
|
|
|
public void markBeforeTime() {
|
|
StackFrame stackFrame = getCurrentStackFrame();
|
|
stackFrame.markBeforeTime();
|
|
}
|
|
|
|
public long getBeforeTime() {
|
|
StackFrame stackFrame = getCurrentStackFrame();
|
|
return stackFrame.getBeforeTime();
|
|
}
|
|
|
|
public void markAfterTime() {
|
|
StackFrame stackFrame = getCurrentStackFrame();
|
|
stackFrame.markAfterTime();
|
|
}
|
|
|
|
public long getAfterTime() {
|
|
StackFrame stackFrame = getCurrentStackFrame();
|
|
return stackFrame.getAfterTime();
|
|
}
|
|
|
|
|
|
public void traceBlockBegin(int stackId) {
|
|
int currentStackIndex = callStack.push();
|
|
StackFrame stackFrame = createSubStackFrame(stackId);
|
|
|
|
if (latestStackIndex != currentStackIndex) {
|
|
latestStackIndex = currentStackIndex;
|
|
SpanEvent spanEvent = ((SubStackFrame) stackFrame).getSpanEvent();
|
|
spanEvent.setDepth(latestStackIndex);
|
|
}
|
|
|
|
callStack.setStackFrame(stackFrame);
|
|
}
|
|
|
|
public void traceRootBlockEnd() {
|
|
traceBlockEnd(ROOT_STACKID);
|
|
}
|
|
|
|
public void traceBlockEnd() {
|
|
traceBlockEnd(NOCHECK_STACKID);
|
|
}
|
|
|
|
|
|
public void traceBlockEnd(int stackId) {
|
|
StackFrame currentStackFrame = callStack.getCurrentStackFrame();
|
|
int stackFrameId = currentStackFrame.getStackFrameId();
|
|
if (stackFrameId != stackId) {
|
|
// 자체 stack dump를 하면 오류발견이 쉬울것으로 생각됨
|
|
if (logger.isLoggable(Level.WARNING)) {
|
|
logger.warning("Corrupted CallStack found. StackId not matched. expected:" + stackId + " current:" + stackFrameId);
|
|
}
|
|
}
|
|
if (currentStackFrame instanceof RootStackFrame) {
|
|
logSpan(((RootStackFrame) currentStackFrame).getSpan());
|
|
} else {
|
|
logSpan(((SubStackFrame) currentStackFrame).getSpanEvent());
|
|
}
|
|
callStack.pop();
|
|
}
|
|
|
|
public StackFrame getCurrentStackFrame() {
|
|
return callStack.getCurrentStackFrame();
|
|
}
|
|
|
|
/**
|
|
* Get current TraceID. If it was not set this will return null.
|
|
*
|
|
* @return
|
|
*/
|
|
public TraceID getTraceId() {
|
|
return callStack.getSpan().getTraceID();
|
|
}
|
|
|
|
public void enable() {
|
|
tracingEnabled = true;
|
|
}
|
|
|
|
public void disable() {
|
|
tracingEnabled = false;
|
|
}
|
|
|
|
void logSpan(SpanEvent spanEvent) {
|
|
if (isDebug) {
|
|
Thread th = Thread.currentThread();
|
|
logger.fine("[WRITE SpanEvent]" + spanEvent + ", Thread ID=" + th.getId() + " Name=" + th.getName());
|
|
}
|
|
this.storage.store(spanEvent);
|
|
}
|
|
|
|
void logSpan(Span span) {
|
|
if (isDebug) {
|
|
Thread th = Thread.currentThread();
|
|
logger.info("[WRITE SPAN]" + span + ", Thread ID=" + th.getId() + " Name=" + th.getName());
|
|
}
|
|
this.storage.store(span);
|
|
}
|
|
|
|
public void recordException(Object result) {
|
|
if (result instanceof Throwable) {
|
|
Throwable th = (Throwable) result;
|
|
String drop = StringUtils.drop(th.getMessage(), 256);
|
|
recordAttribute(AnnotationKey.EXCEPTION, drop);
|
|
|
|
Span span = getCallStack().getSpan();
|
|
if (span.getException() == 0) {
|
|
span.setException(1);
|
|
}
|
|
}
|
|
}
|
|
|
|
public void recordApi(MethodDescriptor methodDescriptor) {
|
|
if (methodDescriptor == null) {
|
|
return;
|
|
}
|
|
if (methodDescriptor.getApiId() == 0) {
|
|
recordAttribute(AnnotationKey.API, methodDescriptor.getFullName());
|
|
} else {
|
|
recordAttribute(AnnotationKey.API_DID, methodDescriptor.getApiId());
|
|
}
|
|
}
|
|
|
|
public void recordApi(MethodDescriptor methodDescriptor, Object[] args) {
|
|
// API 저장 방법의 개선 필요.
|
|
recordApi(methodDescriptor);
|
|
recocordArgs(args);
|
|
}
|
|
|
|
public void recordApi(int apiId) {
|
|
recordAttribute(AnnotationKey.API_ID, apiId);
|
|
}
|
|
|
|
public void recordApi(int apiId, Object[] args) {
|
|
recordAttribute(AnnotationKey.API_ID, apiId);
|
|
recocordArgs(args);
|
|
}
|
|
|
|
private void recocordArgs(Object[] args) {
|
|
if (args != null) {
|
|
int min = Math.min(args.length, AnnotationKey.MAX_ARGS_SIZE);
|
|
for (int i = 0; i < min; i++) {
|
|
recordAttribute(AnnotationKey.getArgs(i), args[i]);
|
|
}
|
|
// TODO MAX 사이즈를 넘는건 마크만 해줘야 하나?
|
|
}
|
|
}
|
|
|
|
|
|
public void recordAttribute(final AnnotationKey key, final String value) {
|
|
recordAttribute(key, (Object) value);
|
|
}
|
|
|
|
public ParsingResult recordSqlInfo(String sql) {
|
|
if (sql == null) {
|
|
return null;
|
|
}
|
|
ParsingResult parsingResult = traceContext.parseSql(sql);
|
|
recordSqlParsingResult(parsingResult);
|
|
return parsingResult;
|
|
}
|
|
|
|
public void recordSqlParsingResult(ParsingResult parsingResult) {
|
|
recordAttribute(AnnotationKey.SQL_ID, parsingResult.getSql().hashCode());
|
|
String output = parsingResult.getOutput();
|
|
if (output != null && output.length() != 0) {
|
|
recordAttribute(AnnotationKey.SQL_PARAM, output);
|
|
}
|
|
}
|
|
|
|
public void recordAttribute(final AnnotationKey key, final Object value) {
|
|
// TODO API 단일화 필요.
|
|
StackFrame currentStackFrame = getCurrentStackFrame();
|
|
if (currentStackFrame instanceof RootStackFrame) {
|
|
Span span = ((RootStackFrame) currentStackFrame).getSpan();
|
|
span.addAnnotation(new Annotation(key, value));
|
|
} else {
|
|
SpanEvent spanEvent = ((SubStackFrame) currentStackFrame).getSpanEvent();
|
|
spanEvent.addAnnotation(new Annotation(key, value));
|
|
}
|
|
|
|
}
|
|
|
|
|
|
public void recordServiceType(final ServiceType serviceType) {
|
|
// TODO API 단일화 필요.
|
|
StackFrame currentStackFrame = getCurrentStackFrame();
|
|
if (currentStackFrame instanceof RootStackFrame) {
|
|
Span span = ((RootStackFrame) currentStackFrame).getSpan();
|
|
span.setServiceType(serviceType);
|
|
} else {
|
|
SpanEvent spanEvent = ((SubStackFrame) currentStackFrame).getSpanEvent();
|
|
spanEvent.setServiceType(serviceType);
|
|
}
|
|
|
|
}
|
|
|
|
public void recordRpcName(final String rpc) {
|
|
// TODO API 단일화 필요.
|
|
StackFrame currentStackFrame = getCurrentStackFrame();
|
|
if (currentStackFrame instanceof RootStackFrame) {
|
|
Span span = ((RootStackFrame) currentStackFrame).getSpan();
|
|
span.setRpc(rpc);
|
|
} else {
|
|
SpanEvent spanEvent = ((SubStackFrame) currentStackFrame).getSpanEvent();
|
|
spanEvent.setRpc(rpc);
|
|
}
|
|
|
|
}
|
|
|
|
public void recordDestinationId(final String destinationId) {
|
|
// TODO API 단일화 필요.
|
|
StackFrame currentStackFrame = getCurrentStackFrame();
|
|
if (currentStackFrame instanceof SubStackFrame) {
|
|
SpanEvent spanEvent = ((SubStackFrame) currentStackFrame).getSpanEvent();
|
|
spanEvent.setDestionationId(destinationId);
|
|
}
|
|
}
|
|
|
|
public void recordDestinationAddress(List<String> address) {
|
|
// TODO API 단일화 필요.
|
|
StackFrame currentStackFrame = getCurrentStackFrame();
|
|
if (currentStackFrame instanceof SubStackFrame) {
|
|
SpanEvent spanEvent = ((SubStackFrame) currentStackFrame).getSpanEvent();
|
|
spanEvent.setDestinationAddress();
|
|
}
|
|
}
|
|
|
|
public void recordDestinationAddressList(List<String> addressList) {
|
|
//To change body of created methods use File | Settings | File Templates.
|
|
}
|
|
|
|
public void recordEndPoint(final String endPoint) {
|
|
// TODO API 단일화 필요.
|
|
StackFrame currentStackFrame = getCurrentStackFrame();
|
|
if (currentStackFrame instanceof RootStackFrame) {
|
|
Span span = ((RootStackFrame) currentStackFrame).getSpan();
|
|
span.setEndPoint(endPoint);
|
|
} else {
|
|
SpanEvent spanEvent = ((SubStackFrame) currentStackFrame).getSpanEvent();
|
|
spanEvent.setEndPoint(endPoint);
|
|
}
|
|
}
|
|
|
|
public void recordRemoteAddr(final String remoteAddr) {
|
|
// TODO API 단일화 필요.
|
|
StackFrame currentStackFrame = getCurrentStackFrame();
|
|
if (currentStackFrame instanceof RootStackFrame) {
|
|
Span span = ((RootStackFrame) currentStackFrame).getSpan();
|
|
span.setRemoteAddr(remoteAddr);
|
|
} else {
|
|
// do nothing.
|
|
}
|
|
}
|
|
|
|
public void recordNextSpanId(int spanId) {
|
|
StackFrame currentStackFrame = getCurrentStackFrame();
|
|
if (currentStackFrame instanceof RootStackFrame) {
|
|
logger.log(Level.WARNING, "OMG. Something's going wrong. Current stackframe is root Span. nextSpanId={}", spanId);
|
|
} else {
|
|
SpanEvent spanEvent = ((SubStackFrame) currentStackFrame).getSpanEvent();
|
|
spanEvent.setNextSpanId(spanId);
|
|
}
|
|
}
|
|
|
|
private void annotate(final AnnotationKey key) {
|
|
StackFrame currentStackFrame = getCurrentStackFrame();
|
|
if (currentStackFrame instanceof RootStackFrame) {
|
|
Span span = ((RootStackFrame) currentStackFrame).getSpan();
|
|
span.addAnnotation(new Annotation(key));
|
|
} else {
|
|
SpanEvent spanEvent = ((SubStackFrame) currentStackFrame).getSpanEvent();
|
|
spanEvent.addAnnotation(new Annotation(key));
|
|
}
|
|
|
|
}
|
|
|
|
public void setTraceContext(TraceContext traceContext) {
|
|
this.traceContext = traceContext;
|
|
}
|
|
|
|
|
|
|
|
} |