Skip to content

Commit 7e99562

Browse files
committed
simplify implementation, getDb returns instance if db not found
1 parent 8bf5ad3 commit 7e99562

File tree

4 files changed

+3
-19
lines changed

4 files changed

+3
-19
lines changed

dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/jdbc/DBInfo.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ public String getInstance() {
193193
}
194194

195195
public String getDb() {
196-
return db;
196+
return db != null ? db : instance;
197197
}
198198

199199
public String getHost() {

dd-java-agent/instrumentation/jdbc/src/main/java/datadog/trace/instrumentation/jdbc/DBMCompatibleConnectionInstrumentation.java

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -132,13 +132,7 @@ public static String onEnter(
132132
DECORATE.DBM_ALWAYS_APPEND_SQL_COMMENT || "sqlserver".equals(dbInfo.getType());
133133
sql =
134134
SQLCommenter.inject(
135-
sql,
136-
dbService,
137-
dbInfo.getType(),
138-
dbInfo.getHost(),
139-
DECORATE.isOracle(dbInfo) ? DECORATE.getDbInstance(dbInfo) : dbInfo.getDb(),
140-
null,
141-
append);
135+
sql, dbService, dbInfo.getType(), dbInfo.getHost(), dbInfo.getDb(), null, append);
142136
return inputSql;
143137
}
144138

dd-java-agent/instrumentation/jdbc/src/main/java/datadog/trace/instrumentation/jdbc/JDBCDecorator.java

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -219,13 +219,6 @@ public String getDbService(final DBInfo dbInfo) {
219219
return dbService(dbInfo.getType(), dbInstance(dbInfo));
220220
}
221221

222-
public String getDbInstance(final DBInfo dbInfo) {
223-
if (null == dbInfo) {
224-
return null;
225-
}
226-
return dbInstance(dbInfo);
227-
}
228-
229222
public static DBInfo parseDBInfoFromConnection(final Connection connection) {
230223
if (connection == null || !FETCH_DB_METADATA_ON_QUERY) {
231224
// we can log here, but it risks to be too verbose

dd-java-agent/instrumentation/jdbc/src/main/java/datadog/trace/instrumentation/jdbc/StatementInstrumentation.java

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -147,26 +147,23 @@ public static AgentScope onEnter(
147147
}
148148

149149
final String dbService;
150-
final String dbName;
151150
if (isOracle) {
152151
String oracleService = DECORATE.getDbService(dbInfo);
153152
if (oracleService != null) {
154153
oracleService =
155154
traceConfig(span).getServiceMapping().getOrDefault(oracleService, oracleService);
156155
}
157156
dbService = oracleService;
158-
dbName = DECORATE.getDbInstance(dbInfo);
159157
} else {
160158
dbService = span.getServiceName();
161-
dbName = dbInfo.getDb();
162159
}
163160
sql =
164161
SQLCommenter.inject(
165162
sql,
166163
dbService,
167164
dbInfo.getType(),
168165
dbInfo.getHost(),
169-
dbName,
166+
dbInfo.getDb(),
170167
injectTraceInComment ? traceParent : null,
171168
appendComment);
172169
}

0 commit comments

Comments
 (0)