X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fservice%2Funit_of_work.py;h=04459962a734f4ae79f196712189459cc3a4affe;hb=19ee63847f8eb032009e63634b90eb594f3c0408;hp=a06e9e07833cd775a2d54d7278030cc7427696cb;hpb=8be81dfad35b08c4de77168e885bb18253069771;p=pti%2Fo2.git

diff --git a/o2ims/service/unit_of_work.py b/o2ims/service/unit_of_work.py
index a06e9e0..0445996 100644
--- a/o2ims/service/unit_of_work.py
+++ b/o2ims/service/unit_of_work.py
@@ -16,12 +16,18 @@
 from __future__ import annotations
 import abc
 
-from o2ims.domain.ocloud_repo import OcloudRepository
+from o2ims.domain.ocloud_repo import OcloudRepository,\
+    ResourcePoolRepository, ResourceRepository, ResourceTypeRepository,\
+    DeploymentManagerRepository
 from o2ims.domain.stx_repo import StxObjectRepository
 
 
 class AbstractUnitOfWork(abc.ABC):
     oclouds: OcloudRepository
+    resource_types: ResourceTypeRepository
+    resource_pools: ResourcePoolRepository
+    resources: ResourceRepository
+    deployment_managers: DeploymentManagerRepository
     stxobjects: StxObjectRepository
 
     def __enter__(self):
@@ -34,9 +40,24 @@ class AbstractUnitOfWork(abc.ABC):
         self._commit()
 
     def collect_new_events(self):
-        for ocloud in self.oclouds.seen:
-            while ocloud.events:
-                yield ocloud.events.pop(0)
+        for entry in self.oclouds.seen:
+            while entry.events:
+                yield entry.events.pop(0)
+        for entry in self.resource_pools.seen:
+            while entry.events:
+                yield entry.events.pop(0)
+        for entry in self.resources.seen:
+            while entry.events:
+                yield entry.events.pop(0)
+        for entry in self.resource_types.seen:
+            while entry.events:
+                yield entry.events.pop(0)
+        # for entry in self.stxobjects.seen:
+        #     while entry.events:
+        #         yield entry.events.pop(0)
+        for entry in self.deployment_managers.seen:
+            while entry.events:
+                yield entry.events.pop(0)
 
     @abc.abstractmethod
     def _commit(self):