feat: Disable the tracing feature and remove public APIs.

googleapis · May 6, 2024 · a736fcc · a736fcc
1 parent 82feaae
commit a736fcc
Show file tree

Hide file tree

Showing 8 changed files with 24 additions and 1 deletion.
diff --git a/google-cloud-firestore/src/main/java/com/google/cloud/firestore/FirestoreOptions.java b/google-cloud-firestore/src/main/java/com/google/cloud/firestore/FirestoreOptions.java
@@ -127,6 +127,7 @@ com.google.cloud.firestore.telemetry.TraceUtil getTraceUtil() {
     return traceUtil;
   }
 
+  @InternalApi
   @BetaApi
   @Nonnull
   public FirestoreOpenTelemetryOptions getOpenTelemetryOptions() {
@@ -223,6 +224,7 @@ public Builder setDatabaseId(@Nonnull String databaseId) {
      *
      * @param openTelemetryOptions The `FirestoreOpenTelemetryOptions` to use.
      */
+    @InternalApi
     @BetaApi
     @Nonnull
     public Builder setOpenTelemetryOptions(

diff --git a/google-cloud-firestore/src/main/java/com/google/cloud/firestore/telemetry/TraceUtil.java b/google-cloud-firestore/src/main/java/com/google/cloud/firestore/telemetry/TraceUtil.java
@@ -25,6 +25,7 @@
 import javax.annotation.Nullable;
 
 public interface TraceUtil {
+  static final boolean TRACING_FEATURE_ENABLED = false;
   String ATTRIBUTE_SERVICE_PREFIX = "gcp.firestore.";
   String SPAN_NAME_DOC_REF_CREATE = "DocumentReference.Create";
   String SPAN_NAME_DOC_REF_SET = "DocumentReference.Set";
@@ -62,6 +63,11 @@ public interface TraceUtil {
    * @return An instance of the TraceUtil class.
    */
   static TraceUtil getInstance(@Nonnull FirestoreOptions firestoreOptions) {
+    // TODO(tracing): Remove this to enable the feature.
+    if (!TRACING_FEATURE_ENABLED) {
+      return new DisabledTraceUtil();
+    }
+
     boolean createEnabledInstance = firestoreOptions.getOpenTelemetryOptions().isTracingEnabled();
 
     // The environment variable can override options to enable/disable telemetry collection.

diff --git a/...le-cloud-firestore/src/test/java/com/google/cloud/firestore/OpenTelemetryOptionsTest.java b/...le-cloud-firestore/src/test/java/com/google/cloud/firestore/OpenTelemetryOptionsTest.java
@@ -26,8 +26,11 @@
 import javax.annotation.Nullable;
 import org.junit.After;
 import org.junit.Before;
+import org.junit.Ignore;
 import org.junit.Test;
 
+// TODO(tracing): Re-enable this test when the tracing feature is enabled.
+@Ignore
 public class OpenTelemetryOptionsTest {
   @Nullable private Firestore firestore;
 

diff --git a/...oud-firestore/src/test/java/com/google/cloud/firestore/it/ITE2ETracingTestGlobalOtel.java b/...oud-firestore/src/test/java/com/google/cloud/firestore/it/ITE2ETracingTestGlobalOtel.java
@@ -15,10 +15,12 @@
  */
 package com.google.cloud.firestore.it;
 
+import org.junit.Ignore;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
+@Ignore
 public class ITE2ETracingTestGlobalOtel extends ITE2ETracingTest {
   @Override
   protected boolean isUsingGlobalOpenTelemetrySDK() {

diff --git a/...-firestore/src/test/java/com/google/cloud/firestore/it/ITE2ETracingTestNonGlobalOtel.java b/...-firestore/src/test/java/com/google/cloud/firestore/it/ITE2ETracingTestNonGlobalOtel.java
@@ -15,10 +15,12 @@
  */
 package com.google.cloud.firestore.it;
 
+import org.junit.Ignore;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
+@Ignore
 public class ITE2ETracingTestNonGlobalOtel extends ITE2ETracingTest {
   @Override
   protected boolean isUsingGlobalOpenTelemetrySDK() {

diff --git a/...-cloud-firestore/src/test/java/com/google/cloud/firestore/it/ITTracingTestGlobalOtel.java b/...-cloud-firestore/src/test/java/com/google/cloud/firestore/it/ITTracingTestGlobalOtel.java
@@ -15,10 +15,12 @@
  */
 package com.google.cloud.firestore.it;
 
+import org.junit.Ignore;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
+@Ignore
 public class ITTracingTestGlobalOtel extends ITTracingTest {
   @Override
   protected boolean isUsingGlobalOpenTelemetrySDK() {

diff --git a/...oud-firestore/src/test/java/com/google/cloud/firestore/it/ITTracingTestNonGlobalOtel.java b/...oud-firestore/src/test/java/com/google/cloud/firestore/it/ITTracingTestNonGlobalOtel.java
@@ -15,10 +15,12 @@
  */
 package com.google.cloud.firestore.it;
 
+import org.junit.Ignore;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
+@Ignore
 public class ITTracingTestNonGlobalOtel extends ITTracingTest {
   @Override
   protected boolean isUsingGlobalOpenTelemetrySDK() {

diff --git a/google-cloud-firestore/src/test/java/com/google/cloud/firestore/telemetry/TraceUtilTest.java b/google-cloud-firestore/src/test/java/com/google/cloud/firestore/telemetry/TraceUtilTest.java
@@ -56,6 +56,10 @@ public void tracingEnabledOptionsUseEnabledTraceUtil() {
                 .setOpenTelemetryOptions(
                     FirestoreOpenTelemetryOptions.newBuilder().setTracingEnabled(true).build())
                 .build());
-    assertThat(traceUtil instanceof EnabledTraceUtil).isTrue();
+
+    // This is currently disabled because the feature is disabled as a whole.
+    // TODO (tracing): This should be `traceUtil instanceof EnabledTraceUtil
+    //  once the feature is enabled.
+    assertThat(traceUtil instanceof DisabledTraceUtil).isTrue();
   }
 }