]j$3 ( d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl
m Z ddl
m Z ddl
m
Z
n# e$ r dxZ xZ Z
Y nw xY w ddlmZ n # e$ r Y nw xY wdd lmZ dd
lmZ dd lmZ ej d Zd
Z G d de Z edd Z G d de Z d Z G d de Z G d de e Z G d de Z G d d e e Z! dd l"m#Z# G d d e e# Z$n
# e$ r dZ$Y nw xY w dd l%m&Z& G d! d"e e& Z'n
# e$ r dZ'Y nw xY w G d# d$e Z( G d% d&e(e Z) dd'l*m+Z+ G d( d)e(e+ Z,n
# e$ r dZ,Y nw xY w dd*l*m-Z- G d+ d,e(e- Z.dS # e$ r dZ.Y dS w xY w)-aG
Lightweight connection pooling for peewee.
In a multi-threaded application, up to `max_connections` will be opened. Each
thread (or, if using gevent, greenlet) will have it's own connection.
In a single-threaded application, only one connection will be created. It will
be continually recycled until either it exceeds the stale timeout or is closed
explicitly (using `.manual_close()`).
By default, all your application needs to do is ensure that connections are
closed when you are finished with them, and they will be returned to the pool.
For web applications, this typically means that at the beginning of a request,
you will open a connection, and when you return a response, you will close the
connection.
Simple Postgres pool example code:
# Use the special postgresql extensions.
from playhouse.pool import PooledPostgresqlExtDatabase
db = PooledPostgresqlExtDatabase(
'my_app',
max_connections=32,
stale_timeout=300, # 5 minutes.
user='postgres')
class BaseModel(Model):
class Meta:
database = db
That's it!
N)
namedtuple)chain)TRANSACTION_STATUS_IDLE)TRANSACTION_STATUS_INERROR)TRANSACTION_STATUS_UNKNOWN)TransactionStatus)
MySQLDatabase)PostgresqlDatabase)SqliteDatabasez peewee.poolc ` | +t | t t f st | S | S N)
isinstanceintfloat)vals E/opt/imunify360/venv/lib64/python3.11/site-packages/playhouse/pool.pymake_intr > s)
z#U| < <3xx J c e Zd ZdS )MaxConnectionsExceededN__name__
__module__ __qualname__ r r r r D s r r PoolConnection) timestamp
connection checked_outc e Zd Zd ZdS ) _sentinelc dS NTr )selfothers r __lt__z_sentinel.__lt__K s tr N)r r r r& r r r r! r! J s# r r! c F t j fd }|S )Nc ^ | j 5 | g|R i |cd d d S # 1 swxY w Y d S r
)
_pool_lock)r$ argskwargsfns r innerzlocked.<locals>.innerP s
_ - -2d,T,,,V,, - - - - - - - - - - - - - - - - - -s "&&) functoolswraps)r, r- s ` r lockedr0 O s8 _R- - - - - Lr c e Zd Z d fd Z d fd Zd fd Ze fd Zd Zd Z d
Z
ed fd Z ed Z ed
Z
edd Zed Z xZS )PooledDatabase Nc d t | | _ t | | _ t | | _ | j dk rt d | _ t j | _ g | _ i | _ t | _ t t | j |fi | d S Nr inf)r _max_connections_stale_timeout
_wait_timeoutr threadingRLockr) _connections_in_useidconn_keysuperr2 __init__)r$ databasemax_connections
stale_timeouttimeoutr+ __class__s r rA zPooledDatabase.__init__X s ( 9 9 &} 5 5 %g . . " "!&u D #/++
,nd##,X@@@@@@@r c t t | j |fi | |t | | _ |t | | _ |5t | | _ | j dk rt d | _ d S d S d S r5 )r@ r2 initr r7 r8 r9 r )r$ rB rC rD rE connect_kwargsrF s r rH zPooledDatabase.initr s (nd##(DD^DDD &$,_$=$=D ! $"*="9"9D !)'!2!2D !Q&&%*5\\""" &&r Fc | j s(t t | | S t j | j z }|t j k rf t t | | }|S # t
$ r t j d Y nw xY w|t j k ft d )Ng?z:Max connections exceeded, timed out attempting to connect.)r9 r@ r2 connecttimer sleep)r$
reuse_if_openexpiresretrF s r rK zPooledDatabase.connect~ s ! F ..66}EE E)++ 22 ##
ND1199-HH
*
3
## % &> ? ? ?s $(B B/.B/c t t j | j \ }}}|}| | }| | r t
d| d x}}n| j rK| | r6t
d| | |d d x}}n/n/# t $ r! d x}}t
d Y nw xY w|| j r,t | j
| j k rt d t t | }t% j }| | }t
d| t' ||t% j | j
|< |S )NTzConnection %s was closed.z!Connection %s was stale, closing.z No connection available in pool.z Exceeded maximum connections.zCreated new connection %s.)heapqheappopr< r?
_is_closedloggerdebugr8 _is_stale_close
IndexErrorr7 lenr= r r@ r2 _connectrL r )r$ ts_c_connconnkeyrF s r r[ zPooledDatabase._connect s
%
d.? @ @
Av mmD)) ??4(( LL!<cBBB $$B( T^^B-?-?
LL!DcJJJKKd+++ $$B-
T ?@@@
< <$
N %%)>>>,-LMMM ..7799DB--%%C LL5s ; ; ; *2tTY[[ A A S s 4C (C+*C+c @ t j |z
| j k S r
)rL r8 )r$ r s r rW zPooledDatabase._is_stale s i'4+>>>r c dS )NFr r$ r_ s r rT zPooledDatabase._is_closed s ur c dS r# r rc s r
_can_reusezPooledDatabase._can_reuse s tr c | | }|r*t t | | d S || j v r| j | }| j r_| |j rEt
d| t t | | d S | | rKt
d| t j
| j |j t |f d S t
d| d S d S )Nz Closing stale connection %s.zReturning %s to pool.z
Closed %s.)r? r@ r2 rX r= popr8 rW r rU rV re rR heappushr< r! )r$ r_
close_connr` pool_connrF s r rX zPooledDatabase._close sB mmD!! 0 .$ ' ' . .t 4 4 4 4 4
DL
((--I"
0t~~i6I'J'J
0 ;SAAAnd++22488888 &&
0 4c:::t 0 ) 3Y[[$ GI I I I I \3///// !
r c | rdS | }| j | | d | | |d dS )zS
Close the underlying connection without returning it to the pool.
FNTri ) is_closedr r= rg r? closerX rc s r manual_closezPooledDatabase.manual_close s{
>> 5
t,,d333
DT *****r c ^ | j D ] \ }}}| |d g | _ d S NTrl )r< rX )r$ r] r_ s r
close_idlezPooledDatabase.close_idle sC + / /JAq$ KKK . . . . r X c i }t j |z
}d}| j D ]7\ }}|j |k r"| |j d |dz
}2|||< 8|| _ |S )Nr Trl )rL r= itemsr rX r )r$ agein_usecutoffnr` rj s r close_stalezPooledDatabase.close_stale s s"
"l 0 0 2 2 ( (NC$v-- I 0T BBBQ 's r c | | j D ] \ }}}| |d | j D ] }| |j d g | _ i | _ d S rq )rn r< rX r= valuesr )r$ r] r_ rj s r close_allzPooledDatabase.close_all s
+ / /JAq$ KKK . . . . ,,.. ? ?I KK ,K > > > > r )r3 NN)NNNF)rs )r r r rA rH rK r0 r[ rW rT re rX ro rr r{ r~
__classcell__)rF s @r r2 r2 W s} CG A A A A A A4 BF
2
2
2
2
2
2
?
?
?
?
?
? ) ) ) ) V)V? ? ?
0 0 0 0 0 V0 + + V+$ V
V
V r r2 c e Zd Zd ZdS )PooledMySQLDatabasec X | j d dk rd}nd} |j | dS # Y dS xY w)Nr r r FT)server_versionping)r$ r_ r* s r rT z PooledMySQLDatabase._is_closed sL q !Q & &DDD DIt 5 44s
$ )Nr r r rT r r r r r
s#
r r c e Zd Zd Zd ZdS )_PooledPostgresqlDatabasec |j rdS | }|t k rdS |t k r| dS NTF)closedget_transaction_statusr r rollbackr$ r_
txn_statuss r rT z$_PooledPostgresqlDatabase._is_closed sQ ; 40022
3 3 34
2
2
2 MMOOOur c | }|t k rdS |t k r| n |t k r| dS NFT)r r r resetr r r s r re z$_PooledPostgresqlDatabase._can_reuse' s_ 0022
3 3 35
5
5
5 JJLLLL
2
2
2 MMOOOtr Nr r r rT re r r r r r s2 r r c e Zd ZdS )PooledPostgresqlDatabaseNr r r r r r 4 Dr r )PostgresqlExtDatabasec e Zd ZdS )PooledPostgresqlExtDatabaseNr r r r r r : r r )Psycopg3Databasec e Zd Zd Zd ZdS )PooledPsycopg3Databasec |j rdS |j j }|t j k rdS |t j k r| dS r )r pgconntransaction_statusr UNKNOWNIDLEr r s r rT z!PooledPsycopg3Database._is_closedD sP {
t 7J . 666t 0 555
5r c |j j }|t j k rdS |t j k r| n$|t j k r| dS r )r r r r INERRORr r r r s r re z!PooledPsycopg3Database._can_reuseO sa 7J . 666 u 0 888
0 555
4r Nr r r r r r C s2 r r c e Zd Zd ZdS )_PooledSqliteDatabasec & |j dS # Y dS xY wr )
total_changesrc s r rT z _PooledSqliteDatabase._is_closed` s' 5 44s Nr r r r r r _ s# r r c e Zd ZdS )PooledSqliteDatabaseNr r r r r r h r r r )SqliteExtDatabasec e Zd ZdS )PooledSqliteExtDatabaseNr r r r r r n r r r )CSqliteExtDatabasec e Zd ZdS )PooledCSqliteExtDatabaseNr r r r r r v r r r )/__doc__r. rR loggingr: rL collectionsr itertoolsr psycopg2.extensionsr r r ImportError
psycopg.pqr peeweer r
r getLoggerrU r
ValueErrorr r objectr! r0 r2 r r r playhouse.postgres_extr r playhouse.psycopg3_extr r r r playhouse.sqlite_extr r r r r r r <module>r s B " " " " " " .;;;;;;>>>>>>>>>>>>> . . . *.. . & & & &. ,,,,,,, D ! % % % % % % ! ! ! ! ! ! = ) ) / . . . .Z . . . , /> ? ?
s s s s sV s s sl .- 2 8:L '<<<<<<
&?AV
' ' '"& ' "777777 1A 0 " " " !" N 0. #666666
"79J
# # # "#$777777
#8:L
$ $ $ # $si 7 AA A AA&C< <DD
D D*)D* E! !E+*E+/F FF