import sys
from pykafka import KafkaClient
from pykafka.balancedconsumer import BalancedConsumer
from pykafka.simpleconsumer import OwnedPartition, OffsetType
reload(sys)
sys.setdefaultencoding('utf8')
#pykafka, need install PyKafka
class PyKafka:
consumer = None
TOPIC = 'log_download'
BROKER_LIST = '10.23.23.24:9092,10.23.23.21:9092'
ZK_LIST = '10.23.23.24:2181,10.23.23.21:2181/sh-bt'
server = topic = zsServer = None
def __init__(self):
print("begin pykafka")
self.server = self.BROKER_LIST
self.topic = self.TOPIC
self.zkServer= self.ZK_LIST
def getConnect(self):
client = KafkaClient(hosts=self.server)
topic = client.topics[self.topic]
self.consumer = topic.get_balanced_consumer(
consumer_group="zs_download_04", # 自己命令
auto_offset_reset=