nodedb 0.2.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

// SPDX-License-Identifier: BUSL-1.1

//! `ALTER ALERT` DDL handler.
//!
//! Syntax:
//! ```sql
//! ALTER ALERT <name> ENABLE
//! ALTER ALERT <name> DISABLE
//! ```

use pgwire::api::results::{Response, Tag};
use pgwire::error::PgWireResult;

use crate::control::security::identity::AuthenticatedIdentity;
use crate::control::state::SharedState;

use super::super::super::types::{require_tenant_admin, sqlstate_error};

pub fn alter_alert(
    state: &SharedState,
    identity: &AuthenticatedIdentity,
    name: &str,
    action: &str,
) -> PgWireResult<Vec<Response>> {
    require_tenant_admin(identity, "alter alerts")?;

    let tenant_id = identity.tenant_id.as_u64();

    let mut def = state
        .alert_registry
        .get(tenant_id, name)
        .ok_or_else(|| sqlstate_error("42704", &format!("alert '{name}' does not exist")))?;

    match action {
        "ENABLE" => def.enabled = true,
        "DISABLE" => def.enabled = false,
        _ => return Err(sqlstate_error("42601", "expected ENABLE or DISABLE")),
    }

    let catalog = state
        .credentials
        .catalog()
        .as_ref()
        .ok_or_else(|| sqlstate_error("XX000", "system catalog not available"))?;

    catalog
        .put_alert_rule(&def)
        .map_err(|e| sqlstate_error("XX000", &format!("catalog write: {e}")))?;

    state.alert_registry.update(def);

    state.audit_record(
        crate::control::security::audit::AuditEvent::AdminAction,
        Some(identity.tenant_id),
        &identity.username,
        &format!("ALTER ALERT {name}"),
    );

    Ok(vec![Response::Execution(Tag::new("ALTER ALERT"))])
}