kazusa
/
skywalking-java
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

1.优化spanEntry.setSpan方式。

This commit is contained in:
wusheng 2016-01-18 10:56:39 +08:00
parent ab208b1de4
commit 1b6e6e95d7
3 changed files with 11 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/mapper/CallChainMapper.java
View File

@ -101,21 +101,11 @@ public class CallChainMapper extends TableMapper<Text, ChainInfo> {
Map<String, SpanEntry> spanEntryMap = new HashMap<String, SpanEntry>();
for (Span span : spanList) {
SpanEntry spanEntry = spanEntryMap.get(span.getParentLevel() + "." + span.getLevelId());
if (spanEntry != null && span.isReceiver()) {
if (span.isReceiver()) {
spanEntry.setServerSpan(span);
} else {
spanEntry.setClientSpan(span);
}
} else {
if (spanEntry == null) {
spanEntry = new SpanEntry();
if (span.isReceiver()) {
spanEntry.setServerSpan(span);
} else {
spanEntry.setClientSpan(span);
}
spanEntryMap.put(span.getParentLevel() + "." + span.getLevelId(), spanEntry);
}
spanEntry.setSpan(span);
}
return spanEntryMap;
}

14
skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/model/SpanEntry.java
View File

@ -132,13 +132,13 @@ public class SpanEntry {
public Span getServerSpan() {
return serverSpan;
}
public void setClientSpan(Span clientSpan) {
this.clientSpan = clientSpan;
}
public void setServerSpan(Span serverSpan) {
this.serverSpan = serverSpan;
public void setSpan(Span span){
if (span.isReceiver()) {
this.serverSpan = span;
}else{
this.clientSpan = span;
}
}
public String getSpanType() {

14
skywalking-analysis/src/test/java/com/ai/cloud/skywalking/analysis/mapper/CallChainMapperTest.java
View File

@ -108,21 +108,11 @@ public class CallChainMapperTest {
Map<String, SpanEntry> spanEntryMap = new HashMap<String, SpanEntry>();
for (Span span : spanList) {
SpanEntry spanEntry = spanEntryMap.get(span.getParentLevel() + "." + span.getLevelId());
if (spanEntry != null && span.isReceiver()) {
if (span.isReceiver()) {
spanEntry.setServerSpan(span);
} else {
spanEntry.setClientSpan(span);
}
} else {
if (spanEntry == null) {
spanEntry = new SpanEntry();
if (span.isReceiver()) {
spanEntry.setServerSpan(span);
} else {
spanEntry.setClientSpan(span);
}
spanEntryMap.put(span.getParentLevel() + "." + span.getLevelId(), spanEntry);
}
spanEntry.setSpan(span);
}
return spanEntryMap;
}