add enable env for tracing

Author: david-leifker

metadata-ingestion/src/datahub/ingestion/sink/datahub_rest.py

@@ -49,6 +49,11 @@
 )
 
 
+class RestTraceMode(ConfigEnum):
+    ENABLED = auto()
+    DISABLED = auto()
+
+
 class RestSinkEndpoint(ConfigEnum):
     RESTLI = auto()
     OPENAPI = auto()
@@ -75,9 +80,16 @@ class RestSinkMode(ConfigEnum):
 )
 
 
+_DEFAULT_REST_TRACE_MODE = pydantic.parse_obj_as(
+    RestTraceMode,
+    os.getenv("DATAHUB_REST_TRACE_MODE", RestTraceMode.DISABLED),
+)
+
+
 class DatahubRestSinkConfig(DatahubClientConfig):
     mode: RestSinkMode = _DEFAULT_REST_SINK_MODE
     endpoint: RestSinkEndpoint = _DEFAULT_REST_SINK_ENDPOINT
+    default_trace_mode: RestTraceMode = _DEFAULT_REST_TRACE_MODE
 
     # These only apply in async modes.
     max_threads: pydantic.PositiveInt = _DEFAULT_REST_SINK_MAX_THREADS
@@ -184,9 +196,8 @@ def _make_emitter(cls, config: DatahubRestSinkConfig) -> DataHubRestEmitter:
             ca_certificate_path=config.ca_certificate_path,
             client_certificate_path=config.client_certificate_path,
             disable_ssl_verification=config.disable_ssl_verification,
-            openapi_ingestion=True
-            if config.endpoint == RestSinkEndpoint.OPENAPI
-            else False,
+            openapi_ingestion=config.endpoint == RestSinkEndpoint.OPENAPI,
+            default_trace_mode=config.default_trace_mode == RestTraceMode.ENABLED,
         )
 
     @property