优化代码

java110

2022-05-21 c2614ad5c7d1e27fae68e52cd058af363464cdb4

 java110-core/src/main/java/com/java110/core/trace/Java110TraceFactory.java

@@ -3,10 +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.TraceParamDto;
import com.java110.dto.trace.*;
import com.java110.utils.constant.CommonConstant;
import com.java110.utils.constant.EnvironmentConstant;
import com.java110.utils.factory.ApplicationContextFactory;
@@ -152,6 +149,8 @@
        traceDto.setAnnotations(traceAnnotationsDtos);
        traceDto.setTraceId(traceId);

        traceDto.setDbs(new ArrayList<>());

        put(traceDto.getId(), traceDto);
        putSpanId(SPAN_ID, traceDto.getId());
        return traceDto.getId();
@@ -229,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) {