glaredb_rt_native 25.6.3

Native runtime and executors for GlareDB
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

use std::sync::Arc;

use glaredb_core::runtime::pipeline::QueryHandle;
use glaredb_core::runtime::profile_buffer::ProfileBuffer;
use parking_lot::Mutex;

use super::task::TaskState;

/// Query handle for queries being executed on the threaded runtime.
#[derive(Debug)]
pub struct ThreadedQueryHandle {
    /// Registered task states for all pipelines in a query.
    pub(crate) states: Mutex<Vec<Arc<TaskState>>>,
    pub(crate) profiles: ProfileBuffer,
}

impl QueryHandle for ThreadedQueryHandle {
    /// Cancel the query.
    fn cancel(&self) {
        let states = self.states.lock();

        for state in states.iter() {
            let mut sched_guard = state.sched_state.lock();
            sched_guard.canceled = true;
            std::mem::drop(sched_guard);

            // Re-execute the pipeline so it picks up the set bool. This lets us
            // cancel the pipeline regardless of if it's pending.
            Arc::clone(state).schedule();
        }
    }

    fn get_profile_buffer(&self) -> &ProfileBuffer {
        &self.profiles
    }
}