java110
2023-05-16 d516e38fd44e186f393bbb05c4e7ca34b323b609
java110-core/src/main/java/com/java110/core/trace/Java110TraceFactory.java
@@ -3,9 +3,7 @@
import com.alibaba.fastjson.JSONObject;
import com.java110.core.factory.GenerateCodeFactory;
import com.java110.core.log.LoggerFactory;
import com.java110.dto.trace.TraceAnnotationsDto;
import com.java110.dto.trace.TraceDto;
import com.java110.dto.trace.TraceEndpointDto;
import com.java110.dto.trace.*;
import com.java110.utils.constant.CommonConstant;
import com.java110.utils.constant.EnvironmentConstant;
import com.java110.utils.factory.ApplicationContextFactory;
@@ -151,9 +149,33 @@
        traceDto.setAnnotations(traceAnnotationsDtos);
        traceDto.setTraceId(traceId);
        traceDto.setDbs(new ArrayList<>());
        put(traceDto.getId(), traceDto);
        putSpanId(SPAN_ID, traceDto.getId());
        return traceDto.getId();
    }
    /**
     * 添加参数
     *
     * @param traceParamDto
     */
    public static void putParams(TraceParamDto traceParamDto) {
        TraceDto traceDto = getTraceDto();
        if (traceDto == null) {
            return;
        }
        //如果存在 则跳过
        if (traceDto.getParam() != null) {
            return;
        }
        traceDto.setParam(traceParamDto);
        put(getSpanId(SPAN_ID), traceDto);
    }
    /**
@@ -206,6 +228,26 @@
    }
    /**
     * 添加db
     *
     * @param sql
     */
    public static void putDbs(String sql, String param, long duration) {
        TraceDto traceDto = getTraceDto();
        if (traceDto == null) {
            return;
        }
        List<TraceDbDto> dbs = traceDto.getDbs();
        TraceDbDto traceDbDto = new TraceDbDto();
        traceDbDto.setDbSql(sql);
        traceDbDto.setParam(param);
        traceDbDto.setDuration(duration+"");
        dbs.add(traceDbDto);
        traceDto.setDbs(dbs);
    }
    /**
     * 清理事务
     */
    public static void clearTrace(String spanId) {