add abstract method getRpcSystem() (#2149)

This commit is contained in:
汤煜冬 2021-02-01 19:47:27 +08:00 committed by GitHub
parent 93b3a3b289
commit c523fa2eae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 14 additions and 2 deletions

View File

@ -13,4 +13,6 @@ public abstract class RpcClientTracer extends BaseTracer {
protected RpcClientTracer(Tracer tracer) {
super(tracer);
}
protected abstract String getRpcSystem();
}

View File

@ -23,9 +23,14 @@ public class GrpcClientTracer extends RpcClientTracer {
super(tracer);
}
@Override
protected String getRpcSystem() {
return "grpc";
}
public Span startSpan(String name) {
SpanBuilder spanBuilder = tracer.spanBuilder(name).setSpanKind(CLIENT);
spanBuilder.setAttribute(SemanticAttributes.RPC_SYSTEM, "grpc");
spanBuilder.setAttribute(SemanticAttributes.RPC_SYSTEM, getRpcSystem());
return spanBuilder.startSpan();
}

View File

@ -27,7 +27,7 @@ public class RmiClientTracer extends RpcClientTracer {
SpanBuilder spanBuilder =
tracer.spanBuilder(serviceName + "/" + methodName).setSpanKind(CLIENT);
spanBuilder.setAttribute(SemanticAttributes.RPC_SYSTEM, "java_rmi");
spanBuilder.setAttribute(SemanticAttributes.RPC_SYSTEM, getRpcSystem());
spanBuilder.setAttribute(SemanticAttributes.RPC_SERVICE, serviceName);
spanBuilder.setAttribute(SemanticAttributes.RPC_METHOD, methodName);
@ -38,4 +38,9 @@ public class RmiClientTracer extends RpcClientTracer {
protected String getInstrumentationName() {
return "io.opentelemetry.javaagent.rmi";
}
@Override
protected String getRpcSystem() {
return "java_rmi";
}
}