o2common/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
  20 class AbstractUnitOfWork(abc.ABC):
  21     # oclouds: OcloudRepository
  22     # resource_types: ResourceTypeRepository
  23     # resource_pools: ResourcePoolRepository
  24     # resources: ResourceRepository
  25     # deployment_managers: DeploymentManagerRepository
  26     # nfdeployment_descs: object
  27
  28     def __enter__(self):
  29         return self
  30
  31     def __exit__(self, *args):
  32         self.rollback()
  33
  34     def commit(self):
  35         self._commit()
  36
  37     def collect_new_events(self):
  38         return self._collect_new_events()
  39
  40     def _collect_new_events(self):
  41         raise NotImplementedError
  42
  43     @abc.abstractmethod
  44     def _commit(self):
  45         raise NotImplementedError
  46
  47     @abc.abstractmethod
  48     def rollback(self):
  49         raise NotImplementedError