o2app/entrypoints/redis_eventconsumer.py

   1 # Copyright (C) 2021 Wind River Systems, Inc.
   2 #
   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
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   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.
  14
  15 # import json
  16 import redis
  17 import json
  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
  23
  24 from o2common.helper import o2logging
  25 from o2ims.domain.subscription_obj import Message2SMO
  26 logger = o2logging.get_logger(__name__)
  27
  28 r = redis.Redis(**config.get_redis_host_and_port())
  29
  30 apibase = config.get_o2ims_api_base()
  31
  32
  33 def main():
  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
  40     for m in pubsub.listen():
  41         try:
  42             handle_dms_changed(m, bus)
  43         except Exception as ex:
  44             logger.warning("{}".format(str(ex)))
  45             continue
  46
  47
  48 def handle_dms_changed(m, bus):
  49     logger.info("handling %s", m)
  50     channel = m['channel'].decode("UTF-8")
  51     if channel == "NfDeploymentStateChanged":
  52         datastr = m['data']
  53         data = json.loads(datastr)
  54         logger.info('HandleNfDeploymentStateChanged with cmd:{}'.format(data))
  55         cmd = commands.HandleNfDeploymentStateChanged(
  56             NfDeploymentId=data['NfDeploymentId'],
  57             FromState=data['FromState'],
  58             ToState=data['ToState']
  59         )
  60         bus.handle(cmd)
  61     if channel == 'ResourceChanged':
  62         datastr = m['data']
  63         data = json.loads(datastr)
  64         logger.info('ResourceChanged with cmd:{}'.format(data))
  65         ref = apibase + '/resourcePools/' + data['resourcePoolId'] +\
  66             '/resources/' + data['id']
  67         cmd = imscmd.PubMessage2SMO(data=Message2SMO(
  68             id=data['id'], ref=ref,
  69             eventtype=data['notificationEventType'],
  70             updatetime=data['updatetime']))
  71         bus.handle(cmd)
  72     else:
  73         logger.info("unhandled:{}".format(channel))
  74
  75
  76 if __name__ == "__main__":
  77     main()