From 09036cdf24248d571bde520ca0db0fae14f0b21c Mon Sep 17 00:00:00 2001
From: gifrerenom <lluis.gifre@cttc.es>
Date: Tue, 21 Feb 2023 12:52:26 +0000
Subject: [PATCH] Context component:
- Added indexes recommended by Cockroach internal planner to boost performance
---
src/context/service/database/models/_Base.py | 39 +++++++++++++++++++-
1 file changed, 38 insertions(+), 1 deletion(-)
diff --git a/src/context/service/database/models/_Base.py b/src/context/service/database/models/_Base.py
index 4323fb713..973f7b7be 100644
--- a/src/context/service/database/models/_Base.py
+++ b/src/context/service/database/models/_Base.py
@@ -13,10 +13,47 @@
# limitations under the License.
import sqlalchemy
-from sqlalchemy.orm import declarative_base
+from typing import Any, List
+from sqlalchemy.orm import Session, sessionmaker, declarative_base
+from sqlalchemy.sql import text
+from sqlalchemy_cockroachdb import run_transaction
_Base = declarative_base()
+def create_performance_enhancers(db_engine : sqlalchemy.engine.Engine) -> None:
+ def index_storing(
+ index_name : str, table_name : str, index_fields : List[str], storing_fields : List[str]
+ ) -> Any:
+ str_index_fields = ','.join(['"{:s}"'.format(index_field) for index_field in index_fields])
+ str_storing_fields = ','.join(['"{:s}"'.format(storing_field) for storing_field in storing_fields])
+ INDEX_STORING = 'CREATE INDEX IF NOT EXISTS {:s} ON "{:s}" ({:s}) STORING ({:s});'
+ return text(INDEX_STORING.format(index_name, table_name, str_index_fields, str_storing_fields))
+
+ statements = [
+ index_storing('configrule_device_uuid_rec_idx', 'configrule', ['device_uuid'], [
+ 'service_uuid', 'slice_uuid', 'position', 'kind', 'action', 'data', 'created_at', 'updated_at'
+ ]),
+ index_storing('configrule_service_uuid_rec_idx', 'configrule', ['service_uuid'], [
+ 'device_uuid', 'slice_uuid', 'position', 'kind', 'action', 'data', 'created_at', 'updated_at'
+ ]),
+ index_storing('configrule_slice_uuid_rec_idx', 'configrule', ['slice_uuid'], [
+ 'device_uuid', 'service_uuid', 'position', 'kind', 'action', 'data', 'created_at', 'updated_at'
+ ]),
+ index_storing('constraint_service_uuid_rec_idx', 'constraint', ['service_uuid'], [
+ 'slice_uuid', 'position', 'kind', 'data', 'created_at', 'updated_at'
+ ]),
+ index_storing('constraint_slice_uuid_rec_idx', 'constraint', ['slice_uuid'], [
+ 'service_uuid', 'position', 'kind', 'data', 'created_at', 'updated_at'
+ ]),
+ index_storing('endpoint_device_uuid_rec_idx', 'endpoint', ['device_uuid'], [
+ 'topology_uuid', 'name', 'endpoint_type', 'kpi_sample_types', 'created_at', 'updated_at'
+ ]),
+ ]
+ def callback(session : Session) -> bool:
+ for stmt in statements: session.execute(stmt)
+ run_transaction(sessionmaker(bind=db_engine), callback)
+
def rebuild_database(db_engine : sqlalchemy.engine.Engine, drop_if_exists : bool = False):
if drop_if_exists: _Base.metadata.drop_all(db_engine)
_Base.metadata.create_all(db_engine)
+ create_performance_enhancers(db_engine)
--
GitLab