Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
润土的好友猹
SkyWalking
提交
1b499f0d
S
SkyWalking
项目概览
润土的好友猹
/
SkyWalking
与 Fork 源项目一致
Fork自
apache / SkyWalking
通知
9
Star
0
Fork
0
代码
文件
提交
分支
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,发现更多精彩内容 >>
提交
1b499f0d
编写于
1月 22, 2018
作者:
L
lytscu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add [Agent] ServiceComb Plugin #437
上级
6554aec1
变更
9
隐藏空白更改
内联
并排
Showing
9 changed file
with
695 addition
and
1 deletion
+695
-1
apm-protocol/apm-network/src/main/java/org/apache/skywalking/apm/network/trace/component/ComponentsDefine.java
...walking/apm/network/trace/component/ComponentsDefine.java
+4
-1
apm-sniffer/apm-sdk-plugin/pom.xml
apm-sniffer/apm-sdk-plugin/pom.xml
+1
-0
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/pom.xml
...er/apm-sdk-plugin/servicecomb-java-chassis-plugin/pom.xml
+81
-0
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/InvocationInterceptor.java
...walking/apm/plugin/servicecomb/InvocationInterceptor.java
+95
-0
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/NextInterceptor.java
...he/skywalking/apm/plugin/servicecomb/NextInterceptor.java
+117
-0
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/define/InvocationInstrumentation.java
.../plugin/servicecomb/define/InvocationInstrumentation.java
+92
-0
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/main/resources/skywalking-plugin.def
...a-chassis-plugin/src/main/resources/skywalking-plugin.def
+1
-0
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/InvocationInterceptorTest.java
...ing/apm/plugin/servicecomb/InvocationInterceptorTest.java
+150
-0
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/NextInterceptorTest.java
...kywalking/apm/plugin/servicecomb/NextInterceptorTest.java
+154
-0
未找到文件。
apm-protocol/apm-network/src/main/java/org/apache/skywalking/apm/network/trace/component/ComponentsDefine.java
浏览文件 @
1b499f0d
...
...
@@ -78,6 +78,8 @@ public class ComponentsDefine {
public
static
final
OfficialComponent
HTTP_ASYNC_CLIENT
=
new
OfficialComponent
(
26
,
"httpasyncclient"
);
public
static
final
OfficialComponent
SERVICECOMB
=
new
OfficialComponent
(
27
,
"ServiceComb"
);
private
static
ComponentsDefine
INSTANCE
=
new
ComponentsDefine
();
private
String
[]
components
;
...
...
@@ -87,7 +89,7 @@ public class ComponentsDefine {
}
public
ComponentsDefine
()
{
components
=
new
String
[
2
7
];
components
=
new
String
[
2
8
];
addComponent
(
TOMCAT
);
addComponent
(
HTTPCLIENT
);
addComponent
(
DUBBO
);
...
...
@@ -114,6 +116,7 @@ public class ComponentsDefine {
addComponent
(
ELASTIC_JOB
);
addComponent
(
ROCKET_MQ
);
addComponent
(
HTTP_ASYNC_CLIENT
);
addComponent
(
SERVICECOMB
);
}
private
void
addComponent
(
OfficialComponent
component
)
{
...
...
apm-sniffer/apm-sdk-plugin/pom.xml
浏览文件 @
1b499f0d
...
...
@@ -53,6 +53,7 @@
<module>
elastic-job-2.x-plugin
</module>
<module>
mongodb-2.x-plugin
</module>
<module>
httpasyncclient-4.x-plugin
</module>
<module>
servicecomb-java-chassis-plugin
</module>
</modules>
<packaging>
pom
</packaging>
...
...
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/pom.xml
0 → 100644
浏览文件 @
1b499f0d
<?xml version="1.0" encoding="UTF-8"?>
<project
xmlns=
"http://maven.apache.org/POM/4.0.0"
xmlns:xsi=
"http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation=
"http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"
>
<parent>
<artifactId>
apm-sdk-plugin
</artifactId>
<groupId>
org.apache.skywalking
</groupId>
<version>
5.0.0-alpha
</version>
</parent>
<modelVersion>
4.0.0
</modelVersion>
<artifactId>
apm-servicecomb-java-chassis-plugin
</artifactId>
<packaging>
jar
</packaging>
<name>
servicecomb-java-chassis-plugin
</name>
<url>
http://maven.apache.org
</url>
<dependencies>
<dependency>
<groupId>
io.servicecomb
</groupId>
<artifactId>
java-chassis-core
</artifactId>
<version>
0.5.0
</version>
<scope>
provided
</scope>
</dependency>
<dependency>
<groupId>
io.servicecomb
</groupId>
<artifactId>
swagger-invocation-core
</artifactId>
<version>
0.5.0
</version>
<scope>
provided
</scope>
</dependency>
<dependency>
<groupId>
io.servicecomb
</groupId>
<artifactId>
foundation-common
</artifactId>
<version>
0.5.0
</version>
<scope>
test
</scope>
</dependency>
<dependency>
<groupId>
io.servicecomb
</groupId>
<artifactId>
service-registry
</artifactId>
<version>
0.5.0
</version>
<scope>
test
</scope>
</dependency>
<dependency>
<groupId>
io.swagger
</groupId>
<artifactId>
swagger-core
</artifactId>
<version>
1.5.17
</version>
<scope>
test
</scope>
</dependency>
<dependency>
<groupId>
org.slf4j
</groupId>
<artifactId>
log4j-over-slf4j
</artifactId>
<version>
1.7.25
</version>
<scope>
test
</scope>
</dependency>
<dependency>
<groupId>
javax.ws.rs
</groupId>
<artifactId>
jsr311-api
</artifactId>
<version>
1.1.1
</version>
</dependency>
</dependencies>
<build>
<plugins>
<plugin>
<!-- 源码插件 -->
<groupId>
org.apache.maven.plugins
</groupId>
<artifactId>
maven-source-plugin
</artifactId>
<!-- 发布时自动将源码同时发布的配置 -->
<executions>
<execution>
<id>
attach-sources
</id>
<phase>
none
</phase>
<goals>
<goal>
jar
</goal>
</goals>
</execution>
</executions>
</plugin>
</plugins>
</build>
</project>
\ No newline at end of file
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/InvocationInterceptor.java
0 → 100644
浏览文件 @
1b499f0d
/*
* 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
;
import
java.lang.reflect.Method
;
import
javax.ws.rs.core.Response.StatusType
;
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
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
;
import
org.apache.skywalking.apm.agent.core.context.tag.Tags
;
import
org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan
;
import
org.apache.skywalking.apm.agent.core.context.trace.SpanLayer
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.InstanceMethodsAroundInterceptor
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.MethodInterceptResult
;
import
org.apache.skywalking.apm.network.trace.component.ComponentsDefine
;
/**
* {@link InvocationInterceptor} define how to enhance class {@link Invocation#getHandlerChain()}.
*
* @author lytscu
*/
public
class
InvocationInterceptor
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
.
getOperationName
();
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
)
{
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
();
}
return
ret
;
}
@Override
public
void
handleMethodException
(
EnhancedInstance
objInst
,
Method
method
,
Object
[]
allArguments
,
Class
<?>[]
argumentsTypes
,
Throwable
t
)
{
AbstractSpan
span
=
ContextManager
.
activeSpan
();
span
.
errorOccurred
();
span
.
log
(
t
);
}
}
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/NextInterceptor.java
0 → 100644
浏览文件 @
1b499f0d
/*
* 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
;
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
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
;
import
org.apache.skywalking.apm.agent.core.context.tag.Tags
;
import
org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan
;
import
org.apache.skywalking.apm.agent.core.context.trace.SpanLayer
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.InstanceMethodsAroundInterceptor
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.MethodInterceptResult
;
import
org.apache.skywalking.apm.network.trace.component.ComponentsDefine
;
/**
* {@link NextInterceptor} define how to enhance class {@link Invocation#next(io.servicecomb.swagger.invocation.AsyncResponse)}.
*
* @author lytscu
*/
public
class
NextInterceptor
implements
InstanceMethodsAroundInterceptor
{
static
final
ThreadLocal
DEEP
=
new
ThreadLocal
()
{
@Override
protected
Integer
initialValue
()
{
Integer
deepindex
=
0
;
return
deepindex
;
}
};
@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
{
if
(
count
==
2
)
{
String
peer
=
invocation
.
getEndpoint
().
getAddress
().
toString
();
final
ContextCarrier
contextCarrier
=
new
ContextCarrier
();
span
=
ContextManager
.
createExitSpan
(
invocation
.
getOperationName
(),
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
);
}
}
}
@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
);
}
}
return
ret
;
}
@Override
public
void
handleMethodException
(
EnhancedInstance
objInst
,
Method
method
,
Object
[]
allArguments
,
Class
<?>[]
argumentsTypes
,
Throwable
t
)
{
AbstractSpan
span
=
ContextManager
.
activeSpan
();
span
.
errorOccurred
();
span
.
log
(
t
);
}
}
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/main/java/org/apache/skywalking/apm/plugin/servicecomb/define/InvocationInstrumentation.java
0 → 100644
浏览文件 @
1b499f0d
/*
* 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
io.servicecomb.core.Invocation
;
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
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#next(AsyncResponse)}by
* using {@link InvocationInterceptor}, and {@link Invocation#getHandlerChain() by using {@link NextInterceptor},
*
* @author lytscu
*/
public
class
InvocationInstrumentation
extends
ClassInstanceMethodsEnhancePluginDefine
{
private
static
final
String
ENHANCE_CLASS
=
"io.servicecomb.core.Invocation"
;
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"
;
@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
(
"getHandlerChain"
);
}
@Override
public
String
getMethodsInterceptor
()
{
return
PRODUCER_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-java-chassis-plugin/src/main/resources/skywalking-plugin.def
0 → 100644
浏览文件 @
1b499f0d
servicecomb-0.6=org.apache.skywalking.apm.plugin.servicecomb.define.InvocationInstrumentation
apm-sniffer/apm-sdk-plugin/servicecomb-java-chassis-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/InvocationInterceptorTest.java
0 → 100644
浏览文件 @
1b499f0d
/*
* 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
;
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
;
import
javax.ws.rs.core.Response
;
import
org.apache.skywalking.apm.agent.core.boot.ServiceManager
;
import
org.apache.skywalking.apm.agent.core.conf.Config
;
import
org.apache.skywalking.apm.agent.core.context.trace.AbstractTracingSpan
;
import
org.apache.skywalking.apm.agent.core.context.trace.TraceSegment
;
import
org.apache.skywalking.apm.agent.core.context.util.KeyValuePair
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance
;
import
org.apache.skywalking.apm.agent.test.helper.SegmentHelper
;
import
org.apache.skywalking.apm.agent.test.helper.SpanHelper
;
import
org.apache.skywalking.apm.agent.test.tools.AgentServiceRule
;
import
org.apache.skywalking.apm.agent.test.tools.SegmentStorage
;
import
org.apache.skywalking.apm.agent.test.tools.SegmentStoragePoint
;
import
org.apache.skywalking.apm.agent.test.tools.TracingSegmentRunner
;
import
org.junit.Assert
;
import
org.junit.Before
;
import
org.junit.Rule
;
import
org.junit.Test
;
import
org.junit.runner.RunWith
;
import
org.mockito.Mock
;
import
org.powermock.api.mockito.PowerMockito
;
import
org.powermock.modules.junit4.PowerMockRunner
;
import
org.powermock.modules.junit4.PowerMockRunnerDelegate
;
import
static
org
.
hamcrest
.
CoreMatchers
.
is
;
import
static
org
.
hamcrest
.
MatcherAssert
.
assertThat
;
import
static
org
.
mockito
.
Mockito
.
times
;
import
static
org
.
mockito
.
Mockito
.
verify
;
import
static
org
.
mockito
.
Mockito
.
when
;
@RunWith
(
PowerMockRunner
.
class
)
@PowerMockRunnerDelegate
(
TracingSegmentRunner
.
class
)
public
class
InvocationInterceptorTest
{
@SegmentStoragePoint
private
SegmentStorage
segmentStorage
;
@Rule
public
AgentServiceRule
agentServiceRule
=
new
AgentServiceRule
();
private
InvocationInterceptor
invocationInterceptor
;
@Mock
private
OperationMeta
operationMeta
;
@Mock
private
MockInvocation
enhancedInstance
;
@Mock
private
Endpoint
endpoint
;
@Mock
Response
.
StatusType
statusType
;
@Mock
ReferenceConfig
referenceConfig
;
@Mock
private
SwaggerInvocation
swagger
;
private
Object
[]
allArguments
;
private
Class
[]
argumentsType
;
private
Object
[]
swaggerArguments
;
@Mock
private
SchemaMeta
schemaMeta
;
@Before
public
void
setUp
()
throws
Exception
{
ServiceManager
.
INSTANCE
.
boot
();
invocationInterceptor
=
new
InvocationInterceptor
();
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
.
getOperationName
()).
thenReturn
(
"productorTest"
);
when
(
operationMeta
.
getOperationPath
()).
thenReturn
(
"/bmi"
);
when
(
enhancedInstance
.
getOperationMeta
()).
thenReturn
(
operationMeta
);
when
(
enhancedInstance
.
getStatus
()).
thenReturn
(
statusType
);
when
(
statusType
.
getStatusCode
()).
thenReturn
(
200
);
when
(
enhancedInstance
.
getInvocationType
()).
thenReturn
(
InvocationType
.
PRODUCER
);
Config
.
Agent
.
APPLICATION_CODE
=
"serviceComnTestCases-APP"
;
allArguments
=
new
Object
[]
{};
argumentsType
=
new
Class
[]
{};
swaggerArguments
=
new
Class
[]
{};
}
@Test
public
void
testConsumer
()
throws
Throwable
{
invocationInterceptor
.
beforeMethod
(
enhancedInstance
,
null
,
allArguments
,
argumentsType
,
null
);
invocationInterceptor
.
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
();
}
private
void
assertCombSpan
(
AbstractTracingSpan
span
)
{
assertThat
(
span
.
getOperationName
(),
is
(
"productorTest"
));
assertThat
(
SpanHelper
.
getComponentId
(
span
),
is
(
27
));
List
<
KeyValuePair
>
tags
=
SpanHelper
.
getTags
(
span
);
assertThat
(
tags
.
get
(
0
).
getValue
(),
is
(
"/bmi"
));
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-java-chassis-plugin/src/test/java/org/apache/skywalking/apm/plugin/servicecomb/NextInterceptorTest.java
0 → 100644
浏览文件 @
1b499f0d
/*
* 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
;
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
;
import
javax.ws.rs.core.Response
;
import
org.apache.skywalking.apm.agent.core.boot.ServiceManager
;
import
org.apache.skywalking.apm.agent.core.conf.Config
;
import
org.apache.skywalking.apm.agent.core.context.trace.AbstractTracingSpan
;
import
org.apache.skywalking.apm.agent.core.context.trace.TraceSegment
;
import
org.apache.skywalking.apm.agent.core.context.util.KeyValuePair
;
import
org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance
;
import
org.apache.skywalking.apm.agent.test.helper.SegmentHelper
;
import
org.apache.skywalking.apm.agent.test.helper.SpanHelper
;
import
org.apache.skywalking.apm.agent.test.tools.AgentServiceRule
;
import
org.apache.skywalking.apm.agent.test.tools.SegmentStorage
;
import
org.apache.skywalking.apm.agent.test.tools.SegmentStoragePoint
;
import
org.apache.skywalking.apm.agent.test.tools.TracingSegmentRunner
;
import
org.junit.Assert
;
import
org.junit.Before
;
import
org.junit.Rule
;
import
org.junit.Test
;
import
org.junit.runner.RunWith
;
import
org.mockito.Mock
;
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
;
import
static
org
.
mockito
.
Mockito
.
verify
;
import
static
org
.
mockito
.
Mockito
.
when
;
@RunWith
(
PowerMockRunner
.
class
)
@PowerMockRunnerDelegate
(
TracingSegmentRunner
.
class
)
public
class
NextInterceptorTest
{
@SegmentStoragePoint
private
SegmentStorage
segmentStorage
;
@Rule
public
AgentServiceRule
agentServiceRule
=
new
AgentServiceRule
();
private
NextInterceptor
nextInterceptor
;
@Mock
private
OperationMeta
operationMeta
;
@Mock
private
MockInvocation
enhancedInstance
;
@Mock
private
Endpoint
endpoint
;
@Mock
Response
.
StatusType
statusType
;
@Mock
ReferenceConfig
referenceConfig
;
@Mock
private
SwaggerInvocation
swagger
;
private
Object
[]
allArguments
;
private
Class
[]
argumentsType
;
private
Object
[]
swaggerArguments
;
@Mock
private
SchemaMeta
schemaMeta
;
@Before
public
void
setUp
()
throws
Exception
{
ServiceManager
.
INSTANCE
.
boot
();
nextInterceptor
=
new
NextInterceptor
();
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
.
getOperationName
()).
thenReturn
(
"consumerTest"
);
when
(
operationMeta
.
getOperationPath
()).
thenReturn
(
"/bmi"
);
when
(
enhancedInstance
.
getOperationMeta
()).
thenReturn
(
operationMeta
);
when
(
enhancedInstance
.
getStatus
()).
thenReturn
(
statusType
);
when
(
statusType
.
getStatusCode
()).
thenReturn
(
200
);
when
(
enhancedInstance
.
getInvocationType
()).
thenReturn
(
InvocationType
.
CONSUMER
);
Config
.
Agent
.
APPLICATION_CODE
=
"serviceComnTestCases-APP"
;
allArguments
=
new
Object
[]
{};
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
();
}
private
void
assertCombSpan
(
AbstractTracingSpan
span
)
{
assertThat
(
span
.
getOperationName
(),
is
(
"consumerTest"
));
assertThat
(
SpanHelper
.
getComponentId
(
span
),
is
(
27
));
List
<
KeyValuePair
>
tags
=
SpanHelper
.
getTags
(
span
);
assertThat
(
tags
.
get
(
0
).
getValue
(),
is
(
"/bmi"
));
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.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录