@Override
public void insert(TSpan span) {
if (span == null) {
throw new NullPointerException("span must not be null");
}
Trace trace = findTrace(span.getAgentId(), span.getAgentStartTime(), span.getSpanId());
final TransactionId transactionId =
TransactionIdUtils.parseTransactionId(span.getTransactionId());
trace.setTraceAgentId(transactionId.getAgentId());
if (trace.getTraceAgentId() == null) {
trace.setTraceAgentId(span.getAgentId());
}
trace.setTraceAgentStartTime(transactionId.getAgentStartTime());
trace.setTraceTransactionSequence(transactionId.getTransactionSequence());
trace.setSpanId(span.getSpanId());
trace.setParentSpanId(span.getParentSpanId());
trace.setStartTime(span.getStartTime());
trace.setElapsed(span.getElapsed());
trace.setRpc(span.getRpc());
trace.setServiceType(span.getServiceType());
trace.setEndPoint(span.getEndPoint());
trace.setFlag(span.getFlag());
ApiMetaData apiMetaData = findApiMetaData(trace.getAgentId(), span.getApiId());
trace.setApiId(apiMetaData.getId());
trace.setErrCode(span.getErr());
trace.setRemoteAddr(span.getRemoteAddr());
final TIntStringValue exceptionInfo = span.getExceptionInfo();
if (exceptionInfo != null) {
trace.setHasException(true);
trace.setExceptionId(exceptionInfo.getIntValue());
trace.setExceptionMessage(exceptionInfo.getStringValue());
}
long acceptedTime = acceptedTimeService.getAcceptedTime();
trace.setCollectorAcceptTime(acceptedTime);
Trace savedTrace = traceRepository.saveAndFlush(trace);
if (span.getAnnotations() != null) {
List<Annotation> annotations =
span.getAnnotations()
.stream()
.filter(tAnnotation -> tAnnotation != null)
.map(
tAnnotation -> {
Annotation annotation = new Annotation();
annotation.setKey(tAnnotation.getKey());
Object value = transcoder.getMappingValue(tAnnotation);
annotation.setValueType(transcoder.getTypeCode(value));
annotation.setByteValue(transcoder.encode(value, annotation.getValueType()));
annotation.setTraceId(savedTrace.getId());
return annotation;
})
.collect(Collectors.toList());
annotationRepository.save(annotations);
}
addNestedSpanEvent(span, savedTrace);
}
private void addNestedSpanEvent(TSpan span, Trace trace) {
Assert.notNull(trace, "trace must not be null while add span event.");
List<TSpanEvent> spanEventBoList = span.getSpanEventList();
if (CollectionUtils.isEmpty(spanEventBoList)) {
return;
}
long acceptedTime0 = acceptedTimeService.getAcceptedTime();
spanEventBoList
.stream()
.filter(spanEvent -> spanEvent != null)
.forEach(
spanEvent -> {
TraceEvent event = new TraceEvent();
event.setAgentId(trace.getAgentId());
event.setTraceId(trace.getId());
final TransactionId transactionId =
TransactionIdUtils.parseTransactionId(span.getTransactionId());
event.setTraceAgentId(transactionId.getAgentId());
if (event.getTraceAgentId() == null) {
event.setTraceAgentId(span.getAgentId());
}
event.setTraceAgentStartTime(transactionId.getAgentStartTime());
event.setTraceTransactionSequence(transactionId.getTransactionSequence());
event.setSequence(spanEvent.getSequence());
event.setStartElapsed(spanEvent.getStartElapsed());
event.setEndElapsed(spanEvent.getEndElapsed());
event.setRpc(spanEvent.getRpc());
event.setServiceType(spanEvent.getServiceType());
event.setDestinationId(spanEvent.getDestinationId());
event.setEndPoint(spanEvent.getEndPoint());
ApiMetaData apiMetaData = findApiMetaData(trace.getAgentId(), spanEvent.getApiId());
event.setApiId(apiMetaData.getId());
if (spanEvent.isSetDepth()) {
event.setDepth(spanEvent.getDepth());
}
if (spanEvent.isSetNextSpanId()) {
event.setNextSpanId(spanEvent.getNextSpanId());
}
final TIntStringValue exceptionInfo = spanEvent.getExceptionInfo();
if (exceptionInfo != null) {
event.setHasException(true);
event.setExceptionId(exceptionInfo.getIntValue());
event.setExceptionMessage(exceptionInfo.getStringValue());
}
event.setCollectorAcceptTime(acceptedTime0);
event.setTraceId(trace.getId());
TraceEvent saveTraceEvent = traceEventRepository.save(event);
if (spanEvent.getAnnotations() != null) {
List<Annotation> annotations =
spanEvent
.getAnnotations()
.stream()
.filter(tAnnotation -> tAnnotation != null)
.map(
tAnnotation -> {
Annotation annotation = new Annotation();
annotation.setKey(tAnnotation.getKey());
Object value = transcoder.getMappingValue(tAnnotation);
annotation.setValueType(transcoder.getTypeCode(value));
annotation.setByteValue(
transcoder.encode(value, annotation.getValueType()));
annotation.setTraceEventId(saveTraceEvent.getId());
return annotation;
})
.collect(Collectors.toList());
annotationRepository.save(annotations);
}
});
}