1616import io .opentelemetry .context .Context ;
1717import io .opentelemetry .instrumentation .api .instrumenter .AttributesExtractor ;
1818import io .opentelemetry .instrumentation .api .internal .SemconvStability ;
19+ import io .opentelemetry .semconv .DbAttributes ;
1920import io .opentelemetry .semconv .incubating .DbIncubatingAttributes ;
2021import java .util .Arrays ;
2122import java .util .Collection ;
@@ -118,17 +119,17 @@ void shouldExtractAllAttributes() {
118119 assertThat (startAttributes .build ())
119120 .containsOnly (
120121 entry (DbIncubatingAttributes .DB_SYSTEM , "myDb" ),
121- entry (DbIncubatingAttributes .DB_SYSTEM_NAME , "myDb" ),
122+ entry (DbAttributes .DB_SYSTEM_NAME , "myDb" ),
122123 entry (DbIncubatingAttributes .DB_USER , "username" ),
123124 entry (DbIncubatingAttributes .DB_NAME , "potatoes" ),
124125 entry (DbIncubatingAttributes .DB_CONNECTION_STRING , "mydb:///potatoes" ),
125126 entry (DbIncubatingAttributes .DB_STATEMENT , "SELECT * FROM potato WHERE id=?" ),
126127 entry (DbIncubatingAttributes .DB_OPERATION , "SELECT" ),
127128 entry (DbIncubatingAttributes .DB_SQL_TABLE , "potato" ),
128- entry (DbIncubatingAttributes .DB_NAMESPACE , "potatoes" ),
129- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "SELECT * FROM potato WHERE id=?" ),
130- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "SELECT" ),
131- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "potato" ));
129+ entry (DbAttributes .DB_NAMESPACE , "potatoes" ),
130+ entry (DbAttributes .DB_QUERY_TEXT , "SELECT * FROM potato WHERE id=?" ),
131+ entry (DbAttributes .DB_OPERATION_NAME , "SELECT" ),
132+ entry (DbAttributes .DB_COLLECTION_NAME , "potato" ));
132133 } else if (SemconvStability .emitOldDatabaseSemconv ()) {
133134 assertThat (startAttributes .build ())
134135 .containsOnly (
@@ -142,11 +143,11 @@ void shouldExtractAllAttributes() {
142143 } else if (SemconvStability .emitStableDatabaseSemconv ()) {
143144 assertThat (startAttributes .build ())
144145 .containsOnly (
145- entry (DbIncubatingAttributes .DB_SYSTEM_NAME , "myDb" ),
146- entry (DbIncubatingAttributes .DB_NAMESPACE , "potatoes" ),
147- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "SELECT * FROM potato WHERE id=?" ),
148- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "SELECT" ),
149- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "potato" ));
146+ entry (DbAttributes .DB_SYSTEM_NAME , "myDb" ),
147+ entry (DbAttributes .DB_NAMESPACE , "potatoes" ),
148+ entry (DbAttributes .DB_QUERY_TEXT , "SELECT * FROM potato WHERE id=?" ),
149+ entry (DbAttributes .DB_OPERATION_NAME , "SELECT" ),
150+ entry (DbAttributes .DB_COLLECTION_NAME , "potato" ));
150151 }
151152
152153 assertThat (endAttributes .build ().isEmpty ()).isTrue ();
@@ -173,8 +174,8 @@ void shouldNotExtractTableIfAttributeIsNotSet() {
173174 .containsOnly (
174175 entry (DbIncubatingAttributes .DB_STATEMENT , "SELECT *" ),
175176 entry (DbIncubatingAttributes .DB_OPERATION , "SELECT" ),
176- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "SELECT *" ),
177- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "SELECT" ));
177+ entry (DbAttributes .DB_QUERY_TEXT , "SELECT *" ),
178+ entry (DbAttributes .DB_OPERATION_NAME , "SELECT" ));
178179 } else if (SemconvStability .emitOldDatabaseSemconv ()) {
179180 assertThat (attributes .build ())
180181 .containsOnly (
@@ -183,8 +184,8 @@ void shouldNotExtractTableIfAttributeIsNotSet() {
183184 } else if (SemconvStability .emitStableDatabaseSemconv ()) {
184185 assertThat (attributes .build ())
185186 .containsOnly (
186- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "SELECT *" ),
187- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "SELECT" ));
187+ entry (DbAttributes .DB_QUERY_TEXT , "SELECT *" ),
188+ entry (DbAttributes .DB_OPERATION_NAME , "SELECT" ));
188189 }
189190 }
190191
@@ -213,9 +214,9 @@ void shouldExtractTableToSpecifiedKey() {
213214 entry (DbIncubatingAttributes .DB_STATEMENT , "SELECT * FROM table" ),
214215 entry (DbIncubatingAttributes .DB_OPERATION , "SELECT" ),
215216 entry (DbIncubatingAttributes .DB_CASSANDRA_TABLE , "table" ),
216- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "SELECT * FROM table" ),
217- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "SELECT" ),
218- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "table" ));
217+ entry (DbAttributes .DB_QUERY_TEXT , "SELECT * FROM table" ),
218+ entry (DbAttributes .DB_OPERATION_NAME , "SELECT" ),
219+ entry (DbAttributes .DB_COLLECTION_NAME , "table" ));
219220 } else if (SemconvStability .emitOldDatabaseSemconv ()) {
220221 assertThat (attributes .build ())
221222 .containsOnly (
@@ -225,9 +226,9 @@ void shouldExtractTableToSpecifiedKey() {
225226 } else if (SemconvStability .emitStableDatabaseSemconv ()) {
226227 assertThat (attributes .build ())
227228 .containsOnly (
228- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "SELECT * FROM table" ),
229- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "SELECT" ),
230- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "table" ));
229+ entry (DbAttributes .DB_QUERY_TEXT , "SELECT * FROM table" ),
230+ entry (DbAttributes .DB_OPERATION_NAME , "SELECT" ),
231+ entry (DbAttributes .DB_COLLECTION_NAME , "table" ));
231232 }
232233 }
233234
@@ -273,11 +274,11 @@ void shouldExtractSingleQueryBatchAttributes() {
273274 entry (DbIncubatingAttributes .DB_STATEMENT , "INSERT INTO potato VALUES(?)" ),
274275 entry (DbIncubatingAttributes .DB_OPERATION , "INSERT" ),
275276 entry (DbIncubatingAttributes .DB_SQL_TABLE , "potato" ),
276- entry (DbIncubatingAttributes .DB_NAMESPACE , "potatoes" ),
277- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
278- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "BATCH INSERT" ),
279- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "potato" ),
280- entry (DbIncubatingAttributes .DB_OPERATION_BATCH_SIZE , 2L ));
277+ entry (DbAttributes .DB_NAMESPACE , "potatoes" ),
278+ entry (DbAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
279+ entry (DbAttributes .DB_OPERATION_NAME , "BATCH INSERT" ),
280+ entry (DbAttributes .DB_COLLECTION_NAME , "potato" ),
281+ entry (DbAttributes .DB_OPERATION_BATCH_SIZE , 2L ));
281282 } else if (SemconvStability .emitOldDatabaseSemconv ()) {
282283 assertThat (startAttributes .build ())
283284 .containsOnly (
@@ -288,11 +289,11 @@ void shouldExtractSingleQueryBatchAttributes() {
288289 } else if (SemconvStability .emitStableDatabaseSemconv ()) {
289290 assertThat (startAttributes .build ())
290291 .containsOnly (
291- entry (DbIncubatingAttributes .DB_NAMESPACE , "potatoes" ),
292- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
293- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "BATCH INSERT" ),
294- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "potato" ),
295- entry (DbIncubatingAttributes .DB_OPERATION_BATCH_SIZE , 2L ));
292+ entry (DbAttributes .DB_NAMESPACE , "potatoes" ),
293+ entry (DbAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
294+ entry (DbAttributes .DB_OPERATION_NAME , "BATCH INSERT" ),
295+ entry (DbAttributes .DB_COLLECTION_NAME , "potato" ),
296+ entry (DbAttributes .DB_OPERATION_BATCH_SIZE , 2L ));
296297 }
297298
298299 assertThat (endAttributes .build ().isEmpty ()).isTrue ();
@@ -325,22 +326,22 @@ void shouldExtractMultiQueryBatchAttributes() {
325326 assertThat (startAttributes .build ())
326327 .containsOnly (
327328 entry (DbIncubatingAttributes .DB_NAME , "potatoes" ),
328- entry (DbIncubatingAttributes .DB_NAMESPACE , "potatoes" ),
329- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
330- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "BATCH INSERT" ),
331- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "potato" ),
332- entry (DbIncubatingAttributes .DB_OPERATION_BATCH_SIZE , 2L ));
329+ entry (DbAttributes .DB_NAMESPACE , "potatoes" ),
330+ entry (DbAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
331+ entry (DbAttributes .DB_OPERATION_NAME , "BATCH INSERT" ),
332+ entry (DbAttributes .DB_COLLECTION_NAME , "potato" ),
333+ entry (DbAttributes .DB_OPERATION_BATCH_SIZE , 2L ));
333334 } else if (SemconvStability .emitOldDatabaseSemconv ()) {
334335 assertThat (startAttributes .build ())
335336 .containsOnly (entry (DbIncubatingAttributes .DB_NAME , "potatoes" ));
336337 } else if (SemconvStability .emitStableDatabaseSemconv ()) {
337338 assertThat (startAttributes .build ())
338339 .containsOnly (
339- entry (DbIncubatingAttributes .DB_NAMESPACE , "potatoes" ),
340- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
341- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "BATCH INSERT" ),
342- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "potato" ),
343- entry (DbIncubatingAttributes .DB_OPERATION_BATCH_SIZE , 2L ));
340+ entry (DbAttributes .DB_NAMESPACE , "potatoes" ),
341+ entry (DbAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
342+ entry (DbAttributes .DB_OPERATION_NAME , "BATCH INSERT" ),
343+ entry (DbAttributes .DB_COLLECTION_NAME , "potato" ),
344+ entry (DbAttributes .DB_OPERATION_BATCH_SIZE , 2L ));
344345 }
345346
346347 assertThat (endAttributes .build ().isEmpty ()).isTrue ();
@@ -374,10 +375,10 @@ void shouldIgnoreBatchSizeOne() {
374375 entry (DbIncubatingAttributes .DB_STATEMENT , "INSERT INTO potato VALUES(?)" ),
375376 entry (DbIncubatingAttributes .DB_OPERATION , "INSERT" ),
376377 entry (DbIncubatingAttributes .DB_SQL_TABLE , "potato" ),
377- entry (DbIncubatingAttributes .DB_NAMESPACE , "potatoes" ),
378- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
379- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "INSERT" ),
380- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "potato" ));
378+ entry (DbAttributes .DB_NAMESPACE , "potatoes" ),
379+ entry (DbAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
380+ entry (DbAttributes .DB_OPERATION_NAME , "INSERT" ),
381+ entry (DbAttributes .DB_COLLECTION_NAME , "potato" ));
381382 } else if (SemconvStability .emitOldDatabaseSemconv ()) {
382383 assertThat (startAttributes .build ())
383384 .containsOnly (
@@ -388,10 +389,10 @@ void shouldIgnoreBatchSizeOne() {
388389 } else if (SemconvStability .emitStableDatabaseSemconv ()) {
389390 assertThat (startAttributes .build ())
390391 .containsOnly (
391- entry (DbIncubatingAttributes .DB_NAMESPACE , "potatoes" ),
392- entry (DbIncubatingAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
393- entry (DbIncubatingAttributes .DB_OPERATION_NAME , "INSERT" ),
394- entry (DbIncubatingAttributes .DB_COLLECTION_NAME , "potato" ));
392+ entry (DbAttributes .DB_NAMESPACE , "potatoes" ),
393+ entry (DbAttributes .DB_QUERY_TEXT , "INSERT INTO potato VALUES(?)" ),
394+ entry (DbAttributes .DB_OPERATION_NAME , "INSERT" ),
395+ entry (DbAttributes .DB_COLLECTION_NAME , "potato" ));
395396 }
396397
397398 assertThat (endAttributes .build ().isEmpty ()).isTrue ();
0 commit comments