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

1.提交部分TraceSpanTree的序列化代码。解决引用序列化问题。

This commit is contained in:
wusheng 2016-03-02 11:10:36 +08:00
parent 1b60750ea9
commit 06445fc592
4 changed files with 105 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/chainbuild/entity/TraceSpanNode.java
View File

@ -1,5 +1,8 @@
package com.ai.cloud.skywalking.analysis.chainbuild.entity;
import java.util.List;
import com.ai.cloud.skywalking.analysis.chainbuild.exception.TraceSpanTreeSerializeException;
import com.ai.cloud.skywalking.analysis.chainbuild.util.StringUtil;
import com.ai.cloud.skywalking.protocol.CallType;
import com.ai.cloud.skywalking.protocol.Span;
@ -13,6 +16,16 @@ public class TraceSpanNode {
protected TraceSpanNode sub = null;
protected String prevNodeRefToken = null;
protected String nextNodeRefToken = null;
protected String parentNodeRefToken = null;
protected String subNodeRefToken = null;
protected String nodeRefToken = null;
protected boolean visualNode = true;
protected String parentLevel;
@ -57,8 +70,8 @@ public class TraceSpanNode {
*/
protected String applicationId = "";
public TraceSpanNode(TraceSpanNode parent, TraceSpanNode sub, TraceSpanNode prev, TraceSpanNode next, Span span) {
this(parent, sub, prev, next);
public TraceSpanNode(TraceSpanNode parent, TraceSpanNode sub, TraceSpanNode prev, TraceSpanNode next, Span span, List<TraceSpanNode> spanContainer) {
this(parent, sub, prev, next, spanContainer);
this.visualNode = false;
this.parentLevel = span.getParentLevel();
this.levelId = span.getLevelId();
@ -75,30 +88,33 @@ public class TraceSpanNode {
this.spanType = span.getSpanType();
this.businessKey = span.getBusinessKey();
this.applicationId = span.getApplicationId();
//TODO: to set nodeToken
}
public TraceSpanNode(TraceSpanNode parent, TraceSpanNode sub, TraceSpanNode prev, TraceSpanNode next){
protected TraceSpanNode(TraceSpanNode parent, TraceSpanNode sub, TraceSpanNode prev, TraceSpanNode next, List<TraceSpanNode> spanContainer){
this.visualNode = true;
this.parent = parent;
this.setParent(parent);
if(parent != null){
parent.sub = this;
parent.setSub(this);
}
this.sub = sub;
this.setSub(sub);
if(sub != null){
sub.parent = this;
sub.setParent(this);
}
this.prev = prev;
this.setPrev(prev);
if(prev != null){
prev.next = this;
prev.setNext(this);
}
this.next = next;
this.setNext(next);
if(next != null){
next.prev = this;
next.setPrev(this);
}
spanContainer.add(this);
}
protected TraceSpanNode(TraceSpanNode parent, TraceSpanNode sub, TraceSpanNode prev, TraceSpanNode next, String parentLevelId, int levelId){
this(parent, sub, prev, next);
protected TraceSpanNode(TraceSpanNode parent, TraceSpanNode sub, TraceSpanNode prev, TraceSpanNode next, String parentLevelId, int levelId, List<TraceSpanNode> spanContainer){
this(parent, sub, prev, next, spanContainer);
this.parentLevel = parentLevelId;
this.levelId = levelId;
this.callTimes = 0;
@ -134,30 +150,38 @@ public class TraceSpanNode {
}
}
TraceSpanNode next(){
return this.next;
}
TraceSpanNode sub(){
return this.sub;
}
public TraceSpanNode getPrev() {
public TraceSpanNode prev() {
return prev;
}
public TraceSpanNode getNext() {
public TraceSpanNode next() {
return next;
}
public TraceSpanNode getParent() {
public TraceSpanNode parent() {
return parent;
}
public TraceSpanNode getSub() {
public TraceSpanNode sub() {
return sub;
}
public void setPrev(TraceSpanNode prev) {
this.prev = prev;
}
public void setNext(TraceSpanNode next) {
this.next = next;
}
public void setParent(TraceSpanNode parent) {
this.parent = parent;
}
public void setSub(TraceSpanNode sub) {
this.sub = sub;
}
public boolean isVisualNode() {
return visualNode;
}
@ -197,4 +221,26 @@ public class TraceSpanNode {
public String getApplicationId() {
return applicationId;
}
public String getNodeRefToken() throws TraceSpanTreeSerializeException {
if(StringUtil.isBlank(nodeRefToken)){
throw new TraceSpanTreeSerializeException("parentLevel=" + parentLevel + ", levelId=" + levelId + ", viewPointId=" + viewPointId + ", node ref token is null.");
}
return nodeRefToken;
}
void serializeRef() throws TraceSpanTreeSerializeException{
if(prev != null){
prevNodeRefToken = prev.getNodeRefToken();
}
if(parent != null){
parentNodeRefToken = parent.getNodeRefToken();
}
if(next != null){
nextNodeRefToken = next.getNodeRefToken();
}
if(sub != null){
subNodeRefToken = sub.getNodeRefToken();
}
}
}

19
skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/chainbuild/entity/TraceSpanTree.java
View File

@ -1,5 +1,6 @@
package com.ai.cloud.skywalking.analysis.chainbuild.entity;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
@ -8,6 +9,7 @@ import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.ai.cloud.skywalking.analysis.chainbuild.exception.BuildTraceSpanTreeException;
import com.ai.cloud.skywalking.analysis.chainbuild.exception.TraceSpanTreeSerializeException;
import com.ai.cloud.skywalking.analysis.chainbuild.util.StringUtil;
import com.ai.cloud.skywalking.analysis.chainbuild.util.TokenGenerator;
import com.ai.cloud.skywalking.protocol.Span;
@ -20,6 +22,8 @@ public class TraceSpanTree {
private String cid;
private TraceSpanNode treeRoot;
private List<TraceSpanNode> spanContainer = new ArrayList<TraceSpanNode>();
public TraceSpanTree() {
}
@ -40,7 +44,7 @@ public class TraceSpanTree {
}
});
cid = generateChainToken(spanList.get(0));
treeRoot = new TraceSpanNode(null, null, null, null, spanList.get(0));
treeRoot = new TraceSpanNode(null, null, null, null, spanList.get(0), spanContainer);
if (spanList.size() > 1) {
for (int i = 1; i < spanList.size(); i++) {
this.build(spanList.get(i));
@ -65,7 +69,7 @@ public class TraceSpanTree {
TraceSpanNode foundNode = findNodeAndCreateVisualNodeIfNess(
span.getParentLevel(), span.getLevelId() - 1);
if (foundNode != null) {
new TraceSpanNode(null, null, foundNode, foundNode.next(), span);
new TraceSpanNode(null, null, foundNode, foundNode.next(), span, spanContainer);
}
} else {
/**
@ -101,7 +105,7 @@ public class TraceSpanTree {
} else {
// create visual next node
currentNode = new VisualTraceSpanNode(null, null,
currentNode, null, contextParentLevelId, i);
currentNode, null, contextParentLevelId, i, spanContainer);
}
}
contextParentLevelId = contextParentLevelId == "" ? ("" + currentLevelInt)
@ -111,7 +115,7 @@ public class TraceSpanTree {
} else {
// create visual sub node
currentNode = new VisualTraceSpanNode(currentNode, null, null,
null, contextParentLevelId, 0);
null, contextParentLevelId, 0, spanContainer);
}
}
@ -130,5 +134,12 @@ public class TraceSpanTree {
+ level0Span.getTraceId() + " level0 span data is illegal");
}
}
private void beforeSerialize() throws TraceSpanTreeSerializeException{
for(TraceSpanNode treeNode : spanContainer){
treeNode.serializeRef();
}
}
}

8
skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/chainbuild/entity/VisualTraceSpanNode.java
View File

@ -1,11 +1,15 @@
package com.ai.cloud.skywalking.analysis.chainbuild.entity;
import java.util.List;
public class VisualTraceSpanNode extends TraceSpanNode {
protected VisualTraceSpanNode(TraceSpanNode parent, TraceSpanNode sub,
TraceSpanNode prev, TraceSpanNode next, String parentLevelId,
int levelId) {
super(parent, sub, prev, next, parentLevelId, levelId);
int levelId, List<TraceSpanNode> spanContainer) {
super(parent, sub, prev, next, parentLevelId, levelId, spanContainer);
//TODO: to set nodeToken
}

13
skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/chainbuild/exception/TraceSpanTreeSerializeException.java Normal file
View File

@ -0,0 +1,13 @@
package com.ai.cloud.skywalking.analysis.chainbuild.exception;
public class TraceSpanTreeSerializeException extends Exception {
private static final long serialVersionUID = 7857716041262993579L;
public TraceSpanTreeSerializeException(String msg){
super(msg);
}
public TraceSpanTreeSerializeException(String msg, Exception cause){
super(msg, cause);
}
}