Skip to content

Commit 6525b9a

Browse files
committed
Fix tests
1 parent 990b829 commit 6525b9a

File tree

4 files changed

+17
-51
lines changed

4 files changed

+17
-51
lines changed

instrumentation-api-incubator/src/main/java/io/opentelemetry/instrumentation/api/incubator/semconv/db/DbClientCommonAttributesExtractor.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ public void onStart(AttributesBuilder attributes, Context parentContext, REQUEST
4545
internalSet(attributes, DB_NAME, getter.getDbNamespace(request));
4646
internalSet(attributes, DB_CONNECTION_STRING, getter.getConnectionString(request));
4747
}
48-
internalSet(attributes, DB_SYSTEM, getter.getSystem(request));
48+
internalSet(attributes, DB_SYSTEM, getter.getDbSystem(request));
4949
}
5050

5151
@Override

instrumentation-api-incubator/src/main/java/io/opentelemetry/instrumentation/api/incubator/semconv/db/SqlClientAttributesExtractor.java

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -65,28 +65,24 @@ public static <REQUEST, RESPONSE> SqlClientAttributesExtractorBuilder<REQUEST, R
6565
this.statementSanitizationEnabled = statementSanitizationEnabled;
6666
}
6767

68-
@SuppressWarnings("deprecation") // until old semconv are dropped
6968
@Override
7069
public void onStart(AttributesBuilder attributes, Context parentContext, REQUEST request) {
7170
super.onStart(attributes, parentContext, request);
7271

73-
String rawStatement =
74-
SemconvStability.emitStableDatabaseSemconv()
75-
? getter.getRawQueryText(request)
76-
: getter.getRawStatement(request);
77-
SqlStatementInfo sanitizedStatement = sanitizer.sanitize(rawStatement);
72+
String rawQueryText = getter.getRawQueryText(request);
73+
SqlStatementInfo sanitizedStatement = sanitizer.sanitize(rawQueryText);
7874
String operation = sanitizedStatement.getOperation();
7975
if (SemconvStability.emitStableDatabaseSemconv()) {
8076
internalSet(
8177
attributes,
8278
DB_QUERY_TEXT,
83-
statementSanitizationEnabled ? sanitizedStatement.getFullStatement() : rawStatement);
79+
statementSanitizationEnabled ? sanitizedStatement.getFullStatement() : rawQueryText);
8480
internalSet(attributes, DB_OPERATION_NAME, operation);
8581
} else if (SemconvStability.emitOldDatabaseSemconv()) {
8682
internalSet(
8783
attributes,
8884
DB_STATEMENT,
89-
statementSanitizationEnabled ? sanitizedStatement.getFullStatement() : rawStatement);
85+
statementSanitizationEnabled ? sanitizedStatement.getFullStatement() : rawQueryText);
9086
internalSet(attributes, DB_OPERATION, operation);
9187
}
9288
if (!SQL_CALL.equals(operation)) {

instrumentation-api-incubator/src/test/java/io/opentelemetry/instrumentation/api/incubator/semconv/db/DbClientAttributesExtractorTest.java

Lines changed: 7 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
import static io.opentelemetry.sdk.testing.assertj.OpenTelemetryAssertions.assertThat;
99
import static org.assertj.core.api.Assertions.entry;
1010

11+
import io.opentelemetry.api.common.AttributeKey;
1112
import io.opentelemetry.api.common.Attributes;
1213
import io.opentelemetry.api.common.AttributesBuilder;
1314
import io.opentelemetry.context.Context;
@@ -37,16 +38,11 @@ public String getUser(Map<String, String> map) {
3738
return map.get("db.user");
3839
}
3940

40-
@Deprecated
41-
@Override
42-
public String getName(Map<String, String> map) {
43-
return map.get("db.name");
44-
}
45-
4641
@Nullable
4742
@Override
4843
public String getDbNamespace(Map<String, String> map) {
49-
return map.get("db.namespace");
44+
return map.get(
45+
SemconvStabilityUtil.getAttributeKey(AttributeKey.stringKey("db.name")).getKey());
5046
}
5147

5248
@Deprecated
@@ -55,28 +51,18 @@ public String getConnectionString(Map<String, String> map) {
5551
return map.get("db.connection_string");
5652
}
5753

58-
@Deprecated
59-
@Override
60-
public String getStatement(Map<String, String> map) {
61-
return map.get("db.statement");
62-
}
63-
6454
@Nullable
6555
@Override
6656
public String getDbQueryText(Map<String, String> map) {
67-
return map.get("db.query.text");
68-
}
69-
70-
@Deprecated
71-
@Override
72-
public String getOperation(Map<String, String> map) {
73-
return map.get("db.operation");
57+
return map.get(
58+
SemconvStabilityUtil.getAttributeKey(AttributeKey.stringKey("db.statement")).getKey());
7459
}
7560

7661
@Nullable
7762
@Override
7863
public String getDbOperationName(Map<String, String> map) {
79-
return map.get("db.operation.name");
64+
return map.get(
65+
SemconvStabilityUtil.getAttributeKey(AttributeKey.stringKey("db.operation")).getKey());
8066
}
8167
}
8268

instrumentation-api-incubator/src/test/java/io/opentelemetry/instrumentation/api/incubator/semconv/db/SqlClientAttributesExtractorTest.java

Lines changed: 5 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
import static io.opentelemetry.sdk.testing.assertj.OpenTelemetryAssertions.assertThat;
99
import static org.assertj.core.api.Assertions.entry;
1010

11+
import io.opentelemetry.api.common.AttributeKey;
1112
import io.opentelemetry.api.common.Attributes;
1213
import io.opentelemetry.api.common.AttributesBuilder;
1314
import io.opentelemetry.context.Context;
@@ -26,21 +27,10 @@ class SqlClientAttributesExtractorTest {
2627
static final class TestAttributesGetter
2728
implements SqlClientAttributesGetter<Map<String, String>> {
2829

29-
@Deprecated
30-
@Override
31-
public String getRawStatement(Map<String, String> map) {
32-
return map.get("db.statement");
33-
}
34-
3530
@Override
3631
public String getRawQueryText(Map<String, String> map) {
37-
return map.get("db.query.text");
38-
}
39-
40-
@Deprecated
41-
@Override
42-
public String getSystem(Map<String, String> map) {
43-
return map.get("db.system");
32+
return map.get(
33+
SemconvStabilityUtil.getAttributeKey(AttributeKey.stringKey("db.statement")).getKey());
4434
}
4535

4636
@Nullable
@@ -55,17 +45,11 @@ public String getUser(Map<String, String> map) {
5545
return map.get("db.user");
5646
}
5747

58-
@Deprecated
59-
@Nullable
60-
@Override
61-
public String getName(Map<String, String> map) {
62-
return map.get("db.name");
63-
}
64-
6548
@Nullable
6649
@Override
6750
public String getDbNamespace(Map<String, String> map) {
68-
return map.get("db.namespace");
51+
return map.get(
52+
SemconvStabilityUtil.getAttributeKey(AttributeKey.stringKey("db.name")).getKey());
6953
}
7054

7155
@Deprecated

0 commit comments

Comments
 (0)