Python向kafka發訊息

海布里Simple發表於2024-11-03
PythonKafka

後端研發可以提供一個向kafka發訊息的介面，用requests向介面post訊息就行：

import requests
import json
import time

now = int(time.time())
n = 10
while n > 0:
    tt = now - n * 60
    data = {
        "queue": "alarm-dog-alarm-dog-test",
        "payload": "{\"test\":80,\"notice_time\":%d}" % tt
    }
    header = {"Content-Type": "application/json"}
    res = requests.post(url="http://10.90.100.130:8088/v1/kafka/send", headers=header, data=json.dumps(data))
    print(res.status_code)
    print(res.content)
    n -= 1

如果沒有提供介面，可以藉助python-kafka庫連線kafka，模擬生產者向kafka發訊息：

同步傳送訊息：

from kafka import KafkaProducer
import json

# 建立一個KafkaProducer例項，指定Kafka伺服器地址
producer = KafkaProducer(bootstrap_servers='http://10.90.100.130:8088')
# 要傳送的訊息內容
message = {'test': 80, 'notice_time': 5}
# 將訊息轉換為JSON字串格式（也可以是其他格式，如純文字）
message_json = json.dumps(message)
# 傳送訊息到指定的Kafka主題，這裡主題名稱是'my_topic'
producer.send('alarm-dog-alarm-dog-test', value=message_json.encode('utf - 8'))
# 確保所有訊息都已傳送
producer.flush()
# 關閉生產者連線
producer.close()

非同步傳送訊息

from kafka import KafkaProducer
import json
import time

# 建立一個KafkaProducer例項，設定非同步傳送和回撥函式
producer = KafkaProducer(bootstrap_servers='http://10.90.100.130:8088',
          acks='all',
          retries=3,
          value_deliver_callback=lambda m: print(f"訊息已傳送到主題{m.topic()}，分割槽{m.partition()}"))

# 要傳送的訊息內容
message = {'test': 80, 'notice_time': 6}
message_json = json.dumps(message)

# 非同步傳送訊息到'my_topic'主題
future = producer.send('alarm-dog-alarm-dog-test', value=message_json.encode('utf - 8'))
try:
record_metadata = future.get(timeout=10)
print(f"訊息已傳送到主題{record_metadata.topic()}，分割槽{record_metadata.partition()}，偏移量{record_metadata.offset()}")
except Exception as e:
print(f"傳送訊息時出錯: {e}")
# 關閉生產者連線
producer.close()

相關文章