diff --git a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/SpanEvent.java b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/SpanEvent.java index 115ef504af1a..bd52ddbb149e 100644 --- a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/SpanEvent.java +++ b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/SpanEvent.java @@ -52,6 +52,8 @@ public class SpanEvent extends DefaultFrameAttachment { private AsyncId asyncIdObject; + private boolean executeQueryType; + public SpanEvent() { } @@ -204,6 +206,14 @@ public AsyncId getAsyncIdObject() { return asyncIdObject; } + public void setExecuteQueryType(boolean executeQueryType) { + this.executeQueryType = executeQueryType; + } + + public boolean isExecuteQueryType() { + return executeQueryType; + } + @Override public String toString() { return "SpanEvent{" + @@ -221,6 +231,7 @@ public String toString() { ", destinationId='" + destinationId + '\'' + ", apiId=" + apiId + ", exceptionInfo=" + exceptionInfo + + ", executeQueryType=" + executeQueryType + "} "; } } diff --git a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/recorder/WrappedSpanEventRecorder.java b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/recorder/WrappedSpanEventRecorder.java index e7a22c1a84c4..f98c23d7c90f 100644 --- a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/recorder/WrappedSpanEventRecorder.java +++ b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/recorder/WrappedSpanEventRecorder.java @@ -58,8 +58,6 @@ public class WrappedSpanEventRecorder extends AbstractRecorder implements SpanEv private ExceptionContext exceptionContext; - private DatabaseInfo databaseInfo; - public WrappedSpanEventRecorder(TraceRoot traceRoot, AsyncContextFactory asyncContextFactory, StringMetaDataService stringMetaDataService, @@ -89,7 +87,6 @@ public WrappedSpanEventRecorder(TraceRoot traceRoot, public void setWrapped(final SpanEvent spanEvent, ExceptionContext exceptionContext) { this.spanEvent = spanEvent; this.exceptionContext = exceptionContext; - this.databaseInfo = null; } @Override @@ -116,14 +113,14 @@ public void recordSqlParsingResult(ParsingResult parsingResult, String bindValue Annotation sqlAnnotation = this.sqlMetaDataService.newSqlAnnotation(parsingResult, bindValue); spanEvent.addAnnotation(sqlAnnotation); - if (databaseInfo != null && spanEvent.getServiceType() == databaseInfo.getExecuteQueryType().getCode()) { + if (spanEvent.isExecuteQueryType()) { sqlCountService.recordSqlCount(this.traceRoot); } } @Override public void recordDatabaseInfo(DatabaseInfo databaseInfo, boolean executeQueryType) { - this.databaseInfo = databaseInfo; + this.spanEvent.setExecuteQueryType(executeQueryType); recordServiceType(executeQueryType ? databaseInfo.getExecuteQueryType() : databaseInfo.getType()); recordEndPoint(databaseInfo.getMultipleHost()); recordDestinationId(databaseInfo.getDatabaseId());