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