Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
山不在高_有仙则灵
SkyWalking
提交
5fce5dcd
S
SkyWalking
项目概览
山不在高_有仙则灵
/
SkyWalking
与 Fork 源项目一致
Fork自
apache / SkyWalking
通知
12
Star
0
Fork
2
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
S
SkyWalking
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
5fce5dcd
编写于
2月 05, 2018
作者:
L
lytscu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update servicecomb-java-chassis-0.x-plugin
上级
48f8fcee
变更
8
隐藏空白更改
内联
并排
Showing
8 changed file
with
222 addition
and
148 deletion
+222
-148
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/pom.xml
...cecomb-plugin/servicecomb-java-chassis-0.x-plugin/pom.xml
+6
-0
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/ProducerOperationHandlerInterceptor.java
...ugin/servicecomb/ProducerOperationHandlerInterceptor.java
+72
-0
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/TransportClientHandlerInterceptor.java
...plugin/servicecomb/TransportClientHandlerInterceptor.java
+24
-37
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/define/ProducerOperationHandlerInstrumentation.java
...ecomb/define/ProducerOperationHandlerInstrumentation.java
+74
-0
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/define/TransportClientHandlerInstrumentation.java
...icecomb/define/TransportClientHandlerInstrumentation.java
+8
-26
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/resources/skywalking-plugin.def
...assis-0.x-plugin/src/main/resources/skywalking-plugin.def
+2
-1
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/ProducerOperationHandlerInterceptorTest.java
.../servicecomb/ProducerOperationHandlerInterceptorTest.java
+17
-39
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/TransportClientHandlerInterceptorTest.java
...in/servicecomb/TransportClientHandlerInterceptorTest.java
+19
-45
未找到文件。
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/pom.xml
浏览文件 @
5fce5dcd
...
...
@@ -41,6 +41,12 @@
<version>
0.5.0
</version>
<scope>
test
</scope>
</dependency>
<dependency>
<groupId>
io.servicecomb
</groupId>
<artifactId>
common-rest
</artifactId>
<version>
0.5.0
</version>
<scope>
provided
</scope>
</dependency>
<dependency>
<groupId>
io.swagger
</groupId>
<artifactId>
swagger-core
</artifactId>
...
...
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/
Next
Interceptor.java
→
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/
ProducerOperationHandler
Interceptor.java
浏览文件 @
5fce5dcd
...
...
@@ -19,12 +19,7 @@
package
org.apache.skywalking.apm.plugin.servicecomb
;
import
io.servicecomb.core.Invocation
;
import
io.servicecomb.swagger.invocation.InvocationType
;
import
io.servicecomb.swagger.invocation.SwaggerInvocation
;
import
io.servicecomb.swagger.invocation.context.InvocationContext
;
import
java.lang.reflect.Method
;
import
java.net.URI
;
import
javax.ws.rs.core.Response.StatusType
;
import
org.apache.skywalking.apm.agent.core.context.CarrierItem
;
import
org.apache.skywalking.apm.agent.core.context.ContextCarrier
;
import
org.apache.skywalking.apm.agent.core.context.ContextManager
;
...
...
@@ -37,76 +32,33 @@ import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.MethodInt
import
org.apache.skywalking.apm.network.trace.component.ComponentsDefine
;
/**
* {
@link NextInterceptor} define how to enhance class {@link Invocation#next(io.servicecomb.swagger.invocation.AsyncResponse
)}.
* {
org.apache.skywalking.apm.plugin.servicecomb.} define how to enhance class {@link Invocation#getHandlerContext(
)}.
*
* @author lytscu
*/
public
class
NextInterceptor
implements
InstanceMethodsAroundInterceptor
{
static
final
ThreadLocal
DEEP
=
new
ThreadLocal
()
{
@Override
protected
Integer
initialValue
()
{
Integer
deepindex
=
0
;
return
deepindex
;
}
};
public
class
ProducerOperationHandlerInterceptor
implements
InstanceMethodsAroundInterceptor
{
@Override
public
void
beforeMethod
(
EnhancedInstance
objInst
,
Method
method
,
Object
[]
allArguments
,
Class
<?>[]
argumentsTypes
,
MethodInterceptResult
result
)
throws
Throwable
{
SwaggerInvocation
swagger
=
(
SwaggerInvocation
)
objInst
;
InvocationType
type
=
swagger
.
getInvocationType
();
Invocation
invocation
=
(
Invocation
)
objInst
;
AbstractSpan
span
;
boolean
isConsumer
=
type
.
equals
(
InvocationType
.
CONSUMER
);
if
(
isConsumer
)
{
Integer
count
=
(
Integer
)
DEEP
.
get
();
try
{
//When count = 2, you can get the peer
if
(
count
==
2
)
{
URI
uri
=
new
URI
(
invocation
.
getEndpoint
().
toString
());
String
peer
=
uri
.
getHost
()
+
":"
+
uri
.
getPort
();
final
ContextCarrier
contextCarrier
=
new
ContextCarrier
();
span
=
ContextManager
.
createExitSpan
(
invocation
.
getMicroserviceQualifiedName
(),
contextCarrier
,
peer
);
CarrierItem
next
=
contextCarrier
.
items
();
while
(
next
.
hasNext
())
{
next
=
next
.
next
();
invocation
.
getContext
().
put
(
next
.
getHeadKey
(),
next
.
getHeadValue
());
}
String
url
=
invocation
.
getOperationMeta
().
getOperationPath
();
Tags
.
URL
.
set
(
span
,
url
);
span
.
setComponent
(
ComponentsDefine
.
SERVICECOMB
);
SpanLayer
.
asRPCFramework
(
span
);
}
}
finally
{
count
++;
DEEP
.
set
(
count
);
}
Invocation
invocation
=
(
Invocation
)
allArguments
[
0
];
ContextCarrier
contextCarrier
=
new
ContextCarrier
();
CarrierItem
next
=
contextCarrier
.
items
();
while
(
next
.
hasNext
())
{
next
=
next
.
next
();
next
.
setHeadValue
(
invocation
.
getContext
().
get
(
next
.
getHeadKey
()));
}
String
operationName
=
invocation
.
getMicroserviceQualifiedName
();
AbstractSpan
span
=
ContextManager
.
createEntrySpan
(
operationName
,
contextCarrier
);
String
url
=
invocation
.
getOperationMeta
().
getOperationPath
();
Tags
.
URL
.
set
(
span
,
url
);
span
.
setComponent
(
ComponentsDefine
.
SERVICECOMB
);
SpanLayer
.
asRPCFramework
(
span
);
}
@Override
public
Object
afterMethod
(
EnhancedInstance
objInst
,
Method
method
,
Object
[]
allArguments
,
Class
<?>[]
argumentsTypes
,
Object
ret
)
throws
Throwable
{
SwaggerInvocation
swagger
=
(
SwaggerInvocation
)
objInst
;
InvocationType
type
=
swagger
.
getInvocationType
();
boolean
isConsumer
=
type
.
equals
(
InvocationType
.
CONSUMER
);
if
(
isConsumer
)
{
Integer
count
=
(
Integer
)
DEEP
.
get
();
try
{
if
(
count
==
1
)
{
AbstractSpan
span
=
ContextManager
.
activeSpan
();
StatusType
statusType
=
((
InvocationContext
)
objInst
).
getStatus
();
int
statusCode
=
statusType
.
getStatusCode
();
if
(
statusCode
>=
400
)
{
span
.
errorOccurred
();
Tags
.
STATUS_CODE
.
set
(
span
,
Integer
.
toString
(
statusCode
));
}
ContextManager
.
stopSpan
();
}
}
finally
{
count
--;
DEEP
.
set
(
count
);
}
}
ContextManager
.
stopSpan
();
return
ret
;
}
...
...
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/
Invocation
Interceptor.java
→
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/
TransportClientHandler
Interceptor.java
浏览文件 @
5fce5dcd
...
...
@@ -19,11 +19,8 @@
package
org.apache.skywalking.apm.plugin.servicecomb
;
import
io.servicecomb.core.Invocation
;
import
io.servicecomb.swagger.invocation.InvocationType
;
import
io.servicecomb.swagger.invocation.SwaggerInvocation
;
import
io.servicecomb.swagger.invocation.context.InvocationContext
;
import
java.lang.reflect.Method
;
import
java
x.ws.rs.core.Response.StatusType
;
import
java
.net.URI
;
import
org.apache.skywalking.apm.agent.core.context.CarrierItem
;
import
org.apache.skywalking.apm.agent.core.context.ContextCarrier
;
import
org.apache.skywalking.apm.agent.core.context.ContextManager
;
...
...
@@ -36,52 +33,42 @@ import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.MethodInt
import
org.apache.skywalking.apm.network.trace.component.ComponentsDefine
;
/**
* {@link
InvocationInterceptor} define how to enhance class {@link Invocation#getHandlerContext(
)}.
* {@link
TransportClientHandlerInterceptor} define how to enhance class {@link Invocation#next(io.servicecomb.swagger.invocation.AsyncResponse
)}.
*
* @author lytscu
*/
public
class
Invocation
Interceptor
implements
InstanceMethodsAroundInterceptor
{
public
class
TransportClientHandler
Interceptor
implements
InstanceMethodsAroundInterceptor
{
@Override
public
void
beforeMethod
(
EnhancedInstance
objInst
,
Method
method
,
Object
[]
allArguments
,
Class
<?>[]
argumentsTypes
,
MethodInterceptResult
result
)
throws
Throwable
{
SwaggerInvocation
swagger
=
(
SwaggerInvocation
)
objInst
;
InvocationType
type
=
swagger
.
getInvocationType
();
Invocation
invocation
=
(
Invocation
)
objInst
;
AbstractSpan
span
;
boolean
isConsumer
=
type
.
equals
(
InvocationType
.
CONSUMER
);
if
(!
isConsumer
)
{
ContextCarrier
contextCarrier
=
new
ContextCarrier
();
CarrierItem
next
=
contextCarrier
.
items
();
while
(
next
.
hasNext
())
{
next
=
next
.
next
();
next
.
setHeadValue
(
invocation
.
getContext
().
get
(
next
.
getHeadKey
()));
}
String
operationName
=
invocation
.
getMicroserviceQualifiedName
();
span
=
ContextManager
.
createEntrySpan
(
operationName
,
contextCarrier
);
String
url
=
invocation
.
getOperationMeta
().
getOperationPath
();
Tags
.
URL
.
set
(
span
,
url
);
span
.
setComponent
(
ComponentsDefine
.
SERVICECOMB
);
SpanLayer
.
asRPCFramework
(
span
);
Invocation
invocation
=
(
Invocation
)
allArguments
[
0
];
URI
uri
=
new
URI
(
invocation
.
getEndpoint
().
toString
());
String
peer
=
uri
.
getHost
()
+
":"
+
uri
.
getPort
();
final
ContextCarrier
contextCarrier
=
new
ContextCarrier
();
AbstractSpan
span
=
ContextManager
.
createExitSpan
(
invocation
.
getMicroserviceQualifiedName
(),
contextCarrier
,
peer
);
CarrierItem
next
=
contextCarrier
.
items
();
while
(
next
.
hasNext
())
{
next
=
next
.
next
();
invocation
.
getContext
().
put
(
next
.
getHeadKey
(),
next
.
getHeadValue
());
}
String
url
=
invocation
.
getOperationMeta
().
getOperationPath
();
Tags
.
URL
.
set
(
span
,
url
);
span
.
setComponent
(
ComponentsDefine
.
SERVICECOMB
);
SpanLayer
.
asRPCFramework
(
span
);
}
@Override
public
Object
afterMethod
(
EnhancedInstance
objInst
,
Method
method
,
Object
[]
allArguments
,
Class
<?>[]
argumentsTypes
,
Object
ret
)
throws
Throwable
{
SwaggerInvocation
swagger
=
(
SwaggerInvocation
)
objInst
;
InvocationType
type
=
swagger
.
getInvocationType
();
boolean
isConsumer
=
type
.
equals
(
InvocationType
.
CONSUMER
);
if
(!
isConsumer
)
{
AbstractSpan
span
=
ContextManager
.
activeSpan
();
StatusType
statusType
=
((
InvocationContext
)
objInst
).
getStatus
();
int
statusCode
=
statusType
.
getStatusCode
();
if
(
statusCode
>=
400
)
{
span
.
errorOccurred
();
Tags
.
STATUS_CODE
.
set
(
span
,
Integer
.
toString
(
statusCode
));
}
ContextManager
.
stopSpan
();
Invocation
invocation
=
(
Invocation
)
allArguments
[
0
];
AbstractSpan
span
=
ContextManager
.
activeSpan
();
int
statusCode
=
invocation
.
getStatus
().
getStatusCode
();
if
(
statusCode
>=
400
)
{
span
.
errorOccurred
();
Tags
.
STATUS_CODE
.
set
(
span
,
Integer
.
toString
(
statusCode
));
}
ContextManager
.
stopSpan
();
return
ret
;
}
...
...
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/define/ProducerOperationHandlerInstrumentation.java
0 → 100644
浏览文件 @
5fce5dcd
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
package
org.apache.skywalking.apm.plugin.servicecomb.define
;
import
net.bytebuddy.description.method.MethodDescription
;
import
net.bytebuddy.matcher.ElementMatcher
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.ConstructorInterceptPoint
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.InstanceMethodsInterceptPoint
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.ClassInstanceMethodsEnhancePluginDefine
;
import
org.apache.skywalking.apm.agent.core.plugin.match.ClassMatch
;
import
org.apache.skywalking.apm.agent.core.plugin.match.NameMatch
;
import
static
net
.
bytebuddy
.
matcher
.
ElementMatchers
.
named
;
/**
* {@link ProducerOperationHandlerInstrumentation} presents that skywalking intercept {@link
* ProducerOperationHandler#handle (io.servicecomb.core.Invocation, io.servicecomb.swagger.invocation.AsyncResponse)#}by
* using {@link ProducerOperationHandlerInterceptor}
*
* @author lytscu
*/
public
class
ProducerOperationHandlerInstrumentation
extends
ClassInstanceMethodsEnhancePluginDefine
{
private
static
final
String
ENHANCE_CLASS
=
"io.servicecomb.core.handler.impl.ProducerOperationHandler"
;
private
static
final
String
INTERCEPT_CLASS
=
"org.apache.skywalking.apm.plugin.servicecomb.ProducerOperationHandlerInterceptor"
;
@Override
protected
ClassMatch
enhanceClass
()
{
return
NameMatch
.
byName
(
ENHANCE_CLASS
);
}
@Override
protected
ConstructorInterceptPoint
[]
getConstructorsInterceptPoints
()
{
return
new
ConstructorInterceptPoint
[
0
];
}
@Override
protected
InstanceMethodsInterceptPoint
[]
getInstanceMethodsInterceptPoints
()
{
return
new
InstanceMethodsInterceptPoint
[]
{
new
InstanceMethodsInterceptPoint
()
{
@Override
public
ElementMatcher
<
MethodDescription
>
getMethodsMatcher
()
{
return
named
(
"handle"
);
}
@Override
public
String
getMethodsInterceptor
()
{
return
INTERCEPT_CLASS
;
}
@Override
public
boolean
isOverrideArgs
()
{
return
false
;
}
}
};
}
}
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/define/
Invocation
Instrumentation.java
→
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/define/
TransportClientHandler
Instrumentation.java
浏览文件 @
5fce5dcd
...
...
@@ -18,7 +18,7 @@
package
org.apache.skywalking.apm.plugin.servicecomb.define
;
import
io.servicecomb.core.
Invocation
;
import
io.servicecomb.core.
handler.impl.TransportClientHandler
;
import
net.bytebuddy.description.method.MethodDescription
;
import
net.bytebuddy.matcher.ElementMatcher
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.ConstructorInterceptPoint
;
...
...
@@ -26,23 +26,20 @@ import org.apache.skywalking.apm.agent.core.plugin.interceptor.InstanceMethodsIn
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.ClassInstanceMethodsEnhancePluginDefine
;
import
org.apache.skywalking.apm.agent.core.plugin.match.ClassMatch
;
import
org.apache.skywalking.apm.agent.core.plugin.match.NameMatch
;
import
org.apache.skywalking.apm.plugin.servicecomb.InvocationInterceptor
;
import
static
net
.
bytebuddy
.
matcher
.
ElementMatchers
.
named
;
/**
* {@link
InvocationInstrumentation} presents that skywalking intercept {@link io.servicecomb.core.Invocation#getHandlerContext
}by
* using {@link
InvocationInterceptor}, and {@link Invocation#next by using {@link NextInterceptor},
* {@link
TransportClientHandlerInstrumentation} presents that skywalking intercept {@link TransportClientHandler
}by
* using {@link
TransportClientHandlerInterceptor }
*
* @author lytscu
*/
public
class
Invocation
Instrumentation
extends
ClassInstanceMethodsEnhancePluginDefine
{
public
class
TransportClientHandler
Instrumentation
extends
ClassInstanceMethodsEnhancePluginDefine
{
private
static
final
String
ENHANCE_CLASS
=
"io.servicecomb.core.
Invocation
"
;
private
static
final
String
ENHANCE_CLASS
=
"io.servicecomb.core.
handler.impl.TransportClientHandler
"
;
private
static
final
String
PRODUCER_INTERCEPT_CLASS
=
"org.apache.skywalking.apm.plugin.servicecomb.InvocationInterceptor"
;
private
static
final
String
CONSUMER_INTERCEPT_CLASS
=
"org.apache.skywalking.apm.plugin.servicecomb.NextInterceptor"
;
private
static
final
String
INTERCEPT_CLASS
=
"org.apache.skywalking.apm.plugin.servicecomb.TransportClientHandlerInterceptor"
;
@Override
protected
ClassMatch
enhanceClass
()
{
...
...
@@ -60,33 +57,18 @@ public class InvocationInstrumentation extends ClassInstanceMethodsEnhancePlugin
new
InstanceMethodsInterceptPoint
()
{
@Override
public
ElementMatcher
<
MethodDescription
>
getMethodsMatcher
()
{
return
named
(
"
getHandlerContext
"
);
return
named
(
"
handle
"
);
}
@Override
public
String
getMethodsInterceptor
()
{
return
PRODUCER_
INTERCEPT_CLASS
;
return
INTERCEPT_CLASS
;
}
@Override
public
boolean
isOverrideArgs
()
{
return
false
;
}
},
new
InstanceMethodsInterceptPoint
()
{
@Override
public
ElementMatcher
<
MethodDescription
>
getMethodsMatcher
()
{
return
named
(
"next"
);
}
@Override
public
String
getMethodsInterceptor
()
{
return
CONSUMER_INTERCEPT_CLASS
;
}
@Override
public
boolean
isOverrideArgs
()
{
return
false
;
}
}
};
}
}
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/main/resources/skywalking-plugin.def
浏览文件 @
5fce5dcd
servicecomb-0.x=org.apache.skywalking.apm.plugin.servicecomb.define.InvocationInstrumentation
servicecomb-0.x=org.apache.skywalking.apm.plugin.servicecomb.define.ProducerOperationHandlerInstrumentation
servicecomb-0.x=org.apache.skywalking.apm.plugin.servicecomb.define.TransportClientHandlerInstrumentation
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/
Invocation
InterceptorTest.java
→
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/
ProducerOperationHandler
InterceptorTest.java
浏览文件 @
5fce5dcd
...
...
@@ -22,7 +22,6 @@ import io.servicecomb.core.Endpoint;
import
io.servicecomb.core.Invocation
;
import
io.servicecomb.core.definition.OperationMeta
;
import
io.servicecomb.core.definition.SchemaMeta
;
import
io.servicecomb.core.provider.consumer.ReferenceConfig
;
import
io.servicecomb.swagger.invocation.InvocationType
;
import
io.servicecomb.swagger.invocation.SwaggerInvocation
;
import
java.util.List
;
...
...
@@ -57,30 +56,23 @@ import static org.mockito.Mockito.when;
@RunWith
(
PowerMockRunner
.
class
)
@PowerMockRunnerDelegate
(
TracingSegmentRunner
.
class
)
public
class
InvocationInterceptorTest
{
@SegmentStoragePoint
private
SegmentStorage
segmentStorage
;
public
class
ProducerOperationHandlerInterceptorTest
{
@Rule
public
AgentServiceRule
agentServiceRule
=
new
AgentServiceRule
();
private
InvocationInterceptor
invocationInterceptor
;
@Mock
private
OperationMeta
operationMeta
;
Response
.
StatusType
statusType
;
@SegmentStoragePoint
private
SegmentStorage
segmentStorage
;
private
ProducerOperationHandlerInterceptor
invocationInterceptor
;
@Mock
private
MockInvocation
enhancedInstance
;
private
OperationMeta
operationMeta
;
@Mock
private
Endpoint
endpoint
;
private
EnhancedInstance
enhancedInstance
;
@Mock
Response
.
StatusType
statusType
;
private
Invocation
invocation
;
@Mock
ReferenceConfig
referenceConfig
;
private
Endpoint
endpoint
;
@Mock
private
SwaggerInvocation
swagger
;
private
Object
[]
allArguments
;
...
...
@@ -93,20 +85,20 @@ public class InvocationInterceptorTest {
@Before
public
void
setUp
()
throws
Exception
{
ServiceManager
.
INSTANCE
.
boot
();
invocationInterceptor
=
new
Invocation
Interceptor
();
invocationInterceptor
=
new
ProducerOperationHandler
Interceptor
();
PowerMockito
.
mock
(
Invocation
.
class
);
when
(
operationMeta
.
getSchemaMeta
()).
thenReturn
(
schemaMeta
);
when
(
endpoint
.
getAddress
()).
thenReturn
(
"0.0.0.0:7777"
);
when
(
enhancedInstance
.
getEndpoint
()).
thenReturn
(
endpoint
);
when
(
enhancedInstance
.
getMicroserviceQualifiedName
()).
thenReturn
(
"productorTest"
);
when
(
invocation
.
getEndpoint
()).
thenReturn
(
endpoint
);
when
(
invocation
.
getMicroserviceQualifiedName
()).
thenReturn
(
"productorTest"
);
when
(
operationMeta
.
getOperationPath
()).
thenReturn
(
"/bmi"
);
when
(
enhancedInstance
.
getOperationMeta
()).
thenReturn
(
operationMeta
);
when
(
enhancedInstance
.
getStatus
()).
thenReturn
(
statusType
);
when
(
invocation
.
getOperationMeta
()).
thenReturn
(
operationMeta
);
when
(
invocation
.
getStatus
()).
thenReturn
(
statusType
);
when
(
statusType
.
getStatusCode
()).
thenReturn
(
200
);
when
(
enhancedInstance
.
getInvocationType
()).
thenReturn
(
InvocationType
.
PRODUCER
);
when
(
invocation
.
getInvocationType
()).
thenReturn
(
InvocationType
.
PRODUCER
);
Config
.
Agent
.
APPLICATION_CODE
=
"serviceComnTestCases-APP"
;
allArguments
=
new
Object
[]
{};
allArguments
=
new
Object
[]
{
invocation
,
};
argumentsType
=
new
Class
[]
{};
swaggerArguments
=
new
Class
[]
{};
}
...
...
@@ -122,7 +114,7 @@ public class InvocationInterceptorTest {
List
<
AbstractTracingSpan
>
spans
=
SegmentHelper
.
getSpans
(
traceSegment
);
assertCombSpan
(
spans
.
get
(
0
));
verify
(
enhancedInstance
,
times
(
1
)).
getContext
();
verify
(
invocation
,
times
(
1
)).
getContext
();
}
private
void
assertCombSpan
(
AbstractTracingSpan
span
)
{
...
...
@@ -133,18 +125,4 @@ public class InvocationInterceptorTest {
assertThat
(
span
.
isEntry
(),
is
(
true
));
}
private
class
MockInvocation
extends
Invocation
implements
EnhancedInstance
{
public
MockInvocation
(
ReferenceConfig
referenceConfig
,
OperationMeta
operationMeta
,
Object
[]
swaggerArguments
)
{
super
(
referenceConfig
,
operationMeta
,
swaggerArguments
);
}
@Override
public
Object
getSkyWalkingDynamicField
()
{
return
null
;
}
@Override
public
void
setSkyWalkingDynamicField
(
Object
value
)
{
}
}
}
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/
Next
InterceptorTest.java
→
apm-sniffer/apm-sdk-plugin/servicecomb-plugin/servicecomb-java-chassis-0.x-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/
TransportClientHandler
InterceptorTest.java
浏览文件 @
5fce5dcd
...
...
@@ -49,7 +49,6 @@ import org.powermock.api.mockito.PowerMockito;
import
org.powermock.modules.junit4.PowerMockRunner
;
import
org.powermock.modules.junit4.PowerMockRunnerDelegate
;
import
static
org
.
apache
.
skywalking
.
apm
.
plugin
.
servicecomb
.
NextInterceptor
.
DEEP
;
import
static
org
.
hamcrest
.
CoreMatchers
.
is
;
import
static
org
.
hamcrest
.
MatcherAssert
.
assertThat
;
import
static
org
.
mockito
.
Mockito
.
times
;
...
...
@@ -58,30 +57,25 @@ import static org.mockito.Mockito.when;
@RunWith
(
PowerMockRunner
.
class
)
@PowerMockRunnerDelegate
(
TracingSegmentRunner
.
class
)
public
class
NextInterceptorTest
{
@SegmentStoragePoint
private
SegmentStorage
segmentStorage
;
public
class
TransportClientHandlerInterceptorTest
{
@Rule
public
AgentServiceRule
agentServiceRule
=
new
AgentServiceRule
();
private
NextInterceptor
nextInterceptor
;
@Mock
private
OperationMeta
operationMeta
;
Response
.
StatusType
statusType
;
@Mock
private
MockInvocation
enhancedInstance
;
ReferenceConfig
referenceConfig
;
@SegmentStoragePoint
private
SegmentStorage
segmentStorage
;
private
TransportClientHandlerInterceptor
nextInterceptor
;
@Mock
private
Endpoint
endpoint
;
private
OperationMeta
operationMeta
;
@Mock
Response
.
StatusType
statusType
;
private
EnhancedInstance
enhancedInstance
;
@Mock
ReferenceConfig
referenceConfig
;
private
Invocation
invocation
;
@Mock
private
Endpoint
endpoint
;
@Mock
private
SwaggerInvocation
swagger
;
private
Object
[]
allArguments
;
...
...
@@ -94,39 +88,33 @@ public class NextInterceptorTest {
@Before
public
void
setUp
()
throws
Exception
{
ServiceManager
.
INSTANCE
.
boot
();
nextInterceptor
=
new
Next
Interceptor
();
nextInterceptor
=
new
TransportClientHandler
Interceptor
();
PowerMockito
.
mock
(
Invocation
.
class
);
when
(
operationMeta
.
getSchemaMeta
()).
thenReturn
(
schemaMeta
);
when
(
endpoint
.
getAddress
()).
thenReturn
(
"0.0.0.0:7777"
);
when
(
enhancedInstance
.
getEndpoint
()).
thenReturn
(
endpoint
);
when
(
enhancedInstance
.
getMicroserviceQualifiedName
()).
thenReturn
(
"consumerTest"
);
when
(
invocation
.
getEndpoint
()).
thenReturn
(
endpoint
);
when
(
invocation
.
getMicroserviceQualifiedName
()).
thenReturn
(
"consumerTest"
);
when
(
operationMeta
.
getOperationPath
()).
thenReturn
(
"/bmi"
);
when
(
enhancedInstance
.
getOperationMeta
()).
thenReturn
(
operationMeta
);
when
(
enhancedInstance
.
getStatus
()).
thenReturn
(
statusType
);
when
(
invocation
.
getOperationMeta
()).
thenReturn
(
operationMeta
);
when
(
invocation
.
getStatus
()).
thenReturn
(
statusType
);
when
(
statusType
.
getStatusCode
()).
thenReturn
(
200
);
when
(
enhancedInstance
.
getInvocationType
()).
thenReturn
(
InvocationType
.
CONSUMER
);
when
(
invocation
.
getInvocationType
()).
thenReturn
(
InvocationType
.
CONSUMER
);
Config
.
Agent
.
APPLICATION_CODE
=
"serviceComnTestCases-APP"
;
allArguments
=
new
Object
[]
{};
allArguments
=
new
Object
[]
{
invocation
,
};
argumentsType
=
new
Class
[]
{};
swaggerArguments
=
new
Class
[]
{};
}
@Test
public
void
testConsumer
()
throws
Throwable
{
Integer
count
=
2
;
DEEP
.
set
(
count
);
nextInterceptor
.
beforeMethod
(
enhancedInstance
,
null
,
allArguments
,
argumentsType
,
null
);
count
=
1
;
DEEP
.
set
(
count
);
nextInterceptor
.
afterMethod
(
enhancedInstance
,
null
,
allArguments
,
argumentsType
,
null
);
Assert
.
assertThat
(
segmentStorage
.
getTraceSegments
().
size
(),
is
(
1
));
TraceSegment
traceSegment
=
segmentStorage
.
getTraceSegments
().
get
(
0
);
List
<
AbstractTracingSpan
>
spans
=
SegmentHelper
.
getSpans
(
traceSegment
);
assertCombSpan
(
spans
.
get
(
0
));
verify
(
enhancedInstance
,
times
(
1
)).
getContext
();
verify
(
invocation
,
times
(
1
)).
getContext
();
}
private
void
assertCombSpan
(
AbstractTracingSpan
span
)
{
...
...
@@ -137,18 +125,4 @@ public class NextInterceptorTest {
assertThat
(
span
.
isExit
(),
is
(
true
));
}
private
class
MockInvocation
extends
Invocation
implements
EnhancedInstance
{
public
MockInvocation
(
ReferenceConfig
referenceConfig
,
OperationMeta
operationMeta
,
Object
[]
swaggerArguments
)
{
super
(
referenceConfig
,
operationMeta
,
swaggerArguments
);
}
@Override
public
Object
getSkyWalkingDynamicField
()
{
return
null
;
}
@Override
public
void
setSkyWalkingDynamicField
(
Object
value
)
{
}
}
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录