Skip to content

Commit 4a7f845

Browse files
committed
cleaning up code
1 parent 03d99f7 commit 4a7f845

File tree

2 files changed

+9
-8
lines changed

2 files changed

+9
-8
lines changed

awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/OtlpAwsSpanExporter.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@
4848
*/
4949
@Immutable
5050
public class OtlpAwsSpanExporter implements SpanExporter {
51+
private static final String SERVICE_NAME = "xray";
5152
private static final Logger logger = LoggerFactory.getLogger(OtlpAwsSpanExporter.class);
5253

5354
private final OtlpHttpSpanExporter parentExporter;
@@ -129,7 +130,7 @@ public Map<String, String> get() {
129130
b ->
130131
b.identity(credentials)
131132
.request(httpRequest)
132-
.putProperty(AwsV4HttpSigner.SERVICE_SIGNING_NAME, "xray")
133+
.putProperty(AwsV4HttpSigner.SERVICE_SIGNING_NAME, SERVICE_NAME)
133134
.putProperty(
134135
AwsV4HttpSigner.REGION_NAME, OtlpAwsSpanExporter.this.awsRegion)
135136
.payload(() -> new ByteArrayInputStream(encodedSpans.toByteArray())));

awsagentprovider/src/test/java/software/amazon/opentelemetry/javaagent/providers/OtlpAwsSpanExporterTest.java

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949

5050
@ExtendWith(MockitoExtension.class)
5151
public class OtlpAwsSpanExporterTest {
52-
private static final String OTLP_CW_ENDPOINT = "https://xray.us-east-1.amazonaws.com/v1/traces";
52+
private static final String XRAY_OTLP_ENDPOINT = "https://xray.us-east-1.amazonaws.com/v1/traces";
5353
private static final String AUTHORIZATION_HEADER = "Authorization";
5454
private static final String X_AMZ_DATE_HEADER = "X-Amz-Date";
5555
private static final String X_AMZ_SECURITY_TOKEN_HEADER = "X-Amz-Security-Token";
@@ -65,7 +65,7 @@ public class OtlpAwsSpanExporterTest {
6565
.request(
6666
SdkHttpFullRequest.builder()
6767
.method(SdkHttpMethod.POST)
68-
.uri(URI.create(OTLP_CW_ENDPOINT))
68+
.uri(URI.create(XRAY_OTLP_ENDPOINT))
6969
.putHeader(AUTHORIZATION_HEADER, EXPECTED_AUTH_HEADER)
7070
.putHeader(X_AMZ_DATE_HEADER, EXPECTED_AUTH_X_AMZ_DATE)
7171
.putHeader(X_AMZ_SECURITY_TOKEN_HEADER, EXPECTED_AUTH_SECURITY_TOKEN)
@@ -115,7 +115,7 @@ void afterEach() {
115115
@Test
116116
void testAwsSpanExporterAddsSigV4Headers() {
117117

118-
SpanExporter exporter = new OtlpAwsSpanExporter(OTLP_CW_ENDPOINT);
118+
SpanExporter exporter = new OtlpAwsSpanExporter(XRAY_OTLP_ENDPOINT);
119119
when(this.credentialsProvider.resolveCredentials()).thenReturn(this.credentials);
120120
when(this.signer.sign((Consumer<Builder<AwsCredentialsIdentity>>) any()))
121121
.thenReturn(this.signedRequest);
@@ -135,7 +135,7 @@ void testAwsSpanExporterAddsSigV4Headers() {
135135

136136
@Test
137137
void testAwsSpanExporterExportCorrectlyAddsDifferentSigV4Headers() {
138-
SpanExporter exporter = new OtlpAwsSpanExporter(OTLP_CW_ENDPOINT);
138+
SpanExporter exporter = new OtlpAwsSpanExporter(XRAY_OTLP_ENDPOINT);
139139

140140
for (int i = 0; i < 10; i += 1) {
141141
String newAuthHeader = EXPECTED_AUTH_HEADER + i;
@@ -147,7 +147,7 @@ void testAwsSpanExporterExportCorrectlyAddsDifferentSigV4Headers() {
147147
.request(
148148
SdkHttpFullRequest.builder()
149149
.method(SdkHttpMethod.POST)
150-
.uri(URI.create(OTLP_CW_ENDPOINT))
150+
.uri(URI.create(XRAY_OTLP_ENDPOINT))
151151
.putHeader(AUTHORIZATION_HEADER, newAuthHeader)
152152
.putHeader(X_AMZ_DATE_HEADER, newXAmzDate)
153153
.putHeader(X_AMZ_SECURITY_TOKEN_HEADER, newXAmzSecurityToken)
@@ -177,7 +177,7 @@ void testAwsSpanExporterDoesNotAddSigV4HeadersIfFailureToRetrieveCredentials() {
177177
when(this.credentialsProvider.resolveCredentials())
178178
.thenThrow(SdkClientException.builder().message("bad credentials").build());
179179

180-
SpanExporter exporter = new OtlpAwsSpanExporter(OTLP_CW_ENDPOINT);
180+
SpanExporter exporter = new OtlpAwsSpanExporter(XRAY_OTLP_ENDPOINT);
181181

182182
exporter.export(List.of());
183183

@@ -198,7 +198,7 @@ void testAwsSpanExporterDoesNotAddSigV4HeadersIfFailureToSignHeaders() {
198198
when(this.signer.sign((Consumer<Builder<AwsCredentialsIdentity>>) any()))
199199
.thenThrow(SdkClientException.builder().message("bad signature").build());
200200

201-
SpanExporter exporter = new OtlpAwsSpanExporter(OTLP_CW_ENDPOINT);
201+
SpanExporter exporter = new OtlpAwsSpanExporter(XRAY_OTLP_ENDPOINT);
202202

203203
exporter.export(List.of());
204204

0 commit comments

Comments
 (0)