o2ims/service/unit_of_work.py

   1 # Copyright (C) 2021 Wind River Systems, Inc.
   2 #
   3 #  Licensed under the Apache License, Version 2.0 (the "License");
   4 #  you may not use this file except in compliance with the License.
   5 #  You may obtain a copy of the License at
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 #  Unless required by applicable law or agreed to in writing, software
  10 #  distributed under the License is distributed on an "AS IS" BASIS,
  11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 #  See the License for the specific language governing permissions and
  13 #  limitations under the License.
  14
  15 # pylint: disable=attribute-defined-outside-init
  16 from __future__ import annotations
  17 import abc
  18
  19 from o2ims.domain.ocloud_repo import OcloudRepository,\
  20     ResourcePoolRepository, ResourceRepository, ResourceTypeRepository,\
  21     DeploymentManagerRepository
  22 from o2ims.domain.stx_repo import StxObjectRepository
  23
  24
  25 class AbstractUnitOfWork(abc.ABC):
  26     oclouds: OcloudRepository
  27     resource_types: ResourceTypeRepository
  28     resource_pools: ResourcePoolRepository
  29     resources: ResourceRepository
  30     deployment_managers: DeploymentManagerRepository
  31     stxobjects: StxObjectRepository
  32
  33     def __enter__(self):
  34         return self
  35
  36     def __exit__(self, *args):
  37         self.rollback()
  38
  39     def commit(self):
  40         self._commit()
  41
  42     def collect_new_events(self):
  43         for entry in self.oclouds.seen:
  44             while entry.events:
  45                 yield entry.events.pop(0)
  46         for entry in self.resource_pools.seen:
  47             while entry.events:
  48                 yield entry.events.pop(0)
  49         for entry in self.resources.seen:
  50             while entry.events:
  51                 yield entry.events.pop(0)
  52         for entry in self.resource_types.seen:
  53             while entry.events:
  54                 yield entry.events.pop(0)
  55         # for entry in self.stxobjects.seen:
  56         #     while entry.events:
  57         #         yield entry.events.pop(0)
  58         for entry in self.deployment_managers.seen:
  59             while entry.events:
  60                 yield entry.events.pop(0)
  61
  62     @abc.abstractmethod
  63     def _commit(self):
  64         raise NotImplementedError
  65
  66     @abc.abstractmethod
  67     def rollback(self):
  68         raise NotImplementedError