talos-python3 3.1.1__tar.gz → 3.1.3__tar.gz

{talos-python3-3.1.1/talos_python3.egg-info → talos_python3-3.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
-Name: talos-python3
-Version: 3.1.1
+Name: talos_python3
+Version: 3.1.3
 Summary: talos python3 sdk
 Home-page: UNKNOWN
 Author: huyumei

{talos-python3-3.1.1 → talos_python3-3.1.3}/setup.py

@@ -1,8 +1,8 @@
 from setuptools import find_packages, setup
 
 setup(
-    name='talos-python3',
-    version='3.1.1',
+    name='talos_python3',
+    version='3.1.3',
     author='huyumei',
     author_email='huyumei@xiaomi.com',
     maintainer='fangchengjin',

talos_python3-3.1.3/talos/example/test.py

@@ -0,0 +1,46 @@
+# Talos Consumer 封装类
+from talos.client.TalosClientConfig import TalosClientConfig
+from talos.example.TalosConsumerDemo import MyMessageProcessorFactory
+from talos.thrift.auth.ttypes import Credential
+from talos.thrift.topic.ttypes import TopicAndPartition
+from talos.consumer.TalosConsumer import TalosConsumer
+from talos.consumer.MessageProcessor import MessageProcessor
+from talos.consumer.MessageProcessorFactory import MessageProcessorFactory
+from talos.thrift.auth.ttypes import UserType
+from atomic import AtomicLong
+import logging
+import traceback
+
+class TalosConsumerWrapper:
+    accessKey = "AKM5DDIDEBDWBBD2PV"
+    accessSecret = "w8nium9H1/1milUJkHAVGhpiDRchLxnd4ZAUbdE6"
+    consumerGroup = "ConsumerGroup-fcj"
+    clientPrefix = "ClientPrefix"
+    pro = {
+        "galaxy.talos.service.endpoint": "http://ap-tjv1autopilotsrv-talos.api.xiaomi.net",
+        "galaxy.talos.client.falcon.monitor.switch": False
+    }
+    count = 1
+
+    def __init__(self):
+        self.consumerConfig = TalosClientConfig(self.pro)
+        self.credential = Credential(UserType.DEV_XIAOMI, self.accessKey, self.accessSecret)
+        self.talosConsumer = None
+
+    def start(self, topic_name, partitionCheckPoint=None):
+        """
+        启动消费者。
+        """
+        # logger.info(f"start consumer: service_manager={service_manager}, factory={factory}, topic_name={topic_name}")
+        self.talosConsumer = TalosConsumer(
+            consumerGroup=self.consumerGroup,
+            consumerConfig=self.consumerConfig,
+            credential=self.credential,
+            topicName=topic_name,
+            messageProcessorFactory= MyMessageProcessorFactory(),
+            clientPrefix=self.clientPrefix
+        )
+        return self.talosConsumer
+
+consumerDemo = TalosConsumerWrapper()
+consumerDemo.start("geo_result_receive_preview")

{talos-python3-3.1.1 → talos_python3-3.1.3}/talos/producer/TalosProducer.py

@@ -236,7 +236,7 @@ class TalosProducer(object):
                     userMessage = UserMessage(message)
                     partitionBufferMap.get(currentPartitionId).append(userMessage)
                 else:
-                    self.check_message_partition_key_validity(message.getPartitionKey())
+                    self.check_message_partition_key_validity(message.partitionKey)
                     # construct UserMessage and dispatch to buffer by partitionId
                     partitionId = self.get_partition_id(message.partitionKey)
                     if partitionId not in partitionBufferMap:
@@ -261,10 +261,8 @@ class TalosProducer(object):
                                                     self.updatePartitionIdMsgNumber)
 
     def check_message_partition_key_validity(self, partitionKey=None):
-        try:
-            assert not partitionKey
-        except AssertionError as e:
-            raise e
+        if not partitionKey:
+            raise InvalidArgumentError("Field \"partitionKey\" must be set")
 
         if len(partitionKey) < self.partitionKeyMinLen or len(partitionKey) > self.partitionKeyMaxLen:
             raise InvalidArgumentError("Invalid partition key which length must be at "

{talos-python3-3.1.1 → talos_python3-3.1.3/talos_python3.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: talos-python3
-Version: 3.1.1
+Version: 3.1.3
 Summary: talos python3 sdk
 Home-page: UNKNOWN
 Author: huyumei

{talos-python3-3.1.1 → talos_python3-3.1.3}/talos_python3.egg-info/SOURCES.txt

@@ -41,6 +41,7 @@ talos/example/TalosAdminDemo.py
 talos/example/TalosConsumerDemo.py
 talos/example/TalosProducerDemo.py
 talos/example/__init__.py
+talos/example/test.py
 talos/producer/BufferedMessageCount.py
 talos/producer/PartitionMessageQueue.py
 talos/producer/PartitionSender.py