refactor out OtelHelper class · googleapis/java-bigquery@7443564 · GitHub | Latest TMZ Celebrity News & Gossip | Watch TMZ Live
Skip to content

Commit 7443564

Browse files
whuffman36whuffman36
committed
refactor out OtelHelper class
1 parent cc84936 commit 7443564

File tree

15 files changed

+651
-745
lines changed

15 files changed

+651
-745
lines changed

‎google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/BigQueryImpl.java

Lines changed: 419 additions & 45 deletions
Large diffs are not rendered by default.

‎google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/DatasetId.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import static com.google.common.base.Strings.isNullOrEmpty;
2121

2222
import com.google.api.services.bigquery.model.DatasetReference;
23+
import io.opentelemetry.api.common.Attributes;
2324
import java.io.Serializable;
2425
import java.util.Objects;
2526

@@ -84,4 +85,11 @@ DatasetReference toPb() {
8485
static DatasetId fromPb(DatasetReference datasetRef) {
8586
return new DatasetId(datasetRef.getProjectId(), datasetRef.getDatasetId());
8687
}
88+
89+
public Attributes getOtelAttributes() {
90+
return Attributes.builder()
91+
.put("project", this.getProject())
92+
.put("dataset", this.getDataset())
93+
.build();
94+
}
8795
}

‎google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/DatasetInfo.java

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import com.google.common.base.MoreObjects;
2828
import com.google.common.collect.ImmutableList;
2929
import com.google.common.collect.Lists;
30+
import io.opentelemetry.api.common.Attributes;
3031
import java.io.Serializable;
3132
import java.util.List;
3233
import java.util.Map;
@@ -746,6 +747,40 @@ public static DatasetInfo of(String datasetId) {
746747
return newBuilder(datasetId).build();
747748
}
748749

750+
public Attributes getOtelAttributes() {
751+
return Attributes.builder()
752+
.put("datasetId", OpenTelemetryHelper.getFieldAsString(this.getDatasetId().getDataset()))
753+
.put(
754+
"defaultTableLifetime",
755+
OpenTelemetryHelper.getFieldAsString(this.getDefaultTableLifetime()))
756+
.put("description", OpenTelemetryHelper.getFieldAsString(this.getDescription()))
757+
.put("etag", OpenTelemetryHelper.getFieldAsString(this.getEtag()))
758+
.put("friendlyName", OpenTelemetryHelper.getFieldAsString(this.getFriendlyName()))
759+
.put("generatedId", OpenTelemetryHelper.getFieldAsString(this.getGeneratedId()))
760+
.put("lastModified", OpenTelemetryHelper.getFieldAsString(this.getLastModified()))
761+
.put("location", OpenTelemetryHelper.getFieldAsString(this.getLocation()))
762+
.put("selfLink", OpenTelemetryHelper.getFieldAsString(this.getSelfLink()))
763+
.put("labels", OpenTelemetryHelper.getFieldAsString(this.getLabels()))
764+
.put(
765+
"defaultEncryptionConfiguration",
766+
OpenTelemetryHelper.getFieldAsString(this.getDefaultEncryptionConfiguration()))
767+
.put(
768+
"defaultPartitionExpirationMs",
769+
OpenTelemetryHelper.getFieldAsString(this.getDefaultPartitionExpirationMs()))
770+
.put("defaultCollation", OpenTelemetryHelper.getFieldAsString(this.getDefaultCollation()))
771+
.put(
772+
"externalDatasetReference",
773+
OpenTelemetryHelper.getFieldAsString(this.getExternalDatasetReference()))
774+
.put(
775+
"storageBillingModel",
776+
OpenTelemetryHelper.getFieldAsString(this.getStorageBillingModel()))
777+
.put(
778+
"maxTimeTravelHours",
779+
OpenTelemetryHelper.getFieldAsString(this.getMaxTimeTravelHours()))
780+
.put("resourceTags", OpenTelemetryHelper.getFieldAsString(this.getResourceTags()))
781+
.build();
782+
}
783+
749784
static DatasetInfo fromPb(Dataset datasetPb) {
750785
return new BuilderImpl(datasetPb).build();
751786
}

‎google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/InsertAllRequest.java

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import com.google.common.collect.ImmutableList;
2323
import com.google.common.collect.ImmutableMap;
2424
import com.google.common.collect.Lists;
25+
import io.opentelemetry.api.common.Attributes;
2526
import java.io.Serializable;
2627
import java.util.Collections;
2728
import java.util.HashMap;
@@ -479,4 +480,15 @@ public boolean equals(Object obj) {
479480
&& Objects.equals(skipInvalidRows, other.skipInvalidRows)
480481
&& Objects.equals(templateSuffix, other.templateSuffix);
481482
}
483+
484+
public Attributes getOtelAttributes() {
485+
return Attributes.builder()
486+
.put("table", OpenTelemetryHelper.getFieldAsString(this.getTable().getTable()))
487+
.put("row", OpenTelemetryHelper.getFieldAsString(this.getRows()))
488+
.put("templateSuffix", OpenTelemetryHelper.getFieldAsString(this.getTemplateSuffix()))
489+
.put(
490+
"ignoreUnknownValues", OpenTelemetryHelper.getFieldAsString(this.ignoreUnknownValues()))
491+
.put("skipInvalidRows", OpenTelemetryHelper.getFieldAsString(this.skipInvalidRows()))
492+
.build();
493+
}
482494
}

‎google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/JobId.java

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222
import com.google.api.services.bigquery.model.JobReference;
2323
import com.google.auto.value.AutoValue;
24+
import io.opentelemetry.api.common.Attributes;
2425
import java.io.Serializable;
2526
import java.util.UUID;
2627
import javax.annotation.Nullable;
@@ -123,4 +124,12 @@ static JobId fromPb(JobReference jobRef) {
123124
.setLocation(jobRef.getLocation())
124125
.build();
125126
}
127+
128+
public Attributes getOtelAttributes() {
129+
return Attributes.builder()
130+
.put("job", OpenTelemetryHelper.getFieldAsString(this.getJob()))
131+
.put("location", OpenTelemetryHelper.getFieldAsString(this.getLocation()))
132+
.put("project", OpenTelemetryHelper.getFieldAsString(this.getProject()))
133+
.build();
134+
}
126135
}

‎google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/JobInfo.java

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import com.google.api.services.bigquery.model.Job;
2020
import com.google.common.base.Function;
2121
import com.google.common.base.MoreObjects;
22+
import io.opentelemetry.api.common.Attributes;
2223
import java.io.Serializable;
2324
import java.util.Objects;
2425

@@ -375,4 +376,17 @@ public static JobInfo of(JobId jobId, JobConfiguration configuration) {
375376
static JobInfo fromPb(Job jobPb) {
376377
return new BuilderImpl(jobPb).build();
377378
}
379+
380+
public Attributes getOtelAttributes() {
381+
return Attributes.builder()
382+
.put("etag", OpenTelemetryHelper.getFieldAsString(this.getEtag()))
383+
.put("generatedId", OpenTelemetryHelper.getFieldAsString(this.getGeneratedId()))
384+
.put("jobId", OpenTelemetryHelper.getFieldAsString(this.getJobId().getJob()))
385+
.put("selfLink", OpenTelemetryHelper.getFieldAsString(this.getSelfLink()))
386+
.put("status", OpenTelemetryHelper.getFieldAsString(this.getStatus()))
387+
.put("statistics", OpenTelemetryHelper.getFieldAsString(this.getStatistics()))
388+
.put("userEmail", OpenTelemetryHelper.getFieldAsString(this.getUserEmail()))
389+
.put("configuration", OpenTelemetryHelper.getFieldAsString(this.getConfiguration()))
390+
.build();
391+
}
378392
}

‎google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/ModelId.java

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222
import com.google.api.services.bigquery.model.ModelReference;
2323
import com.google.common.base.Function;
24+
import io.opentelemetry.api.common.Attributes;
2425
import java.io.Serializable;
2526
import java.util.Objects;
2627

@@ -105,4 +106,12 @@ ModelReference toPb() {
105106
static ModelId fromPb(ModelReference modelRef) {
106107
return new ModelId(modelRef.getProjectId(), modelRef.getDatasetId(), modelRef.getModelId());
107108
}
109+
110+
public Attributes getOtelAttributes() {
111+
return Attributes.builder()
112+
.put("project", this.getProject())
113+
.put("dataset", this.getDataset())
114+
.put("model", this.getModel())
115+
.build();
116+
}
108117
}

‎google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/ModelInfo.java

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
import com.google.common.base.Strings;
2727
import com.google.common.collect.ImmutableList;
2828
import com.google.common.collect.Lists;
29+
import io.opentelemetry.api.common.Attributes;
2930
import java.io.Serializable;
3031
import java.util.Collections;
3132
import java.util.List;
@@ -453,4 +454,25 @@ Model toPb() {
453454
static ModelInfo fromPb(Model modelPb) {
454455
return new BuilderImpl(modelPb).build();
455456
}
457+
458+
public Attributes getOtelAttributes() {
459+
return Attributes.builder()
460+
.put("etag", OpenTelemetryHelper.getFieldAsString(this.getEtag()))
461+
.put("modelId", OpenTelemetryHelper.getFieldAsString(this.getModelId().getModel()))
462+
.put("description", OpenTelemetryHelper.getFieldAsString(this.getDescription()))
463+
.put("modelType", OpenTelemetryHelper.getFieldAsString(this.getModelType()))
464+
.put("friendlyName", OpenTelemetryHelper.getFieldAsString(this.getFriendlyName()))
465+
.put("creationTime", OpenTelemetryHelper.getFieldAsString(this.getCreationTime()))
466+
.put("lastModifiedTime", OpenTelemetryHelper.getFieldAsString(this.getLastModifiedTime()))
467+
.put("expirationTime", OpenTelemetryHelper.getFieldAsString(this.getExpirationTime()))
468+
.put("labels", OpenTelemetryHelper.getFieldAsString(this.getLabels()))
469+
.put("location", OpenTelemetryHelper.getFieldAsString(this.getLocation()))
470+
.put("trainingRunList", OpenTelemetryHelper.getFieldAsString(this.getTrainingRuns()))
471+
.put("featureColumnList", OpenTelemetryHelper.getFieldAsString(this.getFeatureColumns()))
472+
.put("labelColumnList", OpenTelemetryHelper.getFieldAsString(this.getLabelColumns()))
473+
.put(
474+
"encryptionConfiguration",
475+
OpenTelemetryHelper.getFieldAsString(this.getEncryptionConfiguration()))
476+
.build();
477+
}
456478
}

0 commit comments

Comments
 (0)

TMZ Celebrity News – Breaking Stories, Videos & Gossip

Looking for the latest TMZ celebrity news? You've come to the right place. From shocking Hollywood scandals to exclusive videos, TMZ delivers it all in real time.

Whether it’s a red carpet slip-up, a viral paparazzi moment, or a legal drama involving your favorite stars, TMZ news is always first to break the story. Stay in the loop with daily updates, insider tips, and jaw-dropping photos.

🎥 Watch TMZ Live

TMZ Live brings you daily celebrity news and interviews straight from the TMZ newsroom. Don’t miss a beat—watch now and see what’s trending in Hollywood.