Σi$jdZddlZddlZddlZddlmZej ddZeej ddZ ej dd Z ej d d Z ej d d Z ej d dZ eej ddZej ddZej ddZeej ddZej ddZej ddj'dk(ZdZdZdZGddZeZy) u AIVA RLM Nexus — Centralized Connection Factory Story 1.05 — Track A Single source of truth for PostgreSQL, Redis, and Qdrant connections. All connections point to Elestio infrastructure. Includes health check and retry. RULE COMPLIANCE: - Rule 7: No SQLite — all storage uses Elestio Postgres/Redis/Qdrant - Rule 6: E: drive only — module lives at /mnt/e/genesis-system/core/db/ N)OptionalPG_HOSTz(postgresql-genesis-u50607.vm.elestio.appPG_PORT25432PG_USERpostgresPG_PASSzCiBjh6LM7Yuqkq-jo2r7eQDwPG_DB REDIS_HOSTz#redis-genesis-u50607.vm.elestio.app REDIS_PORT26379 REDIS_PASSze2ZyYYr4oWRdASI2CaLc- QDRANT_HOSTz"qdrant-b3knu-u50607.vm.elestio.app QDRANT_PORT6333QDRANT_API_KEY@7b74e6621bd0e6650789f6662bca4cbf4143d3d1d710a0002b3b563973ca6876 QDRANT_HTTPStrueg?cBtjfd}|S)zKRetry a connection function up to MAX_RETRIES times with RETRY_DELAY sleep.cd}tdtdzD] } |i|cSt jdtd||#t$r.}|}|tkrtjt Yd}~cd}~wwxYw)Nz failed after z retries: )range MAX_RETRIES Exceptiontimesleep RETRY_DELAYConnectionError__name__)argskwargs last_errorattemptexcfuncs ,/mnt/e/genesis-system/core/db/connections.pywrapperz_retry..wrapper3s*. Q a0 ,G ,T,V,, ,}}o^K= :, O    , [(JJ{+ ,sA A?$A::A?) functoolswraps)r'r)s` r(_retryr,1s%__T   Nc\eZdZdZd dZedZedZedZde fdZ d d Z y) ConnectionFactoryz Single source of truth for all Elestio database connections. Lazy-initialises each connection on first access and caches it. Call close_all() to release connections when done (e.g. in teardown). returnNc.d|_d|_d|_y)N)_pg_conn _redis_client_qdrant_clientselfs r(__init__zConnectionFactory.__init__Os !"r-cddl}|j|jjrA|jtt t ttd|_d|j_ |jS)z Return a live psycopg2 connection to Elestio PostgreSQL. Re-opens if the cached connection has been closed. Retries up to MAX_RETRIES times on failure. rN hostportuserpassworddatabaseconnect_timeoutF) psycopg2r2closedconnectrrrr r autocommit)r6rAs r( get_postgreszConnectionFactory.get_postgresXs^  == DMM$8$8$,,  " -DM(-DMM $}}r-cddl}|j(|jttt ddd|_|jj |jS)z Return a connected redis.Redis client pointed at Elestio Redis. PINGs the server to confirm liveness. Retries up to MAX_RETRIES times on failure. rNTr9)r;r<r>decode_responsessocket_timeoutsocket_connect_timeout)redisr3Redisr r rping)r6 redis_libs r( get_rediszConnectionFactory.get_redisrs[ "    %!*#!%!') "1"D  !!!!r-cddlm}|j/ttdt d}t r t |d<|di||_|jS)zx Return a QdrantClient connected to Elestio Qdrant. Retries up to MAX_RETRIES times on failure. r QdrantClientr9r;r<timeouthttpsapi_key) qdrant_clientrQr4rrrr)r6rQr#s r( get_qdrantzConnectionFactory.get_qdrantsQ /    &##% F $2y!"."8"8D """r-cdddd} ddl}|jtttt t d}|j}|jd|jddk(sJ|j|jd |d < ddl }|jttt dd }|j#|jd |d < dd lm}t(t*dt,d}t.r t.|d<|di|}|j1d |d<|S#t$rYwxYw#t$rYbwxYw#t$rY|SwxYw)a+ Probe each service with a lightweight command. Returns: {"postgres": bool, "redis": bool, "qdrant": bool} A False value means the service is unreachable or returned an error; it does NOT raise an exception so callers can act on partial health. F)rrJqdrantrNr:zSELECT 1rTr)r;r<r>rHrIrJrPrRrUrZrV)rArCrrrr r cursorexecutefetchonecloserrJrKr r rrLrWrQrrrrget_collections)r6resultrAprobecurrMrQr#s r( health_checkzConnectionFactory.health_checksg$euE  $$7w ! %E ,,.C KK #<<>!$) )) IIK KKM!%F:   %OOj# '( $E JJL KKM"F7O   2$#% F $2y! *6*E  ! ! ##F8  M       &    s8B D)A D8"AE) D54D58 EE EEc2|j1|jjs |jj|j |jjd|_d|_d|_y#t$rYHwxYw#t$rY0wxYw)z)Close and release all cached connections.N)r2rBr_rr3r4r5s r( close_allzConnectionFactory.close_alls == $T]]-A-A  ##%    ) ""((* " !"     s#A; B ; BB BB)r0N) r! __module__ __qualname____doc__r7r,rErNrXdictrdrfrVr-r(r/r/Gs\#  2 " "4 # #2BdBP#r-r/)riosrr*typingrgetenvrintrrr r r r rrrrlowerrrrr,r/ connectionsrVr-r(rqsB   "))II J ibii 7+ , "))Iz * "))I9 : "))Gz *RYY|%J K <1 2 RYY|%< = =-QR YRYY}f56 +-op>V4::<F   ,y#y#F ! r-