|
| 1 | +/* |
| 2 | + * Copyright The OpenTelemetry Authors |
| 3 | + * SPDX-License-Identifier: Apache-2.0 |
| 4 | + */ |
| 5 | + |
1 | 6 | package io.opentelemetry.instrumentation.awssdk.v1_11;
|
2 | 7 |
|
3 | 8 | import com.amazonaws.Request;
|
|
7 | 12 | import io.opentelemetry.context.Context;
|
8 | 13 | import io.opentelemetry.instrumentation.api.instrumenter.AttributesExtractor;
|
9 | 14 | import io.opentelemetry.instrumentation.api.internal.AttributesExtractorUtil;
|
10 |
| -import javax.annotation.Nullable; |
11 | 15 | import java.util.Collections;
|
12 | 16 | import java.util.List;
|
| 17 | +import javax.annotation.Nullable; |
13 | 18 |
|
14 | 19 | public class DynamoAttributesExtractor implements AttributesExtractor<Request<?>, Response<?>> {
|
15 | 20 |
|
16 |
| - private static final AttributeKey<String> DB_SYSTEM = |
17 |
| - AttributeKey.stringKey("db.system"); |
| 21 | + private static final AttributeKey<String> DB_SYSTEM = AttributeKey.stringKey("db.system"); |
18 | 22 | private static final AttributeKey<List<String>> AWS_TABLE_NAMES =
|
19 | 23 | AttributeKey.stringArrayKey("aws.dynamodb.table_names");
|
20 | 24 |
|
21 | 25 | private static final String DYNAMODB = "dynamodb";
|
22 | 26 |
|
23 | 27 | @Override
|
24 | 28 | public void onStart(AttributesBuilder attributes, Context parentContext, Request<?> request) {
|
25 |
| - AttributesExtractorUtil.internalSet(attributes,DB_SYSTEM,DYNAMODB); |
| 29 | + AttributesExtractorUtil.internalSet(attributes, DB_SYSTEM, DYNAMODB); |
26 | 30 | String tableName = RequestAccess.getTableName(request.getOriginalRequest());
|
27 |
| - AttributesExtractorUtil.internalSet(attributes,AWS_TABLE_NAMES, Collections.singletonList(tableName)); |
28 |
| - } |
| 31 | + AttributesExtractorUtil.internalSet( |
| 32 | + attributes, AWS_TABLE_NAMES, Collections.singletonList(tableName)); |
| 33 | + } |
29 | 34 |
|
30 | 35 | @Override
|
31 | 36 | public void onEnd(
|
|
0 commit comments