p.py 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. #!/usr/bin/python
  2. # -*- coding=utf-8 -*-
  3. # @Create Time: 2024-01-10 16:16:18
  4. # @Last Modified time:2024-01-10 16:16:18
  5. import sys, time
  6. from rocketmq.client import Producer, Message
  7. topicName = 'test'
  8. groupName = 'log'
  9. nameserver = 'localhost:9876'
  10. KEYS = 'XXX'
  11. TAGS = 'XXX'
  12. def create_message():
  13. # 组装消息
  14. msg = Message(topicName)
  15. # 设置keys
  16. msg.set_tags(KEYS)
  17. # 设置tags
  18. msg.set_tags(TAGS)
  19. msg.set_property('property', 'test')
  20. # 消息内容
  21. msg.set_body('message body')
  22. return msg
  23. def send_message_sync(count):
  24. # 初始化生产者,并设置生产组消息,组名称使用全称
  25. producer = Producer(groupName)
  26. # 设置服务器地址
  27. producer.set_name_server_address(nameserver)
  28. # 启动生产者
  29. producer.start()
  30. for n in range(count):
  31. msg = create_message()
  32. # 发送同步消息
  33. ret = producer.send_sync(msg)
  34. print(ret.status, ret.msg_id, ret.offset)
  35. # 资源释放
  36. producer.shutdown()
  37. if __name__ == '__main__':
  38. send_message_sync(10)