ZiddlZddlZddlmZmZddlmZmZddlZddl m Z ddl m Z GddZ Gdd eZGd d eZGd d eZdS)N)ABCabstractmethod)AnySet)override) Annotatedc eZdZUdZded<eed<ejed<dddeded e d e f d Z dd edej fdZ ddZ ddZdej fdZddZdS) ConnectionzBA threadpool connection that returns itself to the pool on close()Pool_pool_db_file_connpooldb_fileis_uriargskwargscv||_||_tj|g|Rdd|d||_d|j_dS)NiF)timeoutcheck_same_threaduri)r r sqlite3connectrisolation_level)selfrrrrrs Y/root/projects/butler/venv/lib64/python3.11/site-packages/chromadb/db/impl/sqlite_pool.py__init__zConnection.__init__s`  _  IM  !U  QW   &* """.sqlreturnct|dur|j|S|j||S)N.)rexecute)rr parameterss rr"zConnection.executes;   :%%c** *z!!#z222rNc8|jdSN)rcommitrs rr&zConnection.commit s rc8|jdSr%)rrollbackr's rr)zConnection.rollback#s rc4|jSr%)rcursorr's rr+zConnection.cursor&sz  """rc8|jdS)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 sLL MMMMMM  **%(*26*?B*NQ****33337>3333 #####rr ceZdZdZedededdfdZededede fd Z ed d Z ed e ddfd Z dS)r zCAbstract base class for a pool of connections to a sqlite database.rrr NcdSr%r8rrrs rrz Pool.__init__1 rrrcdS)z"Return a connection from the pool.Nr8)rrrs rrz Pool.connect5  rcdS)z"Close all connections in the pool.Nr8r's rr-z Pool.close:r>rconncdS)z Return a connection to the pool.Nr8rr@s rreturn_to_poolzPool.return_to_pool?r>rr/) r0r1r2r3rr5r6rrr rr-rCr8rrr r .sMM  T d   ^  S C J   ^    ^  : $   ^   rr ceZdZUdZeeejefe d<e j e d<e j e d<e e d<ee d<dde d efd Zed ed ed efdZeded dfdZeddZdS)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_uriFrrct|_tj|_tj|_||_||_dSr%) setrF threadinglocalrHRLockrGr rIr;s rrzLockPool.__init__Qs?EE$?,,_&&   rrrr c:|jt|jdr|jj |jjSt ||j|jg|Ri|}||j_|j tj ||SNr@) rGacquirehasattrrHr@r r rIrFaddweakrefrefrrrnew_connections rrzLockPool.connectXs  4#V , , "1A1F1R#( ('dmT\48A;r@cdSr%r8rBs rrCzPerThreadPool.return_to_poolr<rr]r/)r0r1r2r3rrrTr^r r4rLrcrMr5r6rrrrr-rCr8rrr`r`ysi 5z ABCCCC >   MMMNNNT "S "C "J " " "X "111X1 : $   X   rr`)rrTabcrrtypingrrrL overridesrtyping_extensionsrr r rEr`r8rrrks)########''''''!!!!!!!!H     3   .11111t111h* * * * * D* * * * * r