From 258f2dbde0e8b55ecadaf15ee21599472005b159 Mon Sep 17 00:00:00 2001 From: Waleed Akbar <wakbar@cttc.es> Date: Tue, 23 Jul 2024 12:57:15 +0000 Subject: [PATCH] basic changes in the KPI Manager and Value API --- src/kpi_manager/database/KpiEngine.py | 10 +++++----- .../service/KpiManagerServiceServicerImpl.py | 1 + src/kpi_value_writer/service/KpiValueWriter.py | 5 ++++- 3 files changed, 10 insertions(+), 6 deletions(-) diff --git a/src/kpi_manager/database/KpiEngine.py b/src/kpi_manager/database/KpiEngine.py index 620ac9796..1d4015ff2 100644 --- a/src/kpi_manager/database/KpiEngine.py +++ b/src/kpi_manager/database/KpiEngine.py @@ -19,14 +19,14 @@ LOGGER = logging.getLogger(__name__) APP_NAME = 'tfs' ECHO = False # False: No dump SQL commands and transactions executed -CRDB_URI_TEMPLATE = 'cockroachdb://{:s}:{:s}@127.0.0.1:{:s}/{:s}?sslmode={:s}' +# CRDB_URI_TEMPLATE = 'cockroachdb://{:s}:{:s}@127.0.0.1:{:s}/{:s}?sslmode={:s}' CRDB_NAMESPACE = "crdb" CRDB_SQL_PORT = "26257" CRDB_DATABASE = "kpi" CRDB_USERNAME = "tfs" CRDB_PASSWORD = "tfs123" CRDB_SSLMODE = "require" -# CRDB_URI_TEMPLATE = 'cockroachdb://{:s}:{:s}@cockroachdb-public.{:s}.svc.cluster.local:{:s}/{:s}?sslmode={:s}' +CRDB_URI_TEMPLATE = 'cockroachdb://{:s}:{:s}@cockroachdb-public.{:s}.svc.cluster.local:{:s}/{:s}?sslmode={:s}' class KpiEngine: # def __init__(self): @@ -34,10 +34,10 @@ class KpiEngine: @staticmethod def get_engine() -> sqlalchemy.engine.Engine: - crdb_uri = CRDB_URI_TEMPLATE.format( - CRDB_USERNAME, CRDB_PASSWORD, CRDB_SQL_PORT, CRDB_DATABASE, CRDB_SSLMODE) # crdb_uri = CRDB_URI_TEMPLATE.format( - # CRDB_USERNAME, CRDB_PASSWORD, CRDB_NAMESPACE, CRDB_SQL_PORT, CRDB_DATABASE, CRDB_SSLMODE) + # CRDB_USERNAME, CRDB_PASSWORD, CRDB_SQL_PORT, CRDB_DATABASE, CRDB_SSLMODE) + crdb_uri = CRDB_URI_TEMPLATE.format( + CRDB_USERNAME, CRDB_PASSWORD, CRDB_NAMESPACE, CRDB_SQL_PORT, CRDB_DATABASE, CRDB_SSLMODE) try: # engine = sqlalchemy.create_engine( # crdb_uri, connect_args={'application_name': APP_NAME}, echo=ECHO, future=True) diff --git a/src/kpi_manager/service/KpiManagerServiceServicerImpl.py b/src/kpi_manager/service/KpiManagerServiceServicerImpl.py index 8996a8efb..7e30c688c 100644 --- a/src/kpi_manager/service/KpiManagerServiceServicerImpl.py +++ b/src/kpi_manager/service/KpiManagerServiceServicerImpl.py @@ -59,6 +59,7 @@ class KpiManagerServiceServicerImpl(KpiManagerServiceServicer): response = KpiModel.convert_row_to_KpiDescriptor(row) return response except Exception as e: + print ('Unable to search kpi id. {:}'.format(e)) LOGGER.debug('Unable to search kpi id. {:}'.format(e)) @safe_and_metered_rpc_method(METRICS_POOL, LOGGER) diff --git a/src/kpi_value_writer/service/KpiValueWriter.py b/src/kpi_value_writer/service/KpiValueWriter.py index 82342bbbd..70e018d08 100644 --- a/src/kpi_value_writer/service/KpiValueWriter.py +++ b/src/kpi_value_writer/service/KpiValueWriter.py @@ -94,7 +94,7 @@ class KpiValueWriter: kpi_id = KpiId() kpi_id.kpi_id.uuid = kpi_value_uuid print("KpiId generated: {:}".format(kpi_id)) - print("Kpi manger client created: {:}".format(kpi_manager_client)) + # print("Kpi manger client created: {:}".format(kpi_manager_client)) try: kpi_descriptor_object = KpiDescriptor() @@ -103,7 +103,10 @@ class KpiValueWriter: print("kpi descriptor received: {:}".format(kpi_descriptor_object)) if isinstance (kpi_descriptor_object, KpiDescriptor): LOGGER.info("Extracted row: {:}".format(kpi_descriptor_object)) + print("Extracted row: {:}".format(kpi_descriptor_object)) else: LOGGER.info("Error in extracting row {:}".format(kpi_descriptor_object)) + print("Error in extracting row {:}".format(kpi_descriptor_object)) except Exception as e: + LOGGER.info("Unable to get Descriptor. Error: {:}".format(e)) print ("Unable to get Descriptor. Error: {:}".format(e)) -- GitLab