nodedb 0.4.0

Local-first, real-time, edge-to-cloud hybrid database for multi-modal workloads
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

// SPDX-License-Identifier: BUSL-1.1

//! RAII in-flight guard for pgwire statement execution.
//!
//! Constructed at the top of a query handler and dropped when the statement
//! finishes (including early return, error, or panic-unwind). On construction
//! it bumps the connection's in-flight counter; on drop it decrements it and
//! stamps last-activity to "now". The listener watchdog reads that state to
//! decide idle eligibility, so this guard is what guarantees a long-running
//! statement is never idle-killed and that the idle window only starts once a
//! statement has actually completed.

use std::net::SocketAddr;

use crate::control::server::shared::session::SessionStore;

/// Marks a statement as in flight for the lifetime of the guard.
pub(crate) struct InFlightGuard<'a> {
    sessions: &'a SessionStore,
    addr: SocketAddr,
}

impl<'a> InFlightGuard<'a> {
    /// Begin a request: increment the connection's in-flight counter.
    pub(crate) fn new(sessions: &'a SessionStore, addr: SocketAddr) -> Self {
        sessions.begin_request(&addr);
        Self { sessions, addr }
    }
}

impl Drop for InFlightGuard<'_> {
    fn drop(&mut self) {
        // Decrement in-flight and stamp last-activity on every exit path.
        self.sessions.end_request(&self.addr);
    }
}