From 039aa743f13bd09afd194ceee0c234f486f3f78c Mon Sep 17 00:00:00 2001
From: zhangxin10 <curtain.xin@gmail.com>
Date: Wed, 18 Nov 2015 19:02:00 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E6=88=90=E5=9F=8B=E7=82=B9Filter?=
 =?UTF-8?q?=EF=BC=8C=E5=BE=85=E6=B5=8B=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../cloud/skywalking/model/ContextData.java   |  4 +
 .../plugin/dubbo/SWDubboEnhanceFilter.java    | 82 +++++++++++++++++++
 2 files changed, 86 insertions(+)
 create mode 100644 skywalking-sdk-plugin/dubbo-plugin/src/main/java/com/ai/cloud/skywalking/plugin/dubbo/SWDubboEnhanceFilter.java

diff --git a/skywalking-api/src/main/java/com/ai/cloud/skywalking/model/ContextData.java b/skywalking-api/src/main/java/com/ai/cloud/skywalking/model/ContextData.java
index b3885ec00..c03bfab3b 100644
--- a/skywalking-api/src/main/java/com/ai/cloud/skywalking/model/ContextData.java
+++ b/skywalking-api/src/main/java/com/ai/cloud/skywalking/model/ContextData.java
@@ -35,4 +35,8 @@ public class ContextData {
         return spanType;
     }
 
+    @Override
+    public String toString() {
+        return traceId + "-" + parentLevel + "-" + levelId + "-" + spanType;
+    }
 }
diff --git a/skywalking-sdk-plugin/dubbo-plugin/src/main/java/com/ai/cloud/skywalking/plugin/dubbo/SWDubboEnhanceFilter.java b/skywalking-sdk-plugin/dubbo-plugin/src/main/java/com/ai/cloud/skywalking/plugin/dubbo/SWDubboEnhanceFilter.java
new file mode 100644
index 000000000..3bcbee943
--- /dev/null
+++ b/skywalking-sdk-plugin/dubbo-plugin/src/main/java/com/ai/cloud/skywalking/plugin/dubbo/SWDubboEnhanceFilter.java
@@ -0,0 +1,82 @@
+package com.ai.cloud.skywalking.plugin.dubbo;
+
+import com.ai.cloud.skywalking.buriedpoint.RPCBuriedPointReceiver;
+import com.ai.cloud.skywalking.buriedpoint.RPCBuriedPointSender;
+import com.ai.cloud.skywalking.context.Span;
+import com.ai.cloud.skywalking.model.ContextData;
+import com.ai.cloud.skywalking.model.Identification;
+import com.alibaba.dubbo.rpc.*;
+
+public class SWDubboEnhanceFilter implements Filter {
+
+    public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
+        RpcContext context = RpcContext.getContext();
+        boolean isConsumer = context.isConsumerSide();
+        Result result = null;
+        if (isConsumer) {
+            RPCBuriedPointSender sender = new RPCBuriedPointSender();
+            ContextData contextData = sender.beforeSend(createIdentification(invoker));
+            // 追加参数
+            RpcInvocation rpcInvocation = (RpcInvocation) invocation;
+            rpcInvocation.setAttachment("contextData", contextData.toString());
+            try {
+                //执行结果
+                result = invoker.invoke(invocation);
+                //结果是否包含异常
+                if (result.getException() != null) {
+                    sender.handleException(result.getException());
+                }
+            } catch (RpcException e) {
+                // 自身异常
+                sender.handleException(e);
+                throw e;
+            } finally {
+                sender.afterSend();
+            }
+        } else {
+            // 读取参数
+            RPCBuriedPointReceiver rpcBuriedPointReceiver = new RPCBuriedPointReceiver();
+            RpcInvocation rpcInvocation = (RpcInvocation) invocation;
+            String contextDataStr = rpcInvocation.getAttachment("contextData");
+            ContextData contextData = null;
+            if (contextDataStr != null && contextDataStr.length() > 0) {
+                // 反序列化参数
+                String[] value = contextDataStr.split("-");
+                Span span = new Span(value[0]);
+                span.setParentLevel(value[1]);
+                span.setLevelId(Integer.valueOf(value[2]));
+                span.setSpanType(value[3].charAt(0));
+                contextData = new ContextData(span);
+            }
+
+            rpcBuriedPointReceiver.beforeReceived(contextData, createIdentification(invoker));
+
+            try {
+                //执行结果
+                result = invoker.invoke(invocation);
+                //结果是否包含异常
+                if (result.getException() != null) {
+                    rpcBuriedPointReceiver.handleException(result.getException());
+                }
+            } catch (RpcException e) {
+                // 自身异常
+                rpcBuriedPointReceiver.handleException(e);
+                throw e;
+            } finally {
+                rpcBuriedPointReceiver.afterReceived();
+            }
+        }
+
+        return result;
+    }
+
+    private static Identification createIdentification(Invoker<?> invoker) {
+        StringBuilder businessKey = new StringBuilder();
+        businessKey.append("IP:" + invoker.getUrl().getAddress());
+        businessKey.append("Host:" + invoker.getUrl().getHost());
+        businessKey.append("Port:" + invoker.getUrl().getPort());
+        businessKey.append("Protocol:" + invoker.getUrl().getProtocol());
+        return Identification.newBuilder().viewPoint(invoker.getUrl().getServiceInterface()).businessKey(businessKey.
+                toString()).spanType('D').build();
+    }
+}