From abf3a57baaf9f59948235c75a915016b2148eb7b Mon Sep 17 00:00:00 2001 From: Ed-XCF Date: Thu, 21 Jul 2022 15:28:34 +0800 Subject: [PATCH] support more function --- .../hooks/aliyun_rocketmq.py | 21 ++++++++++++++++--- 1 file changed, 18 insertions(+), 3 deletions(-) diff --git a/aliyun_rocketmq_provider/hooks/aliyun_rocketmq.py b/aliyun_rocketmq_provider/hooks/aliyun_rocketmq.py index 0e9feea..91ca3de 100644 --- a/aliyun_rocketmq_provider/hooks/aliyun_rocketmq.py +++ b/aliyun_rocketmq_provider/hooks/aliyun_rocketmq.py @@ -49,13 +49,28 @@ def run( self, message_body: str, message_tag: str = None, + message_key: str = None, + trans_check_immunity_time: int = None, + start_deliver_time: int = None, + sharding_key: str = None, fail_silently: bool = False ) -> TopicMessage: """Publish the data.""" + message = TopicMessage(message_body) + + if message_tag is not None: + message.set_message_tag(message_tag.lower()) + if message_key is not None: + message.set_message_key(message_key) + if trans_check_immunity_time is not None: + message.set_trans_check_immunity_time(trans_check_immunity_time) + if start_deliver_time is not None: + message.set_start_deliver_time(start_deliver_time) + if sharding_key is not None: + message.set_sharding_key(sharding_key) + try: - conn = self.get_conn() - message = TopicMessage(message_body, (message_tag or "").lower()) - return conn.publish_message(message) + return self.get_conn().publish_message(message) except Exception as e: if not fail_silently: raise