ugddlZddlZddlmZmZddlmZmZddlZddl m Z ddl m Z GddZ Gdd eZGd d eZGd d eZy)N)ABCabstractmethod)AnySet)override) Annotatedc eZdZUdZded<eed<ejed<dddeded e d e f d Z dd ed ejfdZ ddZ ddZd ejfdZddZy) ConnectionzBA threadpool connection that returns itself to the pool on close()Pool_pool_db_file_connpooldb_fileis_uriargskwargsc||_||_tj|g|dd|d||_d|j_y)NiF)timeoutcheck_same_threaduri)r r sqlite3connectrisolation_level)selfrrrrrs Q/var/www/openai/venv/lib/python3.12/site-packages/chromadb/db/impl/sqlite_pool.py__init__zConnection.__init__sO  __  IM !U QW  &* "sqlreturncx|dur|jj|S|jj||S)N.)rexecute)rr parameterss rr"zConnection.executes6  ::%%c* *zz!!#z22rNc8|jjyN)rcommitrs rr&zConnection.commit s rc8|jjyr%)rrollbackr's rr)zConnection.rollback#s rc6|jjSr%)rcursorr's rr+zConnection.cursor&szz  ""rc8|jjy)z(Actually closes the connection to the dbN)rcloser's r close_actualzConnection.close_actual)s r).r N)__name__ __module__ __qualname____doc____annotations__strrr boolrrCursorr"r&r)r+r.rrr r s~L MM   **%(*26*?B*NQ*3337>>3 ##rr cxeZdZdZedededdfdZededede fd Z ed d Z ed e ddfd Z y)r zCAbstract base class for a pool of connections to a sqlite database.rrr Ncyr%r8rrrs rrz Pool.__init__1 rrrcy)z"Return a connection from the pool.Nr8)rrrs rrz Pool.connect5 rcy)z"Close all connections in the pool.Nr8r's rr-z Pool.close:r>rconncy)z Return a connection to the pool.Nr8rr@s rreturn_to_poolzPool.return_to_pool?r>rr/) r0r1r2r3rr5r6rrr rr-rCr8rrr r .sM  T d   S C J     : $  rr ceZdZUdZeeejefe d<e je d<e je d<e e d<ee d<dde defd Zed ed ed efd Zeded dfdZeddZy)LockPoolaIA pool that has a single connection per thread but uses a lock to ensure that only one thread can use it at a time. This is used because sqlite does not support multithreaded access with connection timeouts when using the shared cache mode. We use the shared cache mode to allow multiple threads to share a database. _connections_lock _connectionr _is_urirrct|_tj|_tj |_||_||_yr%) setrF threadinglocalrHRLockrGr rIr;s rrzLockPool.__init__Qs7E$??,__&   rrrr c|jjt|jdr,|jj|jjSt ||j |jg|i|}||j_|jjtj||SNr@) rGacquirehasattrrHr@r r rIrFaddweakrefrefrrrnew_connections rrzLockPool.connectXs  4##V ,1A1A1F1F1R##(( ('dmmT\\48! !rr@NcX |jjy#t$rYywxYwr%)rGrelease RuntimeErrorrBs rrCzLockPool.return_to_pooles)  JJ      s  ))c|jD]}| |j!|jjtj|_ |j jy#t$rYywxYwr%) rFr.clearrLrMrHrGrYrZrBs rr-zLockPool.closelsr%%Dv!##%& !$??,  JJ      s#A>> B  B Fr/)r0r1r2r3rrrT ReferenceTyper r4rLrNrMr5r6rrrrrCr-r8rrrErEEs i 5 5z ABCC ?? M MT "S "C "J " ":$   rrEceZdZUdZeeejefe d<e je d<e je d<e e d<ee d<dde defd Zed ed ed efd ZeddZeded dfdZy) PerThreadPoolzMaintains a connection per thread. For now this does not maintain a cap on the number of connections, but it could be extended to do so and block on connect() if the cap is reached. rFrGrHr _is_uri_rrct|_tj|_tj |_||_||_yr%) rKrFrLrMrHLockrGr rIr;s rrzPerThreadPool.__init__s7E$??,^^%   rrrr ct|jdr,|jj|jjSt||j|j g|i|}||j_|j 5|jjtj|ddd|S#1swY|SxYwrP) rRrHr@r r rIrGrFrSrTrUrVs rrzPerThreadPool.connects 4##V ,1A1A1F1F1R##(( ('dmmT\\48>CNc|j5|jD]}| |j!|jjt j |_dddy#1swYyxYwr%)rGrFr.r\rLrMrHrBs rr-zPerThreadPool.closes\ ZZ))6%F'')*    # # %(0D  ZZsA8A A88Br@cyr%r8rBs rrCzPerThreadPool.return_to_poolr<rr]r/)r0r1r2r3rrrTr^r r4rLrcrMr5r6rrrrr-rCr8rrr`r`ysi 5 5z ABCC >> MNT "S "C "J " "11 : $  rr`)rrTabcrrtypingrrrL overridesrtyping_extensionsrr r rEr`r8rrrksJ#'!!H 3 .1t1h* D* r