zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-16 06:24:12 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update pipfile

Browse source
This commit is contained in:
zandercymatics 2023-10-16 13:33:11 -06:00
parent a9e4d34099
commit 7bb6bb9e7d
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
7 changed files with 36 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/epplibwrapper/utility/pool.py
View file

@ -22,6 +22,7 @@ class EPPConnectionPool(ConnectionPool):
options (dict): Options for the ConnectionPool
base class
"""
def __init__(self, client, login, options: dict):
# For storing shared credentials
self._client = client
@ -51,11 +52,11 @@ class EPPConnectionPool(ConnectionPool):
"""Creates and returns a socket instance"""
socket = Socket(client, login)
return socket
def get_connections(self):
"""Returns the connection queue"""
return self.conn
def kill_all_connections(self):
"""Kills all active connections in the pool."""
try:
@ -66,11 +67,9 @@ class EPPConnectionPool(ConnectionPool):
self.lock.release()
# TODO - connection pool err
except Exception as err:
logger.error(
"Could not kill all connections."
)
logger.error("Could not kill all connections.")
raise err
def repopulate_all_connections(self):
"""Regenerates the connection pool.
If any connections exist, kill them first.
@ -80,6 +79,4 @@ class EPPConnectionPool(ConnectionPool):
for i in range(self.size):
self.lock.acquire()
for i in range(self.size):
gevent.spawn_later(self.SPAWN_FREQUENCY*i, self._addOne)
gevent.spawn_later(self.SPAWN_FREQUENCY * i, self._addOne)