SIENTIAPDE-1325

Update GITHUB_BRANCH in values.yaml to feature/SIENTIAPDE-1325 for specific external operation metrics. Refactor shutdown methods in Activities class to ensure proper closure of MongoDB, Redis, and Gates connections. Adjust tests to reflect these changes and enhance repository pattern implementation for MongoDB and Redis activities.
This commit is contained in:
vitor-aignosi
2025-10-29 16:58:22 -03:00
parent 10b3310677
commit 0c16075823
8 changed files with 152 additions and 127 deletions

View File

@@ -8,17 +8,14 @@ with workflow.unsafe.imports_passed_through():
from datetime import datetime
from typing import Any
from pandas import DataFrame
from pymongo import MongoClient
from sientia_do.notifications.handlers import CoreNotificationHandler as NotificationHandler
from sientia_do.notifications.models import NotificationLevel
from sientia_do.observability.logger import Logger
from sientia_do.repository.mongodb_repository import MongoDBRepository
from sientia_do.observability.sientia_monitoring import SientiaMonitoring
from sientia_do.repository.mongodb_repository import MongoDBRepository
from sientia_do.temporal.constants import DATETIME_FORMAT_MS_WITH_TZ
class MongoDB(SientiaMonitoring):
"""
MongoDB operations for data retrieval and storage.
@@ -53,7 +50,7 @@ class MongoDB(SientiaMonitoring):
Raises:
ConnectionError: If MongoDB connection fails
"""
self.mongodb_repository = MongoDBRepository(
connection_string=connection_string,
database_name=database_name,
@@ -136,7 +133,7 @@ class MongoDB(SientiaMonitoring):
self.debug(f'Loaded data: {data}', metadata=metadata)
return DataFrame(data).to_dict()
return data
except Exception as e:
trace = traceback.format_exc()
self.send_notification(