import array import errno import socket from kafka.base import BaseKafka, logging, StringIO, ConnectionFailure socket_log = logging.getLogger('kafka.socket') __all__ = [ 'Kafka', ] class Kafka(BaseKafka): def __init__(self, *args, **kwargs): BaseKafka.__init__(self, *args, **kwargs) self._socket = None self._overflow = '' self.total_read = 0 # Socket management methods def _connect(self): """ Connect to the Kafka server. """ self._socket = socket.socket() try: self._socket.connect((self.host, self.port)) except Exception, e: self._socket = None raise ConnectionFailure(
import socket from kafka.base import BaseKafka, logging, ConnectionFailure socket_log = logging.getLogger('kafka.iostream') from tornado.iostream import IOStream __all__ = [ 'KafkaTornado', ] class KafkaTornado(BaseKafka): def __init__(self, *args, **kwargs): if 'io_loop' in kwargs: self._io_loop = kwargs['io_loop'] del kwargs['io_loop'] else: self._io_loop = None BaseKafka.__init__(self, *args, **kwargs) self._stream = None # Socket management methods def _connect(self): """ Connect to the Kafka server. """ sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0) try:
import array import errno import socket from kafka.base import BaseKafka, logging, StringIO, ConnectionFailure socket_log = logging.getLogger('kafka.socket') __all__ = [ 'Kafka', ] class Kafka(BaseKafka): def __init__(self, *args, **kwargs): BaseKafka.__init__(self, *args, **kwargs) self._socket = None self._overflow = '' self.total_read = 0 # Socket management methods def _connect(self): """ Connect to the Kafka server. """ self._socket = socket.socket() try: self._socket.connect((self.host, self.port)) except Exception, e: self._socket = None raise ConnectionFailure("Could not connect to kafka at {0}:{1}".format(self.host, self.port))