-
Notifications
You must be signed in to change notification settings - Fork 0
/
simple-producer.py
69 lines (52 loc) · 1.71 KB
/
simple-producer.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import json
from confluent_kafka import Producer
import sys, pymysql
#db connec
def openDb():
global conn, cursor
conn = pymysql.connect(
host="localhost",
user="root",
password="",
db="db_kafka" )
cursor = conn.cursor()
#db closing conn
def closeDb():
global conn, cursor
cursor.close()
conn.close()
def delivery_callback(err, msg):
if err:
sys.stderr.write('%% Message failed delivery: %s\n' % err)
else:
sys.stderr.write('%% Message delivered to %s [%d] @ %d\n' %
(msg.topic(), msg.partition(), msg.offset()))
if __name__ == '__main__':
broker = "localhost:9092"
conf = {'bootstrap.servers': broker}
# Create Producer instance
p = Producer(**conf)
topic = "bukan_testing"
msg = "tidak tahu"
msg = {"message":"test_message"}
msg = json.dumps(msg)
msg = msg.encode("utf-8")
p.produce(topic, msg, callback=delivery_callback)
'''
# Read lines from stdin, produce each line to Kafka
for line in sys.stdin:
try:
# Produce line (without newline)
p.produce(topic, msg, callback=delivery_callback)
except BufferError:
sys.stderr.write('%% Local producer queue is full (%d messages awaiting delivery): try again\n' %
len(p))
# Serve delivery callback queue.
# NOTE: Since produce() is an asynchronous API this poll() call
# will most likely not serve the delivery callback for the
# last produce()d message.
p.poll(0)
'''
# Wait until all messages have been delivered
sys.stderr.write('%% Waiting for %d deliveries\n' % len(p))
p.flush()