Update domain for event handler
[pti/o2.git] / 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 from o2ims.domain.stx_repo import StxObjectRepository
22
23
24 class AbstractUnitOfWork(abc.ABC):
25     oclouds: OcloudRepository
26     resource_types: ResourceTypeRepository
27     resource_pools: ResourcePoolRepository
28     resources: ResourceRepository
29     stxobjects: StxObjectRepository
30
31     def __enter__(self):
32         return self
33
34     def __exit__(self, *args):
35         self.rollback()
36
37     def commit(self):
38         self._commit()
39
40     def collect_new_events(self):
41         for entry in self.oclouds.seen:
42             while entry.events:
43                 yield entry.events.pop(0)
44         for entry in self.resource_pools.seen:
45             while entry.events:
46                 yield entry.events.pop(0)
47         for entry in self.resources.seen:
48             while entry.events:
49                 yield entry.events.pop(0)
50         for entry in self.resource_types.seen:
51             while entry.events:
52                 yield entry.events.pop(0)
53         for entry in self.stxobjects.seen:
54             while entry.events:
55                 yield entry.events.pop(0)
56
57     @abc.abstractmethod
58     def _commit(self):
59         raise NotImplementedError
60
61     @abc.abstractmethod
62     def rollback(self):
63         raise NotImplementedError