diff --git a/python_kafka_logging/KafkaHandler.py b/python_kafka_logging/KafkaHandler.py index 4608457..55b4947 100644 --- a/python_kafka_logging/KafkaHandler.py +++ b/python_kafka_logging/KafkaHandler.py @@ -1,15 +1,14 @@ -from kafka.client import KafkaClient +from kafka.client import SimpleClient from kafka.producer import SimpleProducer, KeyedProducer -from kafka.conn import DEFAULT_SOCKET_TIMEOUT_SECONDS import logging class KafkaLoggingHandler(logging.Handler): - def __init__(self, hosts_list, topic, timeout_secs=DEFAULT_SOCKET_TIMEOUT_SECONDS, **kwargs): + def __init__(self, hosts_list, topic, **kwargs): logging.Handler.__init__(self) - self.kafka_client = KafkaClient(hosts_list, timeout=timeout_secs) + self.kafka_client = SimpleClient(hosts_list) self.key = kwargs.get("key", None) self.kafka_topic_name = topic diff --git a/requirements.txt b/requirements.txt index bd17944..17c5a66 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ -kafka-python==0.9.5 +kafka-python==1.3.1 logstash-formatter==0.5.8