main.py 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. import json
  2. from logs.logger import *
  3. from service.standard_service import StandardService
  4. from data_initialize_standard.results import *
  5. from communication.rabbitmq_link import create_connect, send_data_to_rabbitmq
  6. # from communication.rocketmq_link import start_rocket, send_data_to_rocket
  7. def main(data_input):
  8. dict_input = json.loads(data_input)
  9. logger.critical("============Get data from rabbitmq============")
  10. logger.critical("Input: {}".format(json.dumps(dict_input, ensure_ascii=False)))
  11. try:
  12. if dict_input['controlConfigInfo']['handleService'] == 1:
  13. logger.critical('============算法服务开启,进入水温智控算法============')
  14. mcss = StandardService(dict_input)
  15. dict_results = mcss.main_multi_comp_service()
  16. else:
  17. logger.critical('============算法服务关闭,退出水温智控算法============')
  18. dict_results = {}
  19. logger.critical("============Send calculated results to rabbitmq============")
  20. dict_results = json.dumps(dict_results, ensure_ascii=False)
  21. logger.critical("Output: {}".format(dict_results))
  22. send_data_to_rabbitmq(dict_results)
  23. except Exception as e:
  24. result_depict = '异常中断,退出水温智控算法' + "(" + str(e) + ")"
  25. logger.critical('============{}============'.format(result_depict))
  26. dict_results = DictResultsAbnormal(dict_input, result_depict)
  27. logger.critical("============Send calculated results to rabbitmq============")
  28. dict_results = json.dumps(dict_results, ensure_ascii=False, default=default_dump)
  29. logger.critical("Output: {}".format(dict_results))
  30. send_data_to_rabbitmq(dict_results)
  31. # send_data_to_rocket(dict_results)
  32. if __name__ == '__main__':
  33. logger.critical("========Create connection for rabbitmq========")
  34. create_connect()
  35. # start_rocket()