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

Add a flag to determine whether to bind data

This commit is contained in:
baiyang 2017-04-19 20:22:32 +08:00
parent e8084e548d
commit e42bf6e014
2 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
skywalking-sniffer/skywalking-api/src/main/java/com/a/eye/skywalking/api/conf/Config.java
View File

@ -23,6 +23,13 @@ public class Config {
* Zero and negative number are illegal.
*/
public static int SAMPLING_CYCLE = 1;
/**
* Set Mongodb plugin whether to bind params
* False=Not bound
* True=Binding
*/
public static boolean MONGODB_BINDPARAM = false;
}
public static class Collector {

12
skywalking-sniffer/skywalking-sdk-plugin/mongodb-3.x-plugin/src/main/java/com/a/eye/skywalking/plugin/mongodb/v3/MongoDBMethodInterceptor.java
View File

@ -4,6 +4,7 @@ import java.util.List;
import org.bson.BsonDocument;
import com.a.eye.skywalking.api.conf.Config;
import com.a.eye.skywalking.api.context.ContextManager;
import com.a.eye.skywalking.api.plugin.interceptor.EnhancedClassInstanceContext;
import com.a.eye.skywalking.api.plugin.interceptor.enhance.InstanceMethodInvokeContext;
@ -65,13 +66,18 @@ public class MongoDBMethodInterceptor implements InstanceMethodsAroundIntercepto
public void beforeMethod(final EnhancedClassInstanceContext context,
final InstanceMethodInvokeContext interceptorContext, final MethodInterceptResult result) {
Object[] arguments = interceptorContext.allArguments();
OperationInfo operationInfo = this.getReadOperationInfo(arguments[0]);
Span span = ContextManager.createSpan(METHOD + operationInfo.getMethodName());
Span span = null;
if (Config.Agent.MONGODB_BINDPARAM) {
OperationInfo operationInfo = this.getReadOperationInfo(arguments[0]);
span = ContextManager.createSpan(METHOD + operationInfo.getMethodName());
Tags.DB_STATEMENT.set(span, operationInfo.getMethodName() + " " + operationInfo.getFilter());
} else {
span = ContextManager.createSpan(METHOD + arguments[0].getClass().getSimpleName());
}
Tags.COMPONENT.set(span, MONGODB_COMPONENT);
Tags.DB_TYPE.set(span, MONGODB_COMPONENT);
Tags.SPAN_KIND.set(span, Tags.SPAN_KIND_CLIENT);
Tags.SPAN_LAYER.asDB(span);
Tags.DB_STATEMENT.set(span, operationInfo.getMethodName() + " " + operationInfo.getFilter());
}
@Override