# See the License for the specific language governing permissions and\r
# limitations under the License.\r
\r
-from o2ims.service import unit_of_work\r
+import logging\r
+import uuid\r
+from datetime import datetime\r
\r
+from o2common.service import unit_of_work, messagebus\r
+from o2ims.domain import events\r
+from o2ims.views.ocloud_dto import RegistrationDTO, SubscriptionDTO\r
+from o2ims.domain.subscription_obj import Registration, Subscription\r
\r
-def ocloud_one(ocloudid: str, uow: unit_of_work.SqlAlchemyUnitOfWork):\r
+\r
+def oclouds(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.oclouds.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def ocloud_one(ocloudid: str, uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.oclouds.get(ocloudid)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def resource_types(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.resource_types.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def resource_type_one(resourceTypeId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.resource_types.get(resourceTypeId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def resource_pools(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.resource_pools.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def resource_pool_one(resourcePoolId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.resource_pools.get(resourcePoolId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def resources(resourcePoolId: str, uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.resources.list(resourcePoolId)\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def resource_one(resourceId: str, uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.resources.get(resourceId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def deployment_managers(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.deployment_managers.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def deployment_manager_one(deploymentManagerId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.deployment_managers.get(deploymentManagerId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def subscriptions(uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ li = uow.subscriptions.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def subscription_one(subscriptionId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.subscriptions.get(subscriptionId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def subscription_create(subscriptionDto: SubscriptionDTO.subscription,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+\r
+ sub_uuid = str(uuid.uuid4())\r
+ subscription = Subscription(\r
+ sub_uuid, subscriptionDto['callback'],\r
+ subscriptionDto['consumerSubscriptionId'],\r
+ subscriptionDto['filter'])\r
+ with uow:\r
+ uow.subscriptions.add(subscription)\r
+ uow.commit()\r
+ return {"subscriptionId": sub_uuid}\r
+\r
+\r
+def subscription_delete(subscriptionId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
with uow:\r
- results = uow.session.execute(\r
- """\r
- SELECT oCloudId, name FROM ocloud WHERE oCloudId = :ocloudid\r
- """,\r
- dict(ocloudid=ocloudid),\r
- )\r
- return dict(results[0]) if len(results) > 0 else None\r
+ uow.subscriptions.delete(subscriptionId)\r
+ uow.commit()\r
+ return True\r
\r
\r
-def oclouds(uow: unit_of_work.SqlAlchemyUnitOfWork):\r
+def registrations(uow: unit_of_work.AbstractUnitOfWork):\r
with uow:\r
- results = uow.session.execute(\r
- """\r
- SELECT oCloudId, name FROM ocloud\r
- """,\r
- )\r
- return [dict(r) for r in results]\r
+ li = uow.registrations.list()\r
+ return [r.serialize() for r in li]\r
+\r
+\r
+def registration_one(registrationId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ first = uow.registrations.get(registrationId)\r
+ return first.serialize() if first is not None else None\r
+\r
+\r
+def registration_create(registrationDto: RegistrationDTO.registration,\r
+ bus: messagebus.MessageBus):\r
+\r
+ reg_uuid = str(uuid.uuid4())\r
+ registration = Registration(\r
+ reg_uuid, registrationDto['callback'])\r
+ with bus.uow as uow:\r
+ uow.registrations.add(registration)\r
+ logging.debug('before event length {}'.format(\r
+ len(registration.events)))\r
+ registration.events.append(events.RegistrationChanged(\r
+ reg_uuid,\r
+ datetime.now()))\r
+ logging.debug('after event length {}'.format(len(registration.events)))\r
+ uow.commit()\r
+ _handle_events(bus)\r
+ return {"registrationId": reg_uuid}\r
+\r
+\r
+def registration_delete(registrationId: str,\r
+ uow: unit_of_work.AbstractUnitOfWork):\r
+ with uow:\r
+ uow.registrations.delete(registrationId)\r
+ uow.commit()\r
+ return True\r
+\r
+\r
+def _handle_events(bus: messagebus.MessageBus):\r
+ # handle events\r
+ events = bus.uow.collect_new_events()\r
+ for event in events:\r
+ bus.handle(event)\r
+ return True\r
Code Review / pti / o2.git / blobdiff