SpringCloud/spring-cloud-sleuth
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixing issue #2083 (#2086)

Browse Source
This commit is contained in:
mschielmann
2021-12-28 22:51:23 +01:00
committed by GitHub
parent 093f77130c
commit 4a29094ccb
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/jdbc/TraceListenerStrategy.java
View File

@@ -20,6 +20,7 @@ import java.net.URI;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.Statement;
import java.util.Arrays;
import java.util.List;
import java.util.Locale;
import java.util.Map;
@@ -73,6 +74,10 @@ class TraceListenerStrategy<CON, STMT, RS> {
// string.
private static final Pattern URL_SERVICE_NAME_FINDER = Pattern.compile("sleuthServiceName=(.*?)(?:&|$)");
private static final String DEFAULT_SPAN_NAME = "query";
private static final List<String> KNOWN_SPAN_NAMES = Arrays.asList("select", "insert", "update", "delete");
private final Map<CON, ConnectionInfo> openConnections = new ConcurrentHashMap<>();
private final ThreadLocal<ConnectionInfo> currentConnection = new ThreadLocal<>();
@@ -429,7 +434,15 @@ class TraceListenerStrategy<CON, STMT, RS> {
}
private String spanName(String sql) {
return sql.substring(0, sql.indexOf(' ')).toLowerCase(Locale.ROOT);
String lowercaseSql = sql.toLowerCase(Locale.ROOT);
String spanName = DEFAULT_SPAN_NAME;
for (String spanNameCandidate : KNOWN_SPAN_NAMES) {
if (lowercaseSql.startsWith(spanNameCandidate)) {
spanName = spanNameCandidate;
break;
}
}
return spanName;
}
/**