X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fviews%2Focloud_view.py;h=ae8b204edc74402e3ecf430cbab373bbf841cccf;hb=3da89330f3837ac6cffd2cad4c4018c9f8c3327d;hp=3735298da208f2d83d5d03b4271e942b42bfecc6;hpb=9625c5b766377f641d9641471f10dd491a61447f;p=pti%2Fo2.git

diff --git a/o2ims/views/ocloud_view.py b/o2ims/views/ocloud_view.py
index 3735298..ae8b204 100644
--- a/o2ims/views/ocloud_view.py
+++ b/o2ims/views/ocloud_view.py
@@ -12,11 +12,14 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+import logging
 import uuid
+from datetime import datetime
 
-from o2common.service import unit_of_work
-from o2ims.views.ocloud_dto import SubscriptionDTO
-from o2ims.domain.subscription_obj import Subscription
+from o2common.service import unit_of_work, messagebus
+from o2ims.domain import events
+from o2ims.views.ocloud_dto import RegistrationDTO, SubscriptionDTO
+from o2ims.domain.subscription_obj import Registration, Subscription
 
 
 def oclouds(uow: unit_of_work.AbstractUnitOfWork):
@@ -115,3 +118,51 @@ def subscription_delete(subscriptionId: str,
         uow.subscriptions.delete(subscriptionId)
         uow.commit()
     return True
+
+
+def registrations(uow: unit_of_work.AbstractUnitOfWork):
+    with uow:
+        li = uow.registrations.list()
+    return [r.serialize() for r in li]
+
+
+def registration_one(registrationId: str,
+                     uow: unit_of_work.AbstractUnitOfWork):
+    with uow:
+        first = uow.registrations.get(registrationId)
+        return first.serialize() if first is not None else None
+
+
+def registration_create(registrationDto: RegistrationDTO.registration,
+                        bus: messagebus.MessageBus):
+
+    reg_uuid = str(uuid.uuid4())
+    registration = Registration(
+        reg_uuid, registrationDto['callback'])
+    with bus.uow as uow:
+        uow.registrations.add(registration)
+        logging.debug('before event length {}'.format(
+            len(registration.events)))
+        registration.events.append(events.RegistrationChanged(
+            reg_uuid,
+            datetime.now()))
+        logging.debug('after event length {}'.format(len(registration.events)))
+        uow.commit()
+    _handle_events(bus)
+    return {"registrationId": reg_uuid}
+
+
+def registration_delete(registrationId: str,
+                        uow: unit_of_work.AbstractUnitOfWork):
+    with uow:
+        uow.registrations.delete(registrationId)
+        uow.commit()
+    return True
+
+
+def _handle_events(bus: messagebus.MessageBus):
+    # handle events
+    events = bus.uow.collect_new_events()
+    for event in events:
+        bus.handle(event)
+    return True