kazusa
/
skywalking-java
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Modify afterMethod logic

This commit is contained in:
lytscu 2017-11-13 19:59:20 +08:00
parent c53c4e4b02
commit 213fa33bd4
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
apm-sniffer/apm-sdk-plugin/mongodb-2.x-plugin/src/main/java/org/skywalking/apm/plugin/mongodb/v2/MongoDBV2MethodInterceptor.java
View File

@ -62,21 +62,18 @@ public class MongoDBV2MethodInterceptor implements InstanceMethodsAroundIntercep
@Override public Object afterMethod(EnhancedInstance objInst, Method method, Object[] allArguments,
Class<?>[] argumentsTypes, Object ret) throws Throwable {
AbstractSpan activeSpan = ContextManager.activeSpan();
CommandResult cresult = null;
if (ret instanceof WriteResult) {
WriteResult wresult = (WriteResult)ret;
if (!wresult.getCachedLastError().ok()) {
Tags.STATUS_CODE.set(activeSpan, "failed");
}
cresult = wresult.getCachedLastError();
} else if (ret instanceof CommandResult) {
CommandResult cresult = (CommandResult)ret;
if (!cresult.ok()) {
Tags.STATUS_CODE.set(activeSpan, "failed");
}
cresult = (CommandResult)ret;
} else if (ret instanceof AggregationOutput) {
AggregationOutput aresult = (AggregationOutput)ret;
if (!aresult.getCommandResult().ok()) {
Tags.STATUS_CODE.set(activeSpan, "failed");
}
cresult = aresult.getCommandResult();
}
if (null != cresult && !cresult.ok()) {
activeSpan.tag("CommandError", cresult.getErrorMessage());
}
ContextManager.stopSpan();
return ret;