fix: add guard for mongo client closing by external library

Author: schivmeister

src/dags/operators/DagBatchPipelineOperator.py

@@ -7,6 +7,13 @@
 from airflow.utils.types import DagRunType
 from pymongo import MongoClient
 
+# WORKAROUND: Disable __del__ in MSSDK's MongoDBRepository to prevent premature client closure
+# The MSSDK MongoDBRepository closes the MongoClient in __del__, but it doesn't own the client.
+# When the repository is garbage collected between loop iterations, it closes the shared client.
+from mapping_suite_sdk.core.adapters.repository import MongoDBRepository as _MSSKDMongoDBRepository
+if hasattr(_MSSKDMongoDBRepository, '__del__'):
+    delattr(_MSSKDMongoDBRepository, '__del__')
+
 from src.dags import RUN_MATERIALISED_VIEW_DAG_PARAM, XCOM_SOURCE_RUN_TYPE_KEY
 from src.dags.dags_utils import pull_dag_upstream, push_dag_downstream, get_dag_param, smart_xcom_pull, \
     smart_xcom_push

src/ted_sws/data_manager/adapters/mapping_package_repository.py

@@ -7,6 +7,12 @@
 from pymongo import MongoClient
 
 from mapping_suite_sdk.core.adapters.repository import MongoDBRepository, ModelNotFoundError
+
+# WORKAROUND: Disable __del__ in MSSDK's MongoDBRepository to prevent premature client closure
+# The MSSDK MongoDBRepository closes the MongoClient in __del__, but it doesn't own the client.
+# When the repository is garbage collected, it closes the shared client unexpectedly.
+if hasattr(MongoDBRepository, '__del__'):
+    delattr(MongoDBRepository, '__del__')
 from mapping_suite_sdk.mapping_package_v1.models import MappingPackageV1
 from mapping_suite_sdk.mapping_package_v2.models import MappingPackageV2
 from mapping_suite_sdk.mapping_package_v3.models import MappingPackageV3

src/ted_sws/mapping_suite_processor/services/mapping_package_processor.py

@@ -4,6 +4,13 @@
 
 from pymongo import MongoClient
 
+# WORKAROUND: Disable __del__ in MSSDK's MongoDBRepository to prevent premature client closure  
+# The MSSDK MongoDBRepository closes the MongoClient in __del__, but it doesn't own the client.
+# When the repository is garbage collected, it closes the shared client unexpectedly.
+from mapping_suite_sdk.core.adapters.repository import MongoDBRepository as _MSSKDMongoDBRepository
+if hasattr(_MSSKDMongoDBRepository, '__del__'):
+    delattr(_MSSKDMongoDBRepository, '__del__')
+
 from src.ted_sws import config
 from src.ted_sws.core.model.manifestation import XMLManifestation
 from src.ted_sws.core.model.notice import Notice