ip_to_geo : reset connection after 50 requests

This commit is contained in:
Gregory Soutade 2017-09-05 07:34:27 +02:00
parent b2f37d83f4
commit 9f9bfebe7a

View File

@ -45,12 +45,15 @@ class IPToGeo(object):
4 : 'Bad IP version',
5 : 'Unsupported IP version',
6 : 'IP not found'}
MAX_REQUESTS = 50
def __init__(self, remote_addr='127.0.0.1', remote_port=53333, timeout=None, family=socket.AF_INET):
self._remote_addr = remote_addr
self._remote_port = remote_port
self._timeout = timeout
self._family = family
self._nb_requests_sent = 0
self._create_socket()
@ -108,7 +111,20 @@ class IPToGeo(object):
(cc0, cc1, cc2, cc3) = struct.unpack_from('BBBB', packet, 7*4)
return (ip_res, '%c%c%c%c' % (cc0, cc1, cc2, cc3))
def _send_request(self, packet):
self._nb_requests_sent += 1
if self._nb_requests_sent == self.MAX_REQUESTS:
self.close()
self._create_socket()
self._nb_requests_sent = 0
try:
self._socket.send(packet)
except IOError, e:
# Give another chance (we may have been disconnected due to timeout)
self._create_socket()
self._socket.send(packet)
def ip_to_geo(self, ip):
ip_type = IPToGeo.IPV4
if ip.find('.') >= 0: