a !f @s\dZddlZddlZddlZddlmZddlmZdZGdddeZ Gdd d e Z dS) z,Google Cloud Bigtable HappyBase pool module.N) Connection) _get_instancec@seZdZdZdS)NoConnectionsAvailablezException raised when no connections are available. This happens if a timeout was specified when obtaining a connection, and no connection became available within the specified timeout. N)__name__ __module__ __qualname____doc__r r _/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/bigtable/happybase/pool.pyrsrc@s2eZdZdZddZd ddZejd ddZdS) ConnectionPoolaThread-safe connection pool. .. note:: All keyword arguments are passed unmodified to the :class:`Connection <.happybase.connection.Connection>` constructor **except** for ``autoconnect``. This is because the ``open`` / ``closed`` status of a connection is managed by the pool. In addition, if ``instance`` is not passed, the default / inferred instance is determined by the pool and then passed to each :class:`Connection <.happybase.connection.Connection>` that is created. :type size: int :param size: The maximum number of concurrently open connections. :type kwargs: dict :param kwargs: Keyword arguments passed to :class:`Connection <.happybase.Connection>` constructor. :raises: :class:`TypeError ` if ``size`` is non an integer. :class:`ValueError ` if ``size`` is not positive. cKst|tjstd|tkr$tdt|_tj j j |d|_ t |_|}d|d<d|vrrt|dd|d<tj |D]}tfi|}|j |q~dS) Nz Pool size arg must be an integerzPool size must be positive)maxsizeFZ autoconnectinstancetimeout)r) isinstancesix integer_types TypeError_MIN_POOL_SIZE ValueError threadingLock_lockmovesqueue LifoQueue_queuelocal_thread_connectionsrgetrangerput)selfsizekwargsZconnection_kwargs_ connectionr r r __init__As     zConnectionPool.__init__NcCs8z|jjd|dWStjjjy2tdYn0dS)aAcquire a connection from the pool. :type timeout: int :param timeout: (Optional) Time (in seconds) to wait for a connection to open. :rtype: :class:`Connection <.happybase.Connection>` :returns: An active connection from the queue stored on the pool. :raises: :class:`NoConnectionsAvailable` if ``Queue.get`` fails before the ``timeout`` (only if a timeout is specified). T)blockrz:No connection available from pool within specified timeoutN)rrrrrEmptyr)r"rr r r _acquire_connectionVs z"ConnectionPool._acquire_connectionccs~t|jdd}d}|durVd}||}|j||j_Wdn1sL0Y||V|rz|j`|j|dS)aObtain a connection from the pool. Must be used as a context manager, for example:: with pool.connection() as connection: pass # do something with the connection If ``timeout`` is omitted, this method waits forever for a connection to become available from the local queue. Yields an active :class:`Connection <.happybase.connection.Connection>` from the pool. :type timeout: int :param timeout: (Optional) Time (in seconds) to wait for a connection to open. :raises: :class:`NoConnectionsAvailable` if no connection can be retrieved from the pool before the ``timeout`` (only if a timeout is specified). currentNFT)getattrrr*rr+openrr!)r"rr&Zretrieved_new_cnxnr r r r&hs  &zConnectionPool.connection)N)N) rrrr r'r* contextlibcontextmanagerr&r r r r r 's  r ) r r.rrZ$gcloud.bigtable.happybase.connectionrrr RuntimeErrorrobjectr r r r r s