refactor out OtelHelper class

Author: whuffman36

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

@@ -20,6 +20,7 @@
 import static com.google.common.base.Strings.isNullOrEmpty;
 
 import com.google.api.services.bigquery.model.DatasetReference;
+import io.opentelemetry.api.common.Attributes;
 import java.io.Serializable;
 import java.util.Objects;
 
@@ -84,4 +85,11 @@ DatasetReference toPb() {
   static DatasetId fromPb(DatasetReference datasetRef) {
     return new DatasetId(datasetRef.getProjectId(), datasetRef.getDatasetId());
   }
+
+  public Attributes getOtelAttributes() {
+    return Attributes.builder()
+        .put("project", this.getProject())
+        .put("dataset", this.getDataset())
+        .build();
+  }
 }

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

@@ -27,6 +27,7 @@
 import com.google.common.base.MoreObjects;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Lists;
+import io.opentelemetry.api.common.Attributes;
 import java.io.Serializable;
 import java.util.List;
 import java.util.Map;
@@ -746,6 +747,40 @@ public static DatasetInfo of(String datasetId) {
     return newBuilder(datasetId).build();
   }
 
+  public Attributes getOtelAttributes() {
+    return Attributes.builder()
+        .put("datasetId", OpenTelemetryHelper.getFieldAsString(this.getDatasetId().getDataset()))
+        .put(
+            "defaultTableLifetime",
+            OpenTelemetryHelper.getFieldAsString(this.getDefaultTableLifetime()))
+        .put("description", OpenTelemetryHelper.getFieldAsString(this.getDescription()))
+        .put("etag", OpenTelemetryHelper.getFieldAsString(this.getEtag()))
+        .put("friendlyName", OpenTelemetryHelper.getFieldAsString(this.getFriendlyName()))
+        .put("generatedId", OpenTelemetryHelper.getFieldAsString(this.getGeneratedId()))
+        .put("lastModified", OpenTelemetryHelper.getFieldAsString(this.getLastModified()))
+        .put("location", OpenTelemetryHelper.getFieldAsString(this.getLocation()))
+        .put("selfLink", OpenTelemetryHelper.getFieldAsString(this.getSelfLink()))
+        .put("labels", OpenTelemetryHelper.getFieldAsString(this.getLabels()))
+        .put(
+            "defaultEncryptionConfiguration",
+            OpenTelemetryHelper.getFieldAsString(this.getDefaultEncryptionConfiguration()))
+        .put(
+            "defaultPartitionExpirationMs",
+            OpenTelemetryHelper.getFieldAsString(this.getDefaultPartitionExpirationMs()))
+        .put("defaultCollation", OpenTelemetryHelper.getFieldAsString(this.getDefaultCollation()))
+        .put(
+            "externalDatasetReference",
+            OpenTelemetryHelper.getFieldAsString(this.getExternalDatasetReference()))
+        .put(
+            "storageBillingModel",
+            OpenTelemetryHelper.getFieldAsString(this.getStorageBillingModel()))
+        .put(
+            "maxTimeTravelHours",
+            OpenTelemetryHelper.getFieldAsString(this.getMaxTimeTravelHours()))
+        .put("resourceTags", OpenTelemetryHelper.getFieldAsString(this.getResourceTags()))
+        .build();
+  }
+
   static DatasetInfo fromPb(Dataset datasetPb) {
     return new BuilderImpl(datasetPb).build();
   }

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

@@ -22,6 +22,7 @@
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.Lists;
+import io.opentelemetry.api.common.Attributes;
 import java.io.Serializable;
 import java.util.Collections;
 import java.util.HashMap;
@@ -479,4 +480,15 @@ public boolean equals(Object obj) {
         && Objects.equals(skipInvalidRows, other.skipInvalidRows)
         && Objects.equals(templateSuffix, other.templateSuffix);
   }
+
+  public Attributes getOtelAttributes() {
+    return Attributes.builder()
+        .put("table", OpenTelemetryHelper.getFieldAsString(this.getTable().getTable()))
+        .put("row", OpenTelemetryHelper.getFieldAsString(this.getRows()))
+        .put("templateSuffix", OpenTelemetryHelper.getFieldAsString(this.getTemplateSuffix()))
+        .put(
+            "ignoreUnknownValues", OpenTelemetryHelper.getFieldAsString(this.ignoreUnknownValues()))
+        .put("skipInvalidRows", OpenTelemetryHelper.getFieldAsString(this.skipInvalidRows()))
+        .build();
+  }
 }

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

@@ -21,6 +21,7 @@
 
 import com.google.api.services.bigquery.model.JobReference;
 import com.google.auto.value.AutoValue;
+import io.opentelemetry.api.common.Attributes;
 import java.io.Serializable;
 import java.util.UUID;
 import javax.annotation.Nullable;
@@ -123,4 +124,12 @@ static JobId fromPb(JobReference jobRef) {
         .setLocation(jobRef.getLocation())
         .build();
   }
+
+  public Attributes getOtelAttributes() {
+    return Attributes.builder()
+        .put("job", OpenTelemetryHelper.getFieldAsString(this.getJob()))
+        .put("location", OpenTelemetryHelper.getFieldAsString(this.getLocation()))
+        .put("project", OpenTelemetryHelper.getFieldAsString(this.getProject()))
+        .build();
+  }
 }

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

@@ -19,6 +19,7 @@
 import com.google.api.services.bigquery.model.Job;
 import com.google.common.base.Function;
 import com.google.common.base.MoreObjects;
+import io.opentelemetry.api.common.Attributes;
 import java.io.Serializable;
 import java.util.Objects;
 
@@ -375,4 +376,17 @@ public static JobInfo of(JobId jobId, JobConfiguration configuration) {
   static JobInfo fromPb(Job jobPb) {
     return new BuilderImpl(jobPb).build();
   }
+
+  public Attributes getOtelAttributes() {
+    return Attributes.builder()
+        .put("etag", OpenTelemetryHelper.getFieldAsString(this.getEtag()))
+        .put("generatedId", OpenTelemetryHelper.getFieldAsString(this.getGeneratedId()))
+        .put("jobId", OpenTelemetryHelper.getFieldAsString(this.getJobId().getJob()))
+        .put("selfLink", OpenTelemetryHelper.getFieldAsString(this.getSelfLink()))
+        .put("status", OpenTelemetryHelper.getFieldAsString(this.getStatus()))
+        .put("statistics", OpenTelemetryHelper.getFieldAsString(this.getStatistics()))
+        .put("userEmail", OpenTelemetryHelper.getFieldAsString(this.getUserEmail()))
+        .put("configuration", OpenTelemetryHelper.getFieldAsString(this.getConfiguration()))
+        .build();
+  }
 }

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

@@ -21,6 +21,7 @@
 
 import com.google.api.services.bigquery.model.ModelReference;
 import com.google.common.base.Function;
+import io.opentelemetry.api.common.Attributes;
 import java.io.Serializable;
 import java.util.Objects;
 
@@ -105,4 +106,12 @@ ModelReference toPb() {
   static ModelId fromPb(ModelReference modelRef) {
     return new ModelId(modelRef.getProjectId(), modelRef.getDatasetId(), modelRef.getModelId());
   }
+
+  public Attributes getOtelAttributes() {
+    return Attributes.builder()
+        .put("project", this.getProject())
+        .put("dataset", this.getDataset())
+        .put("model", this.getModel())
+        .build();
+  }
 }

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

@@ -26,6 +26,7 @@
 import com.google.common.base.Strings;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Lists;
+import io.opentelemetry.api.common.Attributes;
 import java.io.Serializable;
 import java.util.Collections;
 import java.util.List;
@@ -453,4 +454,25 @@ Model toPb() {
   static ModelInfo fromPb(Model modelPb) {
     return new BuilderImpl(modelPb).build();
   }
+
+  public Attributes getOtelAttributes() {
+    return Attributes.builder()
+        .put("etag", OpenTelemetryHelper.getFieldAsString(this.getEtag()))
+        .put("modelId", OpenTelemetryHelper.getFieldAsString(this.getModelId().getModel()))
+        .put("description", OpenTelemetryHelper.getFieldAsString(this.getDescription()))
+        .put("modelType", OpenTelemetryHelper.getFieldAsString(this.getModelType()))
+        .put("friendlyName", OpenTelemetryHelper.getFieldAsString(this.getFriendlyName()))
+        .put("creationTime", OpenTelemetryHelper.getFieldAsString(this.getCreationTime()))
+        .put("lastModifiedTime", OpenTelemetryHelper.getFieldAsString(this.getLastModifiedTime()))
+        .put("expirationTime", OpenTelemetryHelper.getFieldAsString(this.getExpirationTime()))
+        .put("labels", OpenTelemetryHelper.getFieldAsString(this.getLabels()))
+        .put("location", OpenTelemetryHelper.getFieldAsString(this.getLocation()))
+        .put("trainingRunList", OpenTelemetryHelper.getFieldAsString(this.getTrainingRuns()))
+        .put("featureColumnList", OpenTelemetryHelper.getFieldAsString(this.getFeatureColumns()))
+        .put("labelColumnList", OpenTelemetryHelper.getFieldAsString(this.getLabelColumns()))
+        .put(
+            "encryptionConfiguration",
+            OpenTelemetryHelper.getFieldAsString(this.getEncryptionConfiguration()))
+        .build();
+  }
 }