Li{hdZddlmZddlZddlZddlmZddlmZddlZddl Zddl ZGddZ y)u Session Store — L4 Cold Ledger session lifecycle manager. Provides a typed, connection-pooled interface for creating, tracking, and closing agent sessions in the Genesis sessions table. sessions — One row per agent session (start/end timestamps, agent ID, arbitrary metadata as JSONB). Usage:: from core.storage.session_store import SessionStore store = SessionStore(connection_params={ "host": "...", "port": 5432, "user": "...", "password": "...", "dbname": "genesis" }) session_id = store.open_session("forge-agent", {"task": "build_widget"}) sessions = store.get_active_sessions() session = store.get_session(session_id) store.close_session(session_id) cleaned = store.cleanup_orphaned_sessions() store.close() Rules enforced (Genesis hardwired): - NO SQLite anywhere in this file - All SQL uses parameterised queries (%s placeholders — never f-strings) - Connection pool uses getconn/putconn in try/finally (no connection leaks) ) annotationsN)datetime)OptionalcXeZdZdZd dZdZd dZdddZddZddZ dd Z dd Z d d Z y) SessionStoreuTyped Postgres client for the Genesis sessions table (L4 Cold Ledger). Thread-safe via ThreadedConnectionPool. Every method acquires a connection from the pool and returns it in a ``try/finally`` block — connection leaks are structurally impossible. Args: connection_params: Dict accepted by ``psycopg2.connect`` — keys: ``host``, ``port``, ``user``, ``password``, ``dbname``. Optionally ``sslmode`` and other libpq parameters. cZ||_tjjdi||_y)N) )_paramspsycopg2poolThreadedConnectionPool)selfconnection_paramss 3/mnt/e/genesis-system/core/storage/session_store.py__init__zSessionStore.__init__9s&( MM 0 0 L:K L c6|jjS)zCReturn a connection from the pool (caller must putconn in finally).)r getconnrs r_acquirezSessionStore._acquireCsyy  ""rc:|jj|y)z3Return a connection to the pool without closing it.N)r putconn)rconns r_releasezSessionStore._releaseGs $rNczttj}d}|tj|nd}|j } |j 5}|j||||fddd|j|j||S#1swY,xYw#|j|wxYw)aCreate a new session row with started_at=NOW() and ended_at=NULL. Args: agent_id: Identifier for the agent opening the session, e.g. ``"forge-agent"`` or ``"parallel-builder"``. metadata: Optional dict of arbitrary key/value data stored as JSONB in the sessions table. Returns: The new session's UUID4 string. zZINSERT INTO sessions (id, started_at, agent_id, metadata) VALUES (%s::uuid, NOW(), %s, %s)N) struuiduuid4jsondumpsrcursorexecutecommitr)ragent_idmetadata session_idsql metadata_jsonrcurs r open_sessionzSessionStore.open_sessionOs&  0 190D 8,$ }}  H# C*h !FG H KKM MM$   H H MM$ s$ B'B0B'B$ B''B:cd}|j} |j5}|j||fddd|j|j |y#1swY+xYw#|j |wxYw)zMark a session as ended by setting ended_at = NOW(). Args: session_id: UUID string of the session to close. z8UPDATE sessions SET ended_at = NOW() WHERE id = %s::uuidN)rr"r#r$r)rr'r(rr*s r close_sessionzSessionStore.close_sessionjsr # }}  0# C*/ 0 KKM MM$   0 0 MM$ s!A.A"A."A+'A..Bcrd}|j} |jtjj5}|j ||j }ddd|j|Dcgc] }t|c}S#1swY3xYw#|j|wxYwcc}w)zReturn all sessions where ended_at IS NULL (currently open). Returns: List of dicts with keys: ``id``, ``started_at``, ``ended_at``, ``agent_id``, ``metadata``. May be empty. zpSELECT id, started_at, ended_at, agent_id, metadata FROM sessions WHERE ended_at IS NULL ORDER BY started_at ASCcursor_factoryN) rr"r extrasRealDictCursorr#fetchallrdict)rr(rr*rowsrows rget_active_sessionsz SessionStore.get_active_sessions|s ' }} HOO,J,JK &s C ||~ & MM$ %)*cS **  & & MM$ *s(*B"B B=B4BBB1cVd}|j} |jtjj5}|j ||f|j }ddd|j|yt|S#1swY(xYw#|j|wxYw)aFetch a single session by its primary key. Args: session_id: UUID string of the session to retrieve. Returns: A dict with keys ``id``, ``started_at``, ``ended_at``, ``agent_id``, ``metadata``, or ``None`` if not found. zUSELECT id, started_at, ended_at, agent_id, metadata FROM sessions WHERE id = %s::uuidr/N) rr"r r1r2r#fetchonerr4)rr'r(rr*r6s r get_sessionzSessionStore.get_sessions # }} HOO,J,JK %s C*/lln % MM$  ;Cy % % MM$ s"*B$B "B BBB(cd}|j} |j5}|j||j}ddd|j |j |S#1swY,xYw#|j |wxYw)a?Close sessions older than 24 hours that were never explicitly closed. Sets ended_at = NOW() for any session where: - ended_at IS NULL (not yet closed), AND - started_at < NOW() - INTERVAL '24 hours' (stale / orphaned) Returns: The number of sessions updated. zhUPDATE sessions SET ended_at = NOW() WHERE ended_at IS NULL AND started_at < NOW() - INTERVAL '24 hours'N)rr"r#rowcountr$r)rr(rr* updated_counts rcleanup_orphaned_sessionsz&SessionStore.cleanup_orphaned_sessionss < }}  -# C #  - KKM MM$   - - MM$ s"A9A-A9-A62A99B cR|jr|jjyy)zBShut down the connection pool, closing all underlying connections.N)r closeallrs rclosezSessionStore.closes 99 II    r)rr4returnNone)rBrC)N)r%rr&r4rBr)r'rrBrC)rBlist)r'rrBzOptional[dict])rBint) __name__ __module__ __qualname____doc__rrrr+r-r7r:r>rArrrr,s4  # 6 $+,4:!rr) rI __future__rr rrtypingrr psycopg2.extras psycopg2.poolrrJrrrOs0># `!`!r