o2app/adapter/unit_of_work.py

   1 # Copyright (C) 2021 Wind River Systems, Inc.\r
   2 #\r
   3 #  Licensed under the Apache License, Version 2.0 (the "License");\r
   4 #  you may not use this file except in compliance with the License.\r
   5 #  You may obtain a copy of the License at\r
   6 #\r
   7 #      http://www.apache.org/licenses/LICENSE-2.0\r
   8 #\r
   9 #  Unless required by applicable law or agreed to in writing, software\r
  10 #  distributed under the License is distributed on an "AS IS" BASIS,\r
  11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  12 #  See the License for the specific language governing permissions and\r
  13 #  limitations under the License.\r
  14 \r
  15 # pylint: disable=attribute-defined-outside-init\r
  16 from __future__ import annotations\r
  17 from sqlalchemy import create_engine\r
  18 from sqlalchemy.orm import sessionmaker\r
  19 from sqlalchemy.orm.session import Session\r
  20 \r
  21 from o2common.config import config\r
  22 from o2common.service.unit_of_work import AbstractUnitOfWork\r
  23 \r
  24 from o2ims.adapter import ocloud_repository\r
  25 from o2dms.adapter import dms_repository\r
  26 \r
  27 from o2common.helper import o2logging\r
  28 logger = o2logging.get_logger(__name__)\r
  29 \r
  30 DEFAULT_SESSION_FACTORY = sessionmaker(\r
  31     bind=create_engine(\r
  32         config.get_postgres_uri(),\r
  33         isolation_level="REPEATABLE READ",\r
  34     )\r
  35 )\r
  36 \r
  37 \r
  38 class SqlAlchemyUnitOfWork(AbstractUnitOfWork):\r
  39     def __init__(self, session_factory=DEFAULT_SESSION_FACTORY):\r
  40         self.session_factory = session_factory\r
  41 \r
  42     def __enter__(self):\r
  43         self.session = self.session_factory()  # type: Session\r
  44         self.oclouds = ocloud_repository\\r
  45             .OcloudSqlAlchemyRepository(self.session)\r
  46         self.resource_types = ocloud_repository\\r
  47             .ResouceTypeSqlAlchemyRepository(self.session)\r
  48         self.resource_pools = ocloud_repository\\r
  49             .ResourcePoolSqlAlchemyRepository(self.session)\r
  50         self.resources = ocloud_repository\\r
  51             .ResourceSqlAlchemyRepository(self.session)\r
  52         self.subscriptions = ocloud_repository\\r
  53             .SubscriptionSqlAlchemyRepository(self.session)\r
  54         self.registrations = ocloud_repository\\r
  55             .RegistrationSqlAlchemyRepository(self.session)\r
  56         self.deployment_managers = ocloud_repository\\r
  57             .DeploymentManagerSqlAlchemyRepository(self.session)\r
  58         self.nfdeployment_descs = dms_repository\\r
  59             .NfDeploymentDescSqlAlchemyRepository(self.session)\r
  60         self.nfdeployments = dms_repository\\r
  61             .NfDeploymentSqlAlchemyRepository(self.session)\r
  62         self.ocloudvresources = dms_repository\\r
  63             .NfOCloudVResourceSqlAlchemyRepository(self.session)\r
  64         return super().__enter__()\r
  65 \r
  66     def __exit__(self, *args):\r
  67         super().__exit__(*args)\r
  68         self.session.close()\r
  69 \r
  70     def _commit(self):\r
  71         self.session.commit()\r
  72 \r
  73     def rollback(self):\r
  74         self.session.rollback()\r
  75 \r
  76     def _collect_new_events(self):\r
  77         for entry in self.oclouds.seen:\r
  78             # while hasattr(entry, 'events') and len(entry.events) > 0:\r
  79             while entry.events is not None and len(entry.events) > 0:\r
  80                 yield entry.events.pop(0)\r
  81         for entry in self.resource_pools.seen:\r
  82             while entry.events is not None and len(entry.events) > 0:\r
  83                 yield entry.events.pop(0)\r
  84         for entry in self.resources.seen:\r
  85             while entry.events is not None and len(entry.events) > 0:\r
  86                 yield entry.events.pop(0)\r
  87         for entry in self.resource_types.seen:\r
  88             while hasattr(entry, 'events') and len(entry.events) > 0:\r
  89                 yield entry.events.pop(0)\r
  90         for entry in self.deployment_managers.seen:\r
  91             while hasattr(entry, 'events') and len(entry.events) > 0:\r
  92                 yield entry.events.pop(0)\r
  93         for entry in self.nfdeployment_descs.seen:\r
  94             while hasattr(entry, 'events') and len(entry.events) > 0:\r
  95                 yield entry.events.pop(0)\r
  96         for entry in self.nfdeployments.seen:\r
  97             while hasattr(entry, 'events') and len(entry.events) > 0:\r
  98                 yield entry.events.pop(0)\r
  99         for entry in self.ocloudvresources.seen:\r
 100             while hasattr(entry, 'events') and len(entry.events) > 0:\r
 101                 yield entry.events.pop(0)\r