SIENTIAPDE-1312
Update dependencies and enhance OpcRepository functionality - Updated sientia-dataops-library version to 1.4.7 in requirements files. - Modified GITHUB_BRANCH in values.yaml for improved pipeline management. - Refactored OpcRepository class to inherit from BaseActivity, adding enhanced logging and error handling during disconnection. - Implemented a disconnection fallback mechanism to ensure graceful handling of OPC server disconnections.
This commit is contained in:
@@ -1,3 +1,5 @@
|
||||
import asyncio
|
||||
import json
|
||||
import time
|
||||
import traceback
|
||||
from datetime import datetime
|
||||
@@ -10,6 +12,7 @@ from asyncua.ua import DataValue, DateTime, Variant, VariantType
|
||||
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.temporal.activities.base import BaseActivity
|
||||
|
||||
from laborious import metrics
|
||||
|
||||
@@ -37,19 +40,19 @@ data_type_map = {
|
||||
}
|
||||
|
||||
|
||||
class OpcRepository:
|
||||
class OpcRepository(BaseActivity):
|
||||
def __init__(
|
||||
self,
|
||||
opc_id: str,
|
||||
url: str,
|
||||
pod_id: str,
|
||||
logger: Logger,
|
||||
notification_handler: NotificationHandler,
|
||||
reconnection_interval: int = 60,
|
||||
server_uri: str | None = None,
|
||||
cert_path: str | None = None,
|
||||
private_key_path: str | None = None,
|
||||
server_cert_path: str | None = None,
|
||||
pod_id: str | None = None,
|
||||
server_cert_path: str | None = None
|
||||
):
|
||||
self.url = url
|
||||
self.id = opc_id
|
||||
@@ -65,6 +68,8 @@ class OpcRepository:
|
||||
self.client: None | Client = None
|
||||
self.pod_id = pod_id
|
||||
|
||||
BaseActivity.__init__(self, logger, notification_handler, set_error_counter=True)
|
||||
|
||||
self.metadata = {
|
||||
'model_name': '-',
|
||||
'model_id': '-',
|
||||
@@ -125,7 +130,14 @@ class OpcRepository:
|
||||
Exception: If the connection to the OPC server fails.
|
||||
"""
|
||||
|
||||
self.client = Client(self.url)
|
||||
self.client = Client(self.url) # type: ignore[attr-defined]
|
||||
|
||||
self.client.name = self.pod_id
|
||||
self.client.application_name = self.pod_id
|
||||
pod_uri = self.pod_id.replace('-', ':')
|
||||
self.client.application_uri = pod_uri
|
||||
self.client.product_uri = pod_uri
|
||||
|
||||
if self.cert_path:
|
||||
await self.set_security()
|
||||
self.logger.custom_info(f'Starting connection to OPC server {self.id}...', self.metadata)
|
||||
@@ -169,6 +181,28 @@ class OpcRepository:
|
||||
'attachment_content': trace,
|
||||
}
|
||||
|
||||
async def disconnection_fallback(self) -> list:
|
||||
"""
|
||||
Tries 5 times to disconnect from the OPC UA server, with a delay of 100ms x try.
|
||||
"""
|
||||
|
||||
assert self.client is not None
|
||||
error_stack = []
|
||||
for i in range(5):
|
||||
try:
|
||||
self.logger.info(f'Disconnecting from OPC UA server, attempt {i+1} of 5')
|
||||
await self.client.disconnect()
|
||||
return []
|
||||
except Exception as e:
|
||||
self.logger.error(f'Failed to disconnect from OPC UA server in attempt {i+1} of 5: {e}')
|
||||
error_stack.append({
|
||||
'attempt': i+1,
|
||||
'error': str(e),
|
||||
'traceback': traceback.format_exc(),
|
||||
})
|
||||
await asyncio.sleep(0.1 * i)
|
||||
return error_stack
|
||||
|
||||
async def disconnect(self):
|
||||
"""
|
||||
Gracefully disconnect from the OPC server.
|
||||
@@ -179,11 +213,20 @@ class OpcRepository:
|
||||
"""
|
||||
if self.client is None:
|
||||
return
|
||||
try:
|
||||
await self.client.disconnect()
|
||||
self.logger.custom_info('Disconnected from OPC server', self.metadata)
|
||||
except Exception as e:
|
||||
self.logger.custom_error(f'Failed to disconnect from OPC server: {e}', self.metadata)
|
||||
|
||||
errors = await self.disconnection_fallback()
|
||||
if errors:
|
||||
self.send_notification(
|
||||
metadata=self.metadata,
|
||||
notification_id=f'OPC_DISCONNECTION_ERROR_{self.id}',
|
||||
message=f'Failed to disconnect from OPC server in 5 attempts: {errors}',
|
||||
block='opc_repository',
|
||||
level=NotificationLevel.ERROR,
|
||||
attachment_content=json.dumps(errors, indent=4),
|
||||
)
|
||||
else:
|
||||
self.logger.warning(f'Disconnected from OPC server {self.id} successfully')
|
||||
|
||||
self.client = None
|
||||
|
||||
async def validate_connection(self) -> tuple[bool, dict[str, Any]]:
|
||||
|
||||
Reference in New Issue
Block a user