def test_correlation_id_rollover(kafka_broker): logging.getLogger('kafka.conn').setLevel(logging.ERROR) from kafka.protocol.metadata import MetadataRequest conn = BrokerConnection('localhost', kafka_broker.port, receive_buffer_bytes=131072, max_in_flight_requests_per_connection=100) req = MetadataRequest([]) while not conn.connected(): conn.connect() futures = collections.deque() start = time.time() done = 0 for i in six.moves.xrange(2**13): if not conn.can_send_more(): conn.recv(timeout=None) futures.append(conn.send(req)) conn.recv() while futures and futures[0].is_done: f = futures.popleft() if not f.succeeded(): raise f.exception done += 1 if time.time() > start + 10: print("%d done" % done) start = time.time() while futures: conn.recv() if futures[0].is_done: f = futures.popleft() if not f.succeeded(): raise f.exception
def test_correlation_id_rollover(kafka_broker): logging.getLogger('kafka.conn').setLevel(logging.ERROR) from kafka.protocol.metadata import MetadataRequest conn = BrokerConnection('localhost', kafka_broker.port, receive_buffer_bytes=131072, max_in_flight_requests_per_connection=100) req = MetadataRequest([]) while not conn.connected(): conn.connect() futures = collections.deque() start = time.time() done = 0 for i in six.moves.xrange(2**13): if not conn.can_send_more(): conn.recv(timeout=None) futures.append(conn.send(req)) conn.recv() while futures and futures[0].is_done: f = futures.popleft() if not f.succeeded(): raise f.exception done += 1 if time.time() > start + 10: print ("%d done" % done) start = time.time() while futures: conn.recv() if futures[0].is_done: f = futures.popleft() if not f.succeeded(): raise f.exception
def _bootstrap(self, hosts): log.info('Bootstrapping cluster metadata from %s', hosts) # Exponential backoff if bootstrap fails backoff_ms = self.config[ 'reconnect_backoff_ms'] * 2**self._bootstrap_fails next_at = self._last_bootstrap + backoff_ms / 1000.0 self._refresh_on_disconnects = False now = time.time() if next_at > now: log.debug("Sleeping %0.4f before bootstrapping again", next_at - now) time.sleep(next_at - now) self._last_bootstrap = time.time() if self.config['api_version'] is None or self.config['api_version'] < ( 0, 10): metadata_request = MetadataRequest[0]([]) else: metadata_request = MetadataRequest[1](None) for host, port, afi in hosts: log.debug("Attempting to bootstrap via node at %s:%s", host, port) cb = functools.partial(WeakMethod(self._conn_state_change), 'bootstrap') bootstrap = BrokerConnection(host, port, afi, state_change_callback=cb, node_id='bootstrap', **self.config) if not bootstrap.connect_blocking(): bootstrap.close() continue future = bootstrap.send(metadata_request) while not future.is_done: self._selector.select(1) for r, f in bootstrap.recv(): f.success(r) if future.failed(): bootstrap.close() continue self.cluster.update_metadata(future.value) log.info('Bootstrap succeeded: found %d brokers and %d topics.', len(self.cluster.brokers()), len(self.cluster.topics())) # A cluster with no topics can return no broker metadata # in that case, we should keep the bootstrap connection if not len(self.cluster.brokers()): self._conns['bootstrap'] = bootstrap else: bootstrap.close() self._bootstrap_fails = 0 break # No bootstrap found... else: log.error('Unable to bootstrap from %s', hosts) # Max exponential backoff is 2^12, x4000 (50ms -> 200s) self._bootstrap_fails = min(self._bootstrap_fails + 1, 12) self._refresh_on_disconnects = True
def _bootstrap(self, hosts): log.info('Bootstrapping cluster metadata from %s', hosts) # Exponential backoff if bootstrap fails backoff_ms = self.config['reconnect_backoff_ms'] * 2 ** self._bootstrap_fails next_at = self._last_bootstrap + backoff_ms / 1000.0 self._refresh_on_disconnects = False now = time.time() if next_at > now: log.debug("Sleeping %0.4f before bootstrapping again", next_at - now) time.sleep(next_at - now) self._last_bootstrap = time.time() if self.config['api_version'] is None or self.config['api_version'] < (0, 10): metadata_request = MetadataRequest[0]([]) else: metadata_request = MetadataRequest[1](None) for host, port, afi in hosts: log.debug("Attempting to bootstrap via node at %s:%s", host, port) cb = functools.partial(WeakMethod(self._conn_state_change), 'bootstrap') bootstrap = BrokerConnection(host, port, afi, state_change_callback=cb, node_id='bootstrap', **self.config) if not bootstrap.connect_blocking(): bootstrap.close() continue future = bootstrap.send(metadata_request) while not future.is_done: self._selector.select(1) for r, f in bootstrap.recv(): f.success(r) if future.failed(): bootstrap.close() continue self.cluster.update_metadata(future.value) log.info('Bootstrap succeeded: found %d brokers and %d topics.', len(self.cluster.brokers()), len(self.cluster.topics())) # A cluster with no topics can return no broker metadata # in that case, we should keep the bootstrap connection if not len(self.cluster.brokers()): self._conns['bootstrap'] = bootstrap else: bootstrap.close() self._bootstrap_fails = 0 break # No bootstrap found... else: log.error('Unable to bootstrap from %s', hosts) # Max exponential backoff is 2^12, x4000 (50ms -> 200s) self._bootstrap_fails = min(self._bootstrap_fails + 1, 12) self._refresh_on_disconnects = True
def test_recv_disconnected(): sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.bind(('127.0.0.1', 0)) port = sock.getsockname()[1] sock.listen(5) conn = BrokerConnection('127.0.0.1', port, socket.AF_INET) timeout = time.time() + 1 while time.time() < timeout: conn.connect() if conn.connected(): break else: assert False, 'Connection attempt to local socket timed-out ?' conn.send(MetadataRequest[0]([])) # Disconnect server socket sock.close() # Attempt to receive should mark connection as disconnected assert conn.connected() conn.recv() assert conn.disconnected()