uiddlZddlmZmZddlmZmZddlZddlm Z GddZ GddeZ Gd d e Z Gd d e Z dS) N)ABCabstractmethod)AnySet)overridec 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 sqlite3connectr isolation_level)selfrrrrrs O/var/www/icac/venv/lib/python3.11/site-packages/chromadb/db/impl/sqlite_pool.py__init__zConnection.__init__s`  _  IM  !U  QW   &* """.sqlreturnct|dur|j|S|j||S)N.)r execute)rr parameterss rr!zConnection.executes;   :%%c** *z!!#z222rNc8|jdSN)r commitrs rr%zConnection.commits rc8|jdSr$)r rollbackr&s rr(zConnection.rollback!s rc4|jSr$)r cursorr&s rr*zConnection.cursor$sz  """rc8|jdS)z(Actually closes the connection to the dbN)r closer&s r close_actualzConnection.close_actual's r).rN)__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.rrrNcdSr$r7rrrs rrz Pool.__init__/ rrrcdS)z"Return a connection from the pool.Nr7)rrrs rrz Pool.connect3  rcdS)z"Close all connections in the pool.Nr7r&s rr,z Pool.close8r=rconncdS)z Return a connection to the pool.Nr7rr?s rreturn_to_poolzPool.return_to_pool=r=rr.) r/r0r1r2rr4r5rrr rr,rBr7rrr r ,sMM  T d   ^  S C J   ^    ^  : $   ^   rr ceZdZUdZeeed<ejed<ej ed<e ed<e ed<dde d e fd Z e d ed ed efdZe ded dfdZe ddZdS)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$) setrE threadinglocalrGRLockrFr rHr:s rrzLockPool.__init__Os?EE$?,,_&&   rrrrc|jt|jdr|jj |jjSt ||j|jg|Ri|}||j_|j ||SNr?) rFacquirehasattrrGr?r r rHrEaddrrrnew_connections rrzLockPool.connectVs  4#V , , "1A1F1R#( ('dmT\48   MMMNNNT "S "C "J " " "X "111X1 : $   X   rr\)rabcrrtypingrrrK overridesrr r rDr\r7rrrfs########!!!!!!!!H     3   .00000t000f) ) ) ) ) D) ) ) ) ) r