Files
opensearch-pyd/elasticsearch/connection/pooling.py
T

21 lines
541 B
Python
Raw Normal View History

2014-09-23 01:13:04 +12:00
try:
import queue
except ImportError:
import Queue as queue
from .base import Connection
class PoolingConnection(Connection):
def __init__(self, *args, **kwargs):
2014-09-23 01:13:04 +12:00
self._free_connections = queue.Queue()
super(PoolingConnection, self).__init__(*args, **kwargs)
def _get_connection(self):
2014-09-23 01:13:04 +12:00
try:
return self._free_connections.get_nowait()
except queue.Empty:
return self._make_connection()
def _release_connection(self, con):
2015-03-23 15:25:45 -07:00
self._free_connections.put(con)