com.aliyun.openservices.log.jaeger.sender.LogItemSpanConverter Maven / Gradle / Ivy

Go to download
package com.aliyun.openservices.log.jaeger.sender;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.aliyun.openservices.log.common.LogItem;
import com.uber.jaeger.LogData;
import com.uber.jaeger.Reference;
import com.uber.jaeger.Span;
import com.uber.jaeger.SpanContext;
import io.opentracing.References;

import java.util.List;
import java.util.Map;

public abstract class LogItemSpanConverter {

  public static LogItem convertSpan(Span span) {
    SpanContext context = span.context();

    boolean oneChildOfParent = span.getReferences().size() == 1
        && References.CHILD_OF.equals(span.getReferences().get(0).getType());

    LogItem logItem = new LogItem();
    logItem.PushBack(Constants.TRACE_ID, String.format("%x", context.getTraceId()));
    logItem.PushBack(Constants.SPAN_ID, String.format("%x", context.getSpanId()));
    logItem.PushBack(Constants.PARENT_SPAN_ID,
        String.format("%x", oneChildOfParent ? context.getParentId() : 0));
    logItem.PushBack(Constants.OPERATION_NAME, span.getOperationName());
    logItem.PushBack(Constants.FLAGS, String.format("%x", context.getFlags()));
    logItem.PushBack(Constants.START_TIME, String.valueOf(span.getStart()));
    logItem.PushBack(Constants.DURATION, String.valueOf(span.getDuration()));

    logItem.PushBack(Constants.REFERENCES, buildReferences(span.getReferences()).toJSONString());
    logItem.PushBack(Constants.TAGS, buildTags(span.getTags()).toJSONString());
    logItem.PushBack(Constants.LOGS, buildLogs(span.getLogs()).toJSONString());
    logItem.PushBack(Constants.PROCESS, buildProcess(span).toJSONString());

    return logItem;
  }

  static JSONObject buildTags(Map tags) {
    JSONObject jsonTags = new JSONObject();
    if (tags != null) {
      for (Map.Entry entry : tags.entrySet()) {
        jsonTags.put(entry.getKey(), entry.getValue());
      }
    }
    return jsonTags;
  }

  static JSONObject buildProcess(Span span) {
    JSONObject jsonProcess = new JSONObject();
    if (span != null) {
      jsonProcess.put(Constants.SERVICE_NAME, span.getServiceName());

      Map tags = span.getTracer().tags();
      JSONObject jsonTags = buildTags(tags);
      jsonProcess.put(Constants.TAGS, jsonTags);
    }
    return jsonProcess;
  }

  static JSONArray buildReferences(List references) {
    JSONArray jsonRefs = new JSONArray();
    if (references != null) {
      for (Reference reference : references) {
        JSONObject jsonRef = new JSONObject();
        jsonRef.put(Constants.REF_TYPE, reference.getType());
        jsonRef
            .put(Constants.TRACE_ID, String.format("%x", reference.getSpanContext().getTraceId()));
        jsonRef.put(Constants.SPAN_ID, String.format("%x", reference.getSpanContext().getSpanId()));
        jsonRefs.add(jsonRef);
      }
    }
    return jsonRefs;
  }

  static JSONArray buildLogs(List logs) {
    JSONArray jsonLogs = new JSONArray();
    if (logs != null) {
      for (LogData logData : logs) {
        JSONObject jsonLog = new JSONObject();
        jsonLog.put(Constants.TIMESTAMP, logData.getTime());
        if (logData.getFields() != null) {
          jsonLog.put(Constants.TAGS, buildTags(logData.getFields()));
        } else {
          JSONObject tags = new JSONObject();
          if (logData.getMessage() != null) {
            tags.put(Constants.EVENT, logData.getMessage());
          }
          jsonLog.put(Constants.TAGS, tags);
        }
        jsonLogs.add(jsonLog);
      }
    }
    return jsonLogs;
  }
}