openstack/monasca-common
Code Issues Proposed changes

Ensures that messages pass to kafka client are bytes string

Browse Source 
Story: 2003178
Task: 23324

Change-Id: I3ae6830fbbe80bf56726886d658c55dda965d717
This commit is contained in:
Adrian Czarnecki 2018-07-26 11:22:03 +02:00
parent 55dbc317ff
commit 304772fa18
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
monasca_common/kafka/producer.py
View File

@ -16,6 +16,7 @@
import logging import logging
import time import time
from oslo_utils import encodeutils
from six import PY3 from six import PY3
import monasca_common.kafka_lib.client as kafka_client import monasca_common.kafka_lib.client as kafka_client
@ -53,11 +54,11 @@ class KafkaProducer(object):
success = False success = False
if key is None: if key is None:
key = int(time.time() * 1000) key = int(time.time() * 1000)
if PY3:
key = bytes(str(key), 'utf-8') messages = [encodeutils.to_utf8(m) for m in messages]
messages = [m.encode("utf-8") for m in messages]
else: key = bytes(str(key), 'utf-8') if PY3 else str(key)
key = str(key)
while not success: while not success:
try: try:
self._producer.send_messages(topic, key, *messages) self._producer.send_messages(topic, key, *messages)