mq_api.py 1.3 KB

123456789101112131415161718192021222324252627282930313233343536
  1. import json
  2. from mq_http_sdk.mq_exception import MQExceptionBase
  3. from mq_http_sdk.mq_producer import TopicMessage
  4. from mq_http_sdk.mq_client import MQClient
  5. # from common.common import Common
  6. class MQ:
  7. instance_id = "MQ_INST_1894469520484605_BXhXuzkZ"
  8. def __init__(self, topic_name) -> None:
  9. self.mq_client = MQClient(
  10. "http://1894469520484605.mqrest.cn-qingdao-public.aliyuncs.com",
  11. "LTAI4G7puhXtLyHzHQpD6H7A",
  12. "nEbq3xWNQd1qLpdy2u71qFweHkZjSG",
  13. )
  14. self.producer = self.mq_client.get_producer(self.instance_id, topic_name)
  15. def send_msg(self, video_dict):
  16. strategy = video_dict["strategy"]
  17. platform = video_dict["platform"]
  18. try:
  19. msg = TopicMessage(json.dumps(video_dict))
  20. msg.set_message_key(
  21. platform + "-" + strategy + "-" + video_dict["out_video_id"]
  22. )
  23. re_msg = self.producer.publish_message(msg)
  24. # Common.logger(strategy, platform).info(
  25. # "Publish Message Succeed. MessageID:%s, BodyMD5:%s\n"
  26. # % (re_msg.message_id, re_msg.message_body_md5)
  27. # )
  28. except MQExceptionBase as e:
  29. print(e)
  30. # Common.logger(strategy, platform).error(
  31. # "Publish Message Fail. Exception:%s\n" % e
  32. # )