o 3Ih@sddlZddlZddlmZmZddlmZmZddlZddl m Z ddl m Z GdddZ Gdd d eZGd d d eZGd d d eZdS)N)ABCabstractmethod)AnySet)override) Annotatedc @seZdZUdZded<eed<ejed<dddeded e d e f d d Z ddedej fddZ dddZ dddZdej fddZdddZdS) ConnectionzBA threadpool connection that returns itself to the pool on close()Pool_pool_db_file_connpooldb_fileis_uriargskwargscOs:||_||_tj|g|Rdd|d||_d|j_dS)NiF)timeoutcheck_same_threaduri)r r sqlite3connectr isolation_level)selfr rrrrr\/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/chromadb/db/impl/sqlite_pool.py__init__s zConnection.__init__.sqlreturncCs"|dur |j|S|j||S)N.)r execute)rr parametersrrrrs zConnection.executeNcC|jdSN)r commitrrrrr" zConnection.commitcCr r!)r rollbackr#rrrr%#r$zConnection.rollbackcCs |jSr!)r cursorr#rrrr&&s zConnection.cursorcCs|jdS)z(Actually closes the connection to the dbN)r closer#rrr close_actual)szConnection.close_actual).rN)__name__ __module__ __qualname____doc____annotations__strrrboolrrCursorrr"r%r&r(rrrrr s*     rc@sheZdZdZedededdfddZeded ede fd d Z edd d Z ede ddfddZ dS)r zCAbstract base class for a pool of connections to a sqlite database.rrrNcCdSr!rrrrrrrr1z Pool.__init__rrcOdS)z"Return a connection from the pool.Nr)rrrrrrr5z Pool.connectcCr5)z"Close all connections in the pool.Nrr#rrrr':r6z Pool.closeconncCr5)z Return a connection to the pool.Nrrr7rrrreturn_to_pool?r6zPool.return_to_poolr)) r*r+r,r-rr/r0rrrrr'r9rrrrr .s r c@seZdZUdZeeejefe d<e j e d<e j e d<e e d<ee d<dde d efd d Zed ed edefddZededdfddZedddZdS)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_uriFrrcC,t|_t|_t|_||_||_dSr!) setr; threadinglocalr=RLockr<r r>r3rrrrQ    zLockPool.__init__rrrcOsf|jt|jdr|jjdur|jjSt||j|jg|Ri|}||j_|j t ||SNr7) r<acquirehasattrr=r7rr r>r;addweakrefrefrrrnew_connectionrrrrXs  zLockPool.connectr7NcCs&z|jWdStyYdSwr!)r<release RuntimeErrorr8rrrr9es  zLockPool.return_to_poolcCsZ|jD] }|dur|q|jt|_z|jWdSty,YdSwr!) r;r(clearrArBr=r<rMrNr8rrrr'ls      zLockPool.closeFr))r*r+r,r-rrrI ReferenceTyperr.rArCrBr/r0rrrrr9r'rrrrr:Es    r:c@seZdZUdZeeejefe d<e j e d<e j e d<e e d<ee d<dde d efd d Zed ed edefddZedddZededdfddZdS) 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. r;r<r=r _is_uri_FrrcCr?r!) r@r;rArBr=Lockr<r r>r3rrrrrDzPerThreadPool.__init__rrrcOst|jdr|jjdur|jjSt||j|jg|Ri|}||j_|j|jt |Wd|S1s;wY|SrE) rGr=r7rr r>r<r;rHrIrJrKrrrrs   zPerThreadPool.connectNcCs`|j#|jD] }|dur|q|jt|_WddS1s)wYdSr!)r<r;r(rOrArBr=r8rrrr's     "zPerThreadPool.closer7cCr2r!rr8rrrr9r4zPerThreadPool.return_to_poolrPr))r*r+r,r-rrrIrQrr.rArTrBr/r0rrrrr'r9rrrrrRys    rR)rrIabcrrtypingrrrA overridesrtyping_extensionsrrr r:rRrrrrs  $4