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.
18 from o2app import bootstrap
19 from o2common.config import config
20 # from o2common.domain import commands
21 from o2dms.domain import commands
22 from o2ims.domain import commands as imscmd
24 from o2common.helper import o2logging
25 from o2ims.domain.subscription_obj import Message2SMO, NotificationEventEnum, RegistrationMessage
26 logger = o2logging.get_logger(__name__)
28 r = redis.Redis(**config.get_redis_host_and_port())
30 apibase = config.get_o2ims_api_base()
34 logger.info("Redis pubsub starting")
35 bus = bootstrap.bootstrap()
36 pubsub = r.pubsub(ignore_subscribe_messages=True)
37 pubsub.subscribe("NfDeploymentStateChanged")
38 pubsub.subscribe('ResourceChanged')
39 pubsub.subscribe('RegistrationChanged')
40 pubsub.subscribe('OcloudChanged')
42 for m in pubsub.listen():
44 handle_dms_changed(m, bus)
45 except Exception as ex:
46 logger.warning("{}".format(str(ex)))
50 def handle_dms_changed(m, bus):
51 logger.info("handling %s", m)
52 channel = m['channel'].decode("UTF-8")
53 if channel == "NfDeploymentStateChanged":
55 data = json.loads(datastr)
56 logger.info('HandleNfDeploymentStateChanged with cmd:{}'.format(data))
57 cmd = commands.HandleNfDeploymentStateChanged(
58 NfDeploymentId=data['NfDeploymentId'],
59 FromState=data['FromState'],
60 ToState=data['ToState']
63 elif channel == 'ResourceChanged':
65 data = json.loads(datastr)
66 logger.info('ResourceChanged with cmd:{}'.format(data))
67 ref = apibase + '/resourcePools/' + data['resourcePoolId'] +\
68 '/resources/' + data['id']
69 cmd = imscmd.PubMessage2SMO(data=Message2SMO(
70 id=data['id'], ref=ref,
71 eventtype=data['notificationEventType'],
72 updatetime=data['updatetime']))
74 elif channel == 'RegistrationChanged':
76 data = json.loads(datastr)
77 logger.info('RegistrationChanged with cmd:{}'.format(data))
78 cmd = imscmd.Register2SMO(data=RegistrationMessage(id=data['id']))
80 elif channel == 'OcloudChanged':
82 data = json.loads(datastr)
83 logger.info('OcloudChanged with cmd:{}'.format(data))
84 if data['notificationEventType'] == NotificationEventEnum.CREATE:
85 cmd = imscmd.Register2SMO(data=RegistrationMessage(is_all=True))
88 logger.info("unhandled:{}".format(channel))
91 if __name__ == "__main__":