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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

// SPDX-License-Identifier: BUSL-1.1

//! Protocol-neutral `ALTER SCHEDULE` DDL handler.
//!
//! Ported from the pgwire `ddl::schedule::alter` handler. The registry lookup,
//! the `propose_and_apply` catalog write, and the in-memory registry update are
//! preserved verbatim; only the result construction changed from pgwire
//! `Response` / `PgWireError` to the protocol-neutral [`DdlResult`] /
//! [`DdlError`].
//!
//! Supports: ENABLE, DISABLE, SET CRON 'expr'.

use crate::control::security::identity::AuthenticatedIdentity;
use crate::control::state::SharedState;
use crate::event::scheduler::cron::CronExpr;

use super::super::super::result::{DdlError, DdlResult};
use super::super::auth_support::status;

/// Handle `ALTER SCHEDULE <name> ENABLE | DISABLE | SET CRON '<expr>'`.
///
/// `name`, `action`, and `cron_expr` come from the typed
/// `AutomationStmt::AlterSchedule` variant.
pub fn alter_schedule(
    state: &SharedState,
    identity: &AuthenticatedIdentity,
    name: &str,
    action: &str,
    cron_expr: Option<&str>,
) -> Result<Vec<DdlResult>, DdlError> {
    let tenant_id = identity.tenant_id.as_u64();

    // Look up the schedule in the registry.
    let mut def = state
        .schedule_registry
        .get(tenant_id, name)
        .ok_or_else(|| DdlError {
            sqlstate: "42704".to_string(),
            message: format!("schedule \"{name}\" does not exist"),
        })?;

    match action {
        "ENABLE" => {
            def.enabled = true;
        }
        "DISABLE" => {
            def.enabled = false;
        }
        "SET" => {
            let new_cron = cron_expr.ok_or_else(|| DdlError {
                sqlstate: "42601".to_string(),
                message: "ALTER SCHEDULE SET CRON requires a quoted cron expression".to_string(),
            })?;

            CronExpr::parse(new_cron).map_err(|e| DdlError {
                sqlstate: "22023".to_string(),
                message: format!("invalid cron expression: {e}"),
            })?;

            def.cron_expr = new_cron.to_string();
        }
        _ => {
            return Err(DdlError {
                sqlstate: "42601".to_string(),
                message: "ALTER SCHEDULE supports: ENABLE, DISABLE, SET CRON 'expr'".to_string(),
            });
        }
    }

    // Persist the updated definition through the same metadata-raft
    // propose path every other parent-replicated ALTER uses, so a
    // cluster deployment converges on the new state cluster-wide and
    // the single-node fallback writes both the primary row and its
    // OWNERS row. The earlier direct `catalog.put_schedule(&def)`
    // call did neither — divergence on replicas, orphan on disk.
    let entry = crate::control::catalog_entry::CatalogEntry::PutSchedule(Box::new(def.clone()));
    super::super::super::catalog::propose_and_apply(state, &entry)?;

    // Update in-memory registry.
    state.schedule_registry.update(def);

    Ok(status("ALTER SCHEDULE"))
}