pinpoint-dubbo插件兼容泛化调用

Posted 2020-10-14 胡大叔

背景

dubbo插件中需要记录当前调用的接口和方法，但是在泛化调用的场景下，记录的接口和方法都变成了

com.alibaba.dubbo.rpc.service.GenericService:$invoke，需要把它转化为真实的API和方法

@Override
    public void after(Object target, Object arg0, Object result, Throwable throwable) {
　　　　　//插件开关
        if(!traceContext.collectSwitch(traceContext.getApplicationName(),"dubbo",null)){
            return ;
        }
        Trace trace = this.getTrace(target);
        if (trace == null) {
            return;
        }

        RpcInvocation invocation = (RpcInvocation) arg0;

        try {
            SpanEventRecorder recorder = trace.currentSpanEventRecorder();

            String endPoint = RpcContext.getContext().getRemoteAddressString();
            recorder.recordApi(descriptor,new Object[]{endPoint});

            if (throwable == null) {
                // RPC client have to record end point (server address)
                recorder.recordEndPoint(endPoint);

                // Optionally, record the destination id (logical name of server. e.g. DB name)
                recorder.recordDestinationId(endPoint);

                URL url = invocation.getInvoker().getUrl();
                // 2017/9/11 上午11:22 记录调用的接口
                String interfaceName = ((Invoker)target).getInterface().getName();
                String methodName = invocation.getMethodName();
                //泛化调用兼容
                if (GENERIC_SERVICE.equals(interfaceName) && INVOKE.equals(methodName)) {
                    interfaceName = invocation.getAttachment("interface");
                    methodName = (String) invocation.getArguments()[0];
                }
                //记录调用的api
                recorder.recordAttribute(DubboConstants.DUBBO_INTERFACE_ANNOTATION_KEY,interfaceName+":"+methodName);
                //记录dubbo调用分组
                recorder.recordAttribute(DubboConstants.DUBBO_GROUP_ANNOTATION_KEY,url.getParameter("group"));
                //记录dubbo调用请求参数
                recorder.recordAttribute(DubboConstants.DUBBO_ARGS_ANNOTATION_KEY, DubboUtils.getParam(invocation.getArguments(),traceContext));

                if (result instanceof RpcResult && trace.isDebug()){
//                if (result instanceof RpcResult ){
                    RpcResult rpcResult = (RpcResult)result;
                    Object value = rpcResult.getValue();
                    recorder.recordAttribute(DubboConstants.DUBBO_RESULT_ANNOTATION_KEY, traceContext.toJsonString(value));
                }else {
//                    recorder.recordAttribute(DubboConstants.DUBBO_RESULT_ANNOTATION_KEY, result);
                }
            } else {
                recorder.recordException(throwable);
            }
        } finally {
            trace.traceBlockEnd();
        }
    }