From 4e7907983a99e695f2acba4dec8bd73b7dbe6b2b Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期四, 29 六月 2023 14:41:33 +0800
Subject: [PATCH] optimize
---
java110-core/src/main/java/com/java110/core/trace/Java110TraceFactory.java | 27 +++++++++++++++++++++++----
1 files changed, 23 insertions(+), 4 deletions(-)
diff --git a/java110-core/src/main/java/com/java110/core/trace/Java110TraceFactory.java b/java110-core/src/main/java/com/java110/core/trace/Java110TraceFactory.java
index 57db2b3..4f279c4 100644
--- a/java110-core/src/main/java/com/java110/core/trace/Java110TraceFactory.java
+++ b/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) {
--
Gitblit v1.8.0