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 `DROP CONSUMER GROUP` DDL handler.
//!
//! Ported from the pgwire `ddl::consumer_group::drop` handler. The token-based
//! syntax check, the direct `catalog.delete_consumer_group` path (NOT a
//! `propose_catalog_entry` proposal — this family writes the catalog directly),
//! the `group_registry.unregister`, the best-effort `offset_store.delete_group`
//! (warn-and-continue, preserved verbatim as the pre-existing behavior), and the
//! `audit_record` call are preserved verbatim; only the result construction
//! changed from pgwire `Response` / `PgWireError` to the protocol-neutral
//! [`DdlResult`] / [`DdlError`].
//!
//! Syntax: `DROP CONSUMER GROUP <name> ON <stream>`

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

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

/// Handle `DROP CONSUMER GROUP <name> ON <stream>`
pub fn drop_consumer_group(
    state: &SharedState,
    identity: &AuthenticatedIdentity,
    parts: &[&str],
) -> Result<Vec<DdlResult>, DdlError> {
    require_tenant_admin(identity, "drop consumer groups")?;

    // parts: ["DROP", "CONSUMER", "GROUP", "<name>", "ON", "<stream>"]
    if parts.len() < 6 || !parts[4].eq_ignore_ascii_case("ON") {
        return Err(DdlError {
            sqlstate: "42601".to_string(),
            message: "expected DROP CONSUMER GROUP <name> ON <stream>".to_string(),
        });
    }

    let group_name = parts[3].to_lowercase();
    let stream_name = parts[5].to_lowercase();
    let tenant_id = identity.tenant_id.as_u64();

    let catalog = state.credentials.catalog();

    let existed = catalog
        .delete_consumer_group(tenant_id, &stream_name, &group_name)
        .map_err(|e| DdlError {
            sqlstate: "XX000".to_string(),
            message: format!("catalog delete: {e}"),
        })?;

    if !existed {
        return Err(DdlError {
            sqlstate: "42704".to_string(),
            message: format!(
                "consumer group '{group_name}' does not exist on stream '{stream_name}'"
            ),
        });
    }

    state
        .group_registry
        .unregister(tenant_id, &stream_name, &group_name);

    // Delete committed offsets for this group.
    if let Err(e) = state
        .offset_store
        .delete_group(tenant_id, &stream_name, &group_name)
    {
        tracing::warn!(
            error = %e,
            "failed to delete offsets for consumer group {group_name}"
        );
    }

    state.audit_record(
        crate::control::security::audit::AuditEvent::AdminAction,
        Some(identity.tenant_id),
        &identity.username,
        &format!("DROP CONSUMER GROUP {group_name} ON {stream_name}"),
    );

    Ok(status("DROP CONSUMER GROUP"))
}