feat(cli): add threads to sink report (#9642)

datahub-project · Jan 17, 2024 · dadd51e · dadd51e
1 parent dfb2f7e
commit dadd51e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/sink/datahub_rest.py b/metadata-ingestion/src/datahub/ingestion/sink/datahub_rest.py
@@ -44,6 +44,7 @@ class DatahubRestSinkConfig(DatahubClientConfig):
 
 @dataclass
 class DataHubRestSinkReport(SinkReport):
+    max_threads: int = -1
     gms_version: str = ""
     pending_requests: int = 0
 
@@ -102,6 +103,7 @@ def __post_init__(self) -> None:
             .get("linkedin/datahub", {})
             .get("version", "")
         )
+        self.report.max_threads = self.config.max_threads
         logger.debug("Setting env variables to override config")
         set_env_variables_override_config(self.config.server, self.config.token)
         logger.debug("Setting gms config")