Add info line on every 1000 lines
This commit is contained in:
@ -27,6 +27,7 @@ class RouterstatsClient():
|
||||
self.connected = False
|
||||
self.sock = None
|
||||
self.passwd_file = passwd_file
|
||||
self.received_lines = 0
|
||||
|
||||
def connect(self):
|
||||
'''Do connect'''
|
||||
@ -37,6 +38,7 @@ class RouterstatsClient():
|
||||
logging.info('Connected to %s', self.host)
|
||||
self.login()
|
||||
self.connected = True
|
||||
self.received_lines = 0
|
||||
except ConnectionRefusedError as error:
|
||||
logging.error('Could not connect to %s:%s', self.host, self.port)
|
||||
raise ConnectionRefusedError from error
|
||||
@ -118,6 +120,9 @@ class RouterstatsClient():
|
||||
#We're done for now, returning value
|
||||
line = line.strip().decode('utf-8')
|
||||
logging.debug('Received from server: %s', line)
|
||||
self.received_lines += 1
|
||||
if not self.received_lines % 1000:
|
||||
logging.info('Received %s lines', self.received_lines)
|
||||
return line
|
||||
else:
|
||||
blanks += 1
|
||||
|
||||
@ -282,6 +282,7 @@ class RequestHandler(socketserver.BaseRequestHandler):
|
||||
|
||||
def handle(self):
|
||||
logging.info('Connected to %s', self.client_address[0])
|
||||
self.sendt_lines = 0
|
||||
self.request.settimeout(5)
|
||||
start_time = datetime.now()
|
||||
if not self.login():
|
||||
@ -339,6 +340,9 @@ class RequestHandler(socketserver.BaseRequestHandler):
|
||||
'''Wrap sendall'''
|
||||
logging.debug('Sending %s', tosend)
|
||||
self.request.sendall(bytes(tosend + "\n", 'utf-8'))
|
||||
self.sendt_lines += 1
|
||||
if not self.sendt_lines % 1000:
|
||||
logging.info('Sendt %s lines', self.sendt_lines)
|
||||
|
||||
def set_queue(self, input_queue, overflowqueue, signal_queue):
|
||||
'''Set Queue for fetching events'''
|
||||
|
||||
Reference in New Issue
Block a user