feat(bigquery): Integrate Otel into retries, jobs, and more (#3842) · googleapis/java-bigquery@4b28c47
@@ -295,7 +295,9 @@ public com.google.api.services.bigquery.model.Dataset call() throws IOException
295295getOptions().getRetrySettings(),
296296BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
297297getOptions().getClock(),
298-EMPTY_RETRY_CONFIG));
298+EMPTY_RETRY_CONFIG,
299+getOptions().isOpenTelemetryTracingEnabled(),
300+getOptions().getOpenTelemetryTracer()));
299301 } catch (BigQueryRetryHelperException e) {
300302throw BigQueryException.translateAndThrow(e);
301303 } finally {
@@ -340,7 +342,9 @@ public com.google.api.services.bigquery.model.Table call() throws IOException {
340342getOptions().getRetrySettings(),
341343BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
342344getOptions().getClock(),
343-EMPTY_RETRY_CONFIG));
345+EMPTY_RETRY_CONFIG,
346+getOptions().isOpenTelemetryTracingEnabled(),
347+getOptions().getOpenTelemetryTracer()));
344348 } catch (BigQueryRetryHelperException e) {
345349throw BigQueryException.translateAndThrow(e);
346350 } finally {
@@ -394,7 +398,9 @@ public com.google.api.services.bigquery.model.Routine call() throws IOException
394398getOptions().getRetrySettings(),
395399BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
396400getOptions().getClock(),
397-EMPTY_RETRY_CONFIG));
401+EMPTY_RETRY_CONFIG,
402+getOptions().isOpenTelemetryTracingEnabled(),
403+getOptions().getOpenTelemetryTracer()));
398404 } catch (BigQueryRetryHelperException e) {
399405throw BigQueryException.translateAndThrow(e);
400406 } finally {
@@ -488,7 +494,9 @@ public com.google.api.services.bigquery.model.Job call() throws IOException {
488494getOptions().getClock(),
489495getBigQueryRetryConfig(optionsMap) != null
490496 ? getBigQueryRetryConfig(optionsMap)
491- : DEFAULT_RETRY_CONFIG));
497+ : DEFAULT_RETRY_CONFIG,
498+getOptions().isOpenTelemetryTracingEnabled(),
499+getOptions().getOpenTelemetryTracer()));
492500 } catch (BigQueryRetryHelperException e) {
493501throw BigQueryException.translateAndThrow(e);
494502 }
@@ -580,7 +588,9 @@ public com.google.api.services.bigquery.model.Dataset call() throws IOException
580588getOptions().getRetrySettings(),
581589BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
582590getOptions().getClock(),
583-EMPTY_RETRY_CONFIG);
591+EMPTY_RETRY_CONFIG,
592+getOptions().isOpenTelemetryTracingEnabled(),
593+getOptions().getOpenTelemetryTracer());
584594return Dataset.fromPb(this, answer);
585595 } catch (BigQueryRetryHelperException e) {
586596if (isRetryErrorCodeHttpNotFound(e)) {
@@ -644,7 +654,9 @@ private static Page<Dataset> listDatasets(
644654serviceOptions.getRetrySettings(),
645655BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
646656serviceOptions.getClock(),
647-EMPTY_RETRY_CONFIG);
657+EMPTY_RETRY_CONFIG,
658+serviceOptions.isOpenTelemetryTracingEnabled(),
659+serviceOptions.getOpenTelemetryTracer());
648660String cursor = result.x();
649661return new PageImpl<>(
650662new DatasetPageFetcher(projectId, serviceOptions, cursor, optionsMap),
@@ -694,7 +706,9 @@ public Boolean call() throws IOException {
694706getOptions().getRetrySettings(),
695707BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
696708getOptions().getClock(),
697-EMPTY_RETRY_CONFIG);
709+EMPTY_RETRY_CONFIG,
710+getOptions().isOpenTelemetryTracingEnabled(),
711+getOptions().getOpenTelemetryTracer());
698712 } catch (BigQueryRetryHelperException e) {
699713if (isRetryErrorCodeHttpNotFound(e)) {
700714return false;
@@ -743,7 +757,9 @@ public Boolean call() throws IOException {
743757getOptions().getRetrySettings(),
744758BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
745759getOptions().getClock(),
746-EMPTY_RETRY_CONFIG);
760+EMPTY_RETRY_CONFIG,
761+getOptions().isOpenTelemetryTracingEnabled(),
762+getOptions().getOpenTelemetryTracer());
747763 } catch (BigQueryRetryHelperException e) {
748764if (isRetryErrorCodeHttpNotFound(e)) {
749765return false;
@@ -787,7 +803,9 @@ public Boolean call() throws IOException {
787803getOptions().getRetrySettings(),
788804BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
789805getOptions().getClock(),
790-EMPTY_RETRY_CONFIG);
806+EMPTY_RETRY_CONFIG,
807+getOptions().isOpenTelemetryTracingEnabled(),
808+getOptions().getOpenTelemetryTracer());
791809 } catch (BigQueryRetryHelperException e) {
792810if (isRetryErrorCodeHttpNotFound(e)) {
793811return false;
@@ -831,7 +849,9 @@ public Boolean call() throws IOException {
831849getOptions().getRetrySettings(),
832850BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
833851getOptions().getClock(),
834-EMPTY_RETRY_CONFIG);
852+EMPTY_RETRY_CONFIG,
853+getOptions().isOpenTelemetryTracingEnabled(),
854+getOptions().getOpenTelemetryTracer());
835855 } catch (BigQueryRetryHelperException e) {
836856if (isRetryErrorCodeHttpNotFound(e)) {
837857return false;
@@ -873,7 +893,9 @@ public Boolean call() throws IOException {
873893getOptions().getRetrySettings(),
874894BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
875895getOptions().getClock(),
876-EMPTY_RETRY_CONFIG);
896+EMPTY_RETRY_CONFIG,
897+getOptions().isOpenTelemetryTracingEnabled(),
898+getOptions().getOpenTelemetryTracer());
877899 } catch (BigQueryRetryHelperException e) {
878900throw BigQueryException.translateAndThrow(e);
879901 } finally {
@@ -912,7 +934,9 @@ public com.google.api.services.bigquery.model.Dataset call() throws IOException
912934getOptions().getRetrySettings(),
913935BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
914936getOptions().getClock(),
915-EMPTY_RETRY_CONFIG));
937+EMPTY_RETRY_CONFIG,
938+getOptions().isOpenTelemetryTracingEnabled(),
939+getOptions().getOpenTelemetryTracer()));
916940 } catch (BigQueryRetryHelperException e) {
917941throw BigQueryException.translateAndThrow(e);
918942 } finally {
@@ -957,7 +981,9 @@ public com.google.api.services.bigquery.model.Table call() throws IOException {
957981getOptions().getRetrySettings(),
958982BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
959983getOptions().getClock(),
960-EMPTY_RETRY_CONFIG));
984+EMPTY_RETRY_CONFIG,
985+getOptions().isOpenTelemetryTracingEnabled(),
986+getOptions().getOpenTelemetryTracer()));
961987 } catch (BigQueryRetryHelperException e) {
962988throw BigQueryException.translateAndThrow(e);
963989 } finally {
@@ -1001,7 +1027,9 @@ public com.google.api.services.bigquery.model.Model call() throws IOException {
10011027getOptions().getRetrySettings(),
10021028BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
10031029getOptions().getClock(),
1004-EMPTY_RETRY_CONFIG));
1030+EMPTY_RETRY_CONFIG,
1031+getOptions().isOpenTelemetryTracingEnabled(),
1032+getOptions().getOpenTelemetryTracer()));
10051033 } catch (BigQueryRetryHelperException e) {
10061034throw BigQueryException.translateAndThrow(e);
10071035 } finally {
@@ -1045,7 +1073,9 @@ public com.google.api.services.bigquery.model.Routine call() throws IOException
10451073getOptions().getRetrySettings(),
10461074BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
10471075getOptions().getClock(),
1048-EMPTY_RETRY_CONFIG));
1076+EMPTY_RETRY_CONFIG,
1077+getOptions().isOpenTelemetryTracingEnabled(),
1078+getOptions().getOpenTelemetryTracer()));
10491079 } catch (BigQueryRetryHelperException e) {
10501080throw BigQueryException.translateAndThrow(e);
10511081 } finally {
@@ -1097,7 +1127,9 @@ public com.google.api.services.bigquery.model.Table call() throws IOException {
10971127getOptions().getRetrySettings(),
10981128BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
10991129getOptions().getClock(),
1100-EMPTY_RETRY_CONFIG);
1130+EMPTY_RETRY_CONFIG,
1131+getOptions().isOpenTelemetryTracingEnabled(),
1132+getOptions().getOpenTelemetryTracer());
11011133return Table.fromPb(this, answer);
11021134 } catch (BigQueryRetryHelperException e) {
11031135if (isRetryErrorCodeHttpNotFound(e)) {
@@ -1154,7 +1186,9 @@ public com.google.api.services.bigquery.model.Model call() throws IOException {
11541186getOptions().getRetrySettings(),
11551187BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
11561188getOptions().getClock(),
1157-EMPTY_RETRY_CONFIG);
1189+EMPTY_RETRY_CONFIG,
1190+getOptions().isOpenTelemetryTracingEnabled(),
1191+getOptions().getOpenTelemetryTracer());
11581192return Model.fromPb(this, answer);
11591193 } catch (BigQueryRetryHelperException e) {
11601194if (isRetryErrorCodeHttpNotFound(e)) {
@@ -1211,7 +1245,9 @@ public com.google.api.services.bigquery.model.Routine call() throws IOException
12111245getOptions().getRetrySettings(),
12121246BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
12131247getOptions().getClock(),
1214-EMPTY_RETRY_CONFIG);
1248+EMPTY_RETRY_CONFIG,
1249+getOptions().isOpenTelemetryTracingEnabled(),
1250+getOptions().getOpenTelemetryTracer());
12151251return Routine.fromPb(this, answer);
12161252 } catch (BigQueryRetryHelperException e) {
12171253if (isRetryErrorCodeHttpNotFound(e)) {
@@ -1427,7 +1463,9 @@ public Tuple<String, Iterable<com.google.api.services.bigquery.model.Table>> cal
14271463serviceOptions.getRetrySettings(),
14281464BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
14291465serviceOptions.getClock(),
1430-EMPTY_RETRY_CONFIG);
1466+EMPTY_RETRY_CONFIG,
1467+serviceOptions.isOpenTelemetryTracingEnabled(),
1468+serviceOptions.getOpenTelemetryTracer());
14311469String cursor = result.x();
14321470Iterable<Table> tables =
14331471Iterables.transform(
@@ -1466,7 +1504,9 @@ public Tuple<String, Iterable<com.google.api.services.bigquery.model.Model>> cal
14661504serviceOptions.getRetrySettings(),
14671505BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
14681506serviceOptions.getClock(),
1469-EMPTY_RETRY_CONFIG);
1507+EMPTY_RETRY_CONFIG,
1508+serviceOptions.isOpenTelemetryTracingEnabled(),
1509+serviceOptions.getOpenTelemetryTracer());
14701510String cursor = result.x();
14711511Iterable<Model> models =
14721512Iterables.transform(
@@ -1505,7 +1545,9 @@ private static Page<Routine> listRoutines(
15051545serviceOptions.getRetrySettings(),
15061546BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
15071547serviceOptions.getClock(),
1508-EMPTY_RETRY_CONFIG);
1548+EMPTY_RETRY_CONFIG,
1549+serviceOptions.isOpenTelemetryTracingEnabled(),
1550+serviceOptions.getOpenTelemetryTracer());
15091551String cursor = result.x();
15101552Iterable<Routine> routines =
15111553Iterables.transform(
@@ -1585,7 +1627,9 @@ public TableDataInsertAllResponse call() throws Exception {
15851627getOptions().getRetrySettings(),
15861628BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
15871629getOptions().getClock(),
1588-EMPTY_RETRY_CONFIG);
1630+EMPTY_RETRY_CONFIG,
1631+getOptions().isOpenTelemetryTracingEnabled(),
1632+getOptions().getOpenTelemetryTracer());
15891633 } catch (BigQueryRetryHelperException e) {
15901634throw BigQueryException.translateAndThrow(e);
15911635 }
@@ -1677,7 +1721,9 @@ public TableDataList call() throws IOException {
16771721serviceOptions.getRetrySettings(),
16781722BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
16791723serviceOptions.getClock(),
1680-EMPTY_RETRY_CONFIG);
1724+EMPTY_RETRY_CONFIG,
1725+serviceOptions.isOpenTelemetryTracingEnabled(),
1726+serviceOptions.getOpenTelemetryTracer());
16811727String cursor = result.getPageToken();
16821728Map<BigQueryRpc.Option, ?> pageOptionMap =
16831729Strings.isNullOrEmpty(cursor) ? optionsMap : optionMap(TableDataListOption.startIndex(0));
@@ -1749,7 +1795,9 @@ public com.google.api.services.bigquery.model.Job call() throws IOException {
17491795getOptions().getRetrySettings(),
17501796BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
17511797getOptions().getClock(),
1752-EMPTY_RETRY_CONFIG);
1798+EMPTY_RETRY_CONFIG,
1799+getOptions().isOpenTelemetryTracingEnabled(),
1800+getOptions().getOpenTelemetryTracer());
17531801return Job.fromPb(this, answer);
17541802 } catch (BigQueryRetryHelperException e) {
17551803if (isRetryErrorCodeHttpNotFound(e)) {
@@ -1804,7 +1852,9 @@ public Tuple<String, Iterable<com.google.api.services.bigquery.model.Job>> call(
18041852serviceOptions.getRetrySettings(),
18051853BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
18061854serviceOptions.getClock(),
1807-EMPTY_RETRY_CONFIG);
1855+EMPTY_RETRY_CONFIG,
1856+serviceOptions.isOpenTelemetryTracingEnabled(),
1857+serviceOptions.getOpenTelemetryTracer());
18081858String cursor = result.x();
18091859Iterable<Job> jobs =
18101860Iterables.transform(
@@ -1857,7 +1907,9 @@ public Boolean call() throws IOException {
18571907getOptions().getRetrySettings(),
18581908BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
18591909getOptions().getClock(),
1860-EMPTY_RETRY_CONFIG);
1910+EMPTY_RETRY_CONFIG,
1911+getOptions().isOpenTelemetryTracingEnabled(),
1912+getOptions().getOpenTelemetryTracer());
18611913 } catch (BigQueryRetryHelperException e) {
18621914if (isRetryErrorCodeHttpNotFound(e)) {
18631915return false;
@@ -1942,7 +1994,9 @@ public com.google.api.services.bigquery.model.QueryResponse call()
19421994getOptions().getRetrySettings(),
19431995BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
19441996getOptions().getClock(),
1945-DEFAULT_RETRY_CONFIG);
1997+DEFAULT_RETRY_CONFIG,
1998+getOptions().isOpenTelemetryTracingEnabled(),
1999+getOptions().getOpenTelemetryTracer());
19462000 } catch (BigQueryRetryHelper.BigQueryRetryHelperException e) {
19472001throw BigQueryException.translateAndThrow(e);
19482002 } finally {
@@ -2117,7 +2171,9 @@ public GetQueryResultsResponse call() throws IOException {
21172171serviceOptions.getRetrySettings(),
21182172BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
21192173serviceOptions.getClock(),
2120-DEFAULT_RETRY_CONFIG);
2174+DEFAULT_RETRY_CONFIG,
2175+serviceOptions.isOpenTelemetryTracingEnabled(),
2176+serviceOptions.getOpenTelemetryTracer());
2121217721222178TableSchema schemaPb = results.getSchema();
21232179@@ -2186,7 +2242,9 @@ public com.google.api.services.bigquery.model.Policy call() throws IOException {
21862242getOptions().getRetrySettings(),
21872243BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
21882244getOptions().getClock(),
2189-EMPTY_RETRY_CONFIG));
2245+EMPTY_RETRY_CONFIG,
2246+getOptions().isOpenTelemetryTracingEnabled(),
2247+getOptions().getOpenTelemetryTracer()));
21902248 } catch (BigQueryRetryHelperException e) {
21912249throw BigQueryException.translateAndThrow(e);
21922250 } finally {
@@ -2230,7 +2288,9 @@ public com.google.api.services.bigquery.model.Policy call() throws IOException {
22302288getOptions().getRetrySettings(),
22312289BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
22322290getOptions().getClock(),
2233-EMPTY_RETRY_CONFIG));
2291+EMPTY_RETRY_CONFIG,
2292+getOptions().isOpenTelemetryTracingEnabled(),
2293+getOptions().getOpenTelemetryTracer()));
22342294 } catch (BigQueryRetryHelperException e) {
22352295throw BigQueryException.translateAndThrow(e);
22362296 } finally {
@@ -2276,7 +2336,9 @@ public com.google.api.services.bigquery.model.TestIamPermissionsResponse call()
22762336getOptions().getRetrySettings(),
22772337BigQueryBaseService.BIGQUERY_EXCEPTION_HANDLER,
22782338getOptions().getClock(),
2279-EMPTY_RETRY_CONFIG);
2339+EMPTY_RETRY_CONFIG,
2340+getOptions().isOpenTelemetryTracingEnabled(),
2341+getOptions().getOpenTelemetryTracer());
22802342return response.getPermissions() == null
22812343 ? ImmutableList.of()
22822344 : ImmutableList.copyOf(response.getPermissions());