Skip to content

Commit d0ea9d3

Browse files
committed
minor cleanup tasks
1 parent be995bb commit d0ea9d3

File tree

4 files changed

+24
-11
lines changed

4 files changed

+24
-11
lines changed

java-bigquery/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/BigQueryRetryAlgorithm.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,6 @@ public boolean shouldRetry(
8282
// Store retry attempt count in thread-local storage for tracing
8383
CURRENT_ATTEMPT.set(attemptCount);
8484

85-
8685
if (LOG.isLoggable(Level.FINEST)) {
8786
LOG.log(
8887
Level.FINEST,

java-bigquery/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/telemetry/HttpTracingRequestInitializer.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,10 @@
1616

1717
package com.google.cloud.bigquery.telemetry;
1818

19-
import com.google.cloud.bigquery.BigQueryRetryAlgorithm;
2019
import com.google.api.client.http.*;
2120
import com.google.api.core.BetaApi;
2221
import com.google.api.core.InternalApi;
22+
import com.google.cloud.bigquery.BigQueryRetryAlgorithm;
2323
import com.google.common.annotations.VisibleForTesting;
2424
import io.opentelemetry.api.common.AttributeKey;
2525
import io.opentelemetry.api.trace.Span;
@@ -93,7 +93,6 @@ public void initialize(HttpRequest request) throws IOException {
9393
HttpUnsuccessfulResponseHandler originalHandler = request.getUnsuccessfulResponseHandler();
9494
request.setUnsuccessfulResponseHandler(
9595
(request1, response, supportsRetry) -> {
96-
System.out.println("laura2 did I get in here for auth?" );
9796
addCommonResponseAttributesToSpan(request1, response, span);
9897
if (originalHandler != null) {
9998
return originalHandler.handleResponse(request1, response, supportsRetry);

java-bigquery/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/spi/v2/HttpBigQueryRpcTest.java

Lines changed: 23 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1122,11 +1122,20 @@ public void testResendCountOnRetry() throws Exception {
11221122

11231123
List<io.opentelemetry.sdk.trace.data.SpanData> spans = spanExporter.getFinishedSpanItems();
11241124
assertThat(spans).isNotEmpty();
1125-
io.opentelemetry.sdk.trace.data.SpanData rpcSpan = spans.stream().filter(s -> s.getName().equals("com.google.cloud.bigquery.BigQueryRpc.getDataset")).findFirst().orElse(null);
1125+
io.opentelemetry.sdk.trace.data.SpanData rpcSpan =
1126+
spans.stream()
1127+
.filter(s -> s.getName().equals("com.google.cloud.bigquery.BigQueryRpc.getDataset"))
1128+
.findFirst()
1129+
.orElse(null);
11261130
assertNotNull(rpcSpan);
1127-
assertEquals(2L, rpcSpan.getAttributes().get(com.google.cloud.bigquery.telemetry.HttpTracingRequestInitializer.HTTP_REQUEST_RESEND_COUNT));
1131+
assertEquals(
1132+
2L,
1133+
rpcSpan
1134+
.getAttributes()
1135+
.get(
1136+
com.google.cloud.bigquery.telemetry.HttpTracingRequestInitializer
1137+
.HTTP_REQUEST_RESEND_COUNT));
11281138

1129-
// Cleanup
11301139
com.google.cloud.bigquery.BigQueryRetryAlgorithm.CURRENT_ATTEMPT.remove();
11311140
}
11321141
}
@@ -1248,11 +1257,19 @@ public void testResendCountNotSetWhenDisabled() throws Exception {
12481257

12491258
List<io.opentelemetry.sdk.trace.data.SpanData> spans = spanExporter.getFinishedSpanItems();
12501259
assertThat(spans).isNotEmpty();
1251-
io.opentelemetry.sdk.trace.data.SpanData rpcSpan = spans.stream().filter(s -> s.getName().equals("com.google.cloud.bigquery.BigQueryRpc.getDataset")).findFirst().orElse(null);
1260+
io.opentelemetry.sdk.trace.data.SpanData rpcSpan =
1261+
spans.stream()
1262+
.filter(s -> s.getName().equals("com.google.cloud.bigquery.BigQueryRpc.getDataset"))
1263+
.findFirst()
1264+
.orElse(null);
12521265
assertNotNull(rpcSpan);
1253-
assertNull(rpcSpan.getAttributes().get(com.google.cloud.bigquery.telemetry.HttpTracingRequestInitializer.HTTP_REQUEST_RESEND_COUNT));
1266+
assertNull(
1267+
rpcSpan
1268+
.getAttributes()
1269+
.get(
1270+
com.google.cloud.bigquery.telemetry.HttpTracingRequestInitializer
1271+
.HTTP_REQUEST_RESEND_COUNT));
12541272

1255-
// Cleanup
12561273
com.google.cloud.bigquery.BigQueryRetryAlgorithm.CURRENT_ATTEMPT.remove();
12571274
}
12581275
}

java-bigquery/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/telemetry/HttpTracingRequestInitializerTest.java

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -335,7 +335,6 @@ public void testAddResponseBodySizeToSpan_NullLength() throws IOException {
335335
assertNull(span.getAttributes().get(HttpTracingRequestInitializer.HTTP_RESPONSE_BODY_SIZE));
336336
}
337337

338-
339338
@Test
340339
public void testResendCountIsSetFromBigQueryRetryAlgorithm() throws IOException {
341340
com.google.cloud.bigquery.BigQueryRetryAlgorithm.CURRENT_ATTEMPT.set(3);
@@ -354,7 +353,6 @@ public void testResendCountIsSetFromBigQueryRetryAlgorithm() throws IOException
354353
assertEquals(
355354
3L, span.getAttributes().get(HttpTracingRequestInitializer.HTTP_REQUEST_RESEND_COUNT));
356355

357-
// Cleanup to prevent leaks
358356
com.google.cloud.bigquery.BigQueryRetryAlgorithm.CURRENT_ATTEMPT.remove();
359357
}
360358

0 commit comments

Comments
 (0)