Skip to content

Commit 61d4637

Browse files
committed
DbInfo simplification
1 parent 0918142 commit 61d4637

7 files changed

Lines changed: 72 additions & 97 deletions

File tree

instrumentation/jdbc/library/src/main/java/io/opentelemetry/instrumentation/jdbc/internal/DataSourceDbAttributesExtractor.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,6 @@ public void onEnd(
5454
}
5555

5656
private static String getName(DbInfo dbInfo) {
57-
String name = dbInfo.getName();
58-
return name == null ? dbInfo.getDb() : name;
57+
return dbInfo.getName();
5958
}
6059
}

instrumentation/jdbc/library/src/main/java/io/opentelemetry/instrumentation/jdbc/internal/JdbcAttributesGetter.java

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
package io.opentelemetry.instrumentation.jdbc.internal;
77

88
import io.opentelemetry.instrumentation.api.incubator.semconv.db.SqlClientAttributesGetter;
9-
import io.opentelemetry.instrumentation.jdbc.internal.dbinfo.DbInfo;
109
import java.sql.SQLException;
1110
import java.util.Collection;
1211
import java.util.Map;
@@ -36,8 +35,7 @@ public String getUser(DbRequest request) {
3635
@Nullable
3736
@Override
3837
public String getDbNamespace(DbRequest request) {
39-
DbInfo dbInfo = request.getDbInfo();
40-
return dbInfo.getName() == null ? dbInfo.getDb() : dbInfo.getName();
38+
return request.getDbInfo().getName();
4139
}
4240

4341
@Deprecated // to be removed in 3.0

instrumentation/jdbc/library/src/main/java/io/opentelemetry/instrumentation/jdbc/internal/JdbcConnectionUrlParser.java

Lines changed: 20 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ DbInfo.Builder doParse(String jdbcUrl, DbInfo.Builder builder) {
6363
path = path.substring(1);
6464
}
6565
if (!path.isEmpty()) {
66-
builder.db(path);
66+
builder.name(path);
6767
}
6868
if (uri.getHost() != null) {
6969
builder.host(uri.getHost());
@@ -91,13 +91,6 @@ DbInfo.Builder doParse(String jdbcUrl, DbInfo.Builder builder) {
9191
}
9292

9393
String[] split = jdbcUrl.split(";", 2);
94-
if (split.length > 1) {
95-
Map<String, String> props = splitQuery(split[1], ";");
96-
populateStandardProperties(builder, props);
97-
if (props.containsKey("instance")) {
98-
builder.name(props.get("instance"));
99-
}
100-
}
10194

10295
String urlServerName = split[0].substring(hostIndex + 17);
10396
if (!urlServerName.isEmpty()) {
@@ -106,10 +99,18 @@ DbInfo.Builder doParse(String jdbcUrl, DbInfo.Builder builder) {
10699

107100
int databaseLoc = serverName.indexOf("/");
108101
if (databaseLoc > 1) {
109-
builder.db(serverName.substring(databaseLoc + 1));
102+
builder.name(serverName.substring(databaseLoc + 1));
110103
serverName = serverName.substring(0, databaseLoc);
111104
}
112105

106+
if (split.length > 1) {
107+
Map<String, String> props = splitQuery(split[1], ";");
108+
populateStandardProperties(builder, props);
109+
if (props.containsKey("instance")) {
110+
builder.name(props.get("instance"));
111+
}
112+
}
113+
113114
int portLoc = serverName.indexOf(":");
114115
if (portLoc > 1) {
115116
builder.port(Integer.parseInt(serverName.substring(portLoc + 1)));
@@ -294,9 +295,9 @@ DbInfo.Builder doParse(String jdbcUrl, DbInfo.Builder builder) {
294295

295296
if (paramLoc > 0) {
296297
populateStandardProperties(builder, splitQuery(jdbcUrl.substring(paramLoc + 1), "&"));
297-
builder.db(jdbcUrl.substring(dbLoc + 1, paramLoc));
298+
builder.name(jdbcUrl.substring(dbLoc + 1, paramLoc));
298299
} else {
299-
builder.db(jdbcUrl.substring(dbLoc + 1));
300+
builder.name(jdbcUrl.substring(dbLoc + 1));
300301
}
301302

302303
if (portLoc > 0) {
@@ -330,9 +331,9 @@ DbInfo.Builder doParse(String jdbcUrl, DbInfo.Builder builder) {
330331

331332
if (paramLoc > 0) {
332333
populateStandardProperties(builder, splitQuery(jdbcUrl.substring(paramLoc + 1), "&"));
333-
builder.db(jdbcUrl.substring(dbLoc + 1, paramLoc));
334+
builder.name(jdbcUrl.substring(dbLoc + 1, paramLoc));
334335
} else if (dbLoc != -1) {
335-
builder.db(jdbcUrl.substring(dbLoc + 1));
336+
builder.name(jdbcUrl.substring(dbLoc + 1));
336337
}
337338

338339
if (jdbcUrl.startsWith("address=")) {
@@ -808,7 +809,10 @@ DbInfo.Builder doParse(String jdbcUrl, DbInfo.Builder builder) {
808809
if (host != null) {
809810
builder.host(host);
810811
}
811-
return builder.name(instance);
812+
if (instance != null) {
813+
builder.name(instance);
814+
}
815+
return builder;
812816
}
813817
},
814818

@@ -1116,10 +1120,10 @@ private static void populateStandardProperties(DbInfo.Builder builder, Map<?, ?>
11161120
}
11171121

11181122
if (props.containsKey("databasename")) {
1119-
builder.db((String) props.get("databasename"));
1123+
builder.name((String) props.get("databasename"));
11201124
}
11211125
if (props.containsKey("databaseName")) {
1122-
builder.db((String) props.get("databaseName"));
1126+
builder.name((String) props.get("databaseName"));
11231127
}
11241128

11251129
if (props.containsKey("servername")) {

instrumentation/jdbc/library/src/main/java/io/opentelemetry/instrumentation/jdbc/internal/dbinfo/DbInfo.java

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,6 @@ public static DbInfo.Builder builder() {
3737
@Nullable
3838
public abstract String getName();
3939

40-
@Nullable
41-
public abstract String getDb();
42-
4340
@Nullable
4441
public abstract String getHost();
4542

@@ -53,7 +50,6 @@ public Builder toBuilder() {
5350
.shortUrl(getShortUrl())
5451
.user(getUser())
5552
.name(getName())
56-
.db(getDb())
5753
.host(getHost())
5854
.port(getPort());
5955
}
@@ -75,8 +71,6 @@ public abstract static class Builder {
7571

7672
public abstract Builder name(String name);
7773

78-
public abstract Builder db(String db);
79-
8074
public abstract Builder host(String host);
8175

8276
public abstract Builder port(Integer port);

instrumentation/jdbc/library/src/test/java/io/opentelemetry/instrumentation/jdbc/datasource/OpenTelemetryDataSourceTest.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,8 +105,7 @@ private static void assertDbInfo(DbInfo dbInfo) {
105105
assertThat(dbInfo.getSubtype()).isNull();
106106
assertThat(dbInfo.getShortUrl()).isEqualTo("postgresql://127.0.0.1:5432");
107107
assertThat(dbInfo.getUser()).isNull();
108-
assertThat(dbInfo.getName()).isNull();
109-
assertThat(dbInfo.getDb()).isEqualTo("dbname");
108+
assertThat(dbInfo.getName()).isEqualTo("dbname");
110109
assertThat(dbInfo.getHost()).isEqualTo("127.0.0.1");
111110
assertThat(dbInfo.getPort()).isEqualTo(5432);
112111
}

0 commit comments

Comments
 (0)