فهرست منبع

etl mq optimize

ehlxr 1 سال پیش
والد
کامیت
b817f6b649
1فایلهای تغییر یافته به همراه5 افزوده شده و 4 حذف شده
  1. 5 4
      common/mq.py

+ 5 - 4
common/mq.py

@@ -16,14 +16,15 @@ class MQ:
                                                     topic_name)
 
     def send_msg(self, video_dict):
+        strategy = video_dict["strategy"]
+        platform = video_dict["platform"]
         try:
             msg = TopicMessage(json.dumps(video_dict))
-            msg.set_message_key(video_dict["strategy"] + "-" +
-                                video_dict["out_video_id"])
+            msg.set_message_key(strategy + "-" + video_dict["out_video_id"])
             re_msg = self.producer.publish_message(msg)
-            Common.logger(video_dict["strategy"], video_dict["platform"]).info(
+            Common.logger(strategy, platform).info(
                 "Publish Message Succeed. MessageID:%s, BodyMD5:%s" %
                 (re_msg.message_id, re_msg.message_body_md5))
         except MQExceptionBase as e:
-            Common.logger(video_dict["strategy"], video_dict["platform"]).error(
+            Common.logger(strategy, platform).error(
                 "Publish Message Fail. Exception:%s" % e)