1 # Copyright (C) 2021 Wind River Systems, Inc.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
16 from typing import Callable
17 from o2ims.adapter import orm, redis_eventpublisher
18 from o2ims.adapter.notifications import AbstractNotifications,\
21 from o2ims.service import handlers, messagebus, unit_of_work
22 from o2ims.adapter.unit_of_work import SqlAlchemyUnitOfWork
23 from o2ims.adapter.clients import orm_stx
27 start_orm: bool = True,
28 uow: unit_of_work.AbstractUnitOfWork = SqlAlchemyUnitOfWork(),
29 notifications: AbstractNotifications = None,
30 publish: Callable = redis_eventpublisher.publish,
31 ) -> messagebus.MessageBus:
33 if notifications is None:
34 notifications = SmoO2Notifications()
37 orm.start_o2ims_mappers()
38 orm_stx.start_o2ims_stx_mappers(uow)
40 dependencies = {"uow": uow, "notifications": notifications,
42 injected_event_handlers = {
44 inject_dependencies(handler, dependencies)
45 for handler in event_handlers
47 for event_type, event_handlers in handlers.EVENT_HANDLERS.items()
49 injected_command_handlers = {
50 command_type: inject_dependencies(handler, dependencies)
51 for command_type, handler in handlers.COMMAND_HANDLERS.items()
54 return messagebus.MessageBus(
56 event_handlers=injected_event_handlers,
57 command_handlers=injected_command_handlers,
61 def inject_dependencies(handler, dependencies):
62 params = inspect.signature(handler).parameters
65 for name, dependency in dependencies.items()
68 return lambda message: handler(message, **deps)