nodedb 0.2.1

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
84
85
86
87
88
89
90
91
92
93
94

// SPDX-License-Identifier: BUSL-1.1

//! `CREATE CONSUMER GROUP` DDL handler.
//!
//! Syntax: `CREATE CONSUMER GROUP <name> ON <stream>`

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

use crate::control::security::identity::AuthenticatedIdentity;
use crate::control::state::SharedState;
use crate::event::cdc::consumer_group::ConsumerGroupDef;

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

/// Handle `CREATE CONSUMER GROUP <name> ON <stream>`.
///
/// `group_name` and `stream_name` come from the typed
/// [`NodedbStatement::CreateConsumerGroup`] variant.
pub fn create_consumer_group(
    state: &SharedState,
    identity: &AuthenticatedIdentity,
    group_name: &str,
    stream_name: &str,
) -> PgWireResult<Vec<Response>> {
    require_tenant_admin(identity, "create consumer groups")?;

    let group_name = group_name.to_lowercase();
    let stream_name = stream_name.to_lowercase();
    let tenant_id = identity.tenant_id.as_u64();

    // Verify the stream or topic exists.
    // Consumer groups can be created on change streams or durable topics.
    let is_stream = state.stream_registry.get(tenant_id, &stream_name).is_some();
    let is_topic = state
        .ep_topic_registry
        .get(tenant_id, &stream_name)
        .is_some();
    // Topics use "topic:<name>" as buffer key — check with prefix too.
    let topic_bare = stream_name.strip_prefix("topic:").unwrap_or(&stream_name);
    let is_topic = is_topic || state.ep_topic_registry.get(tenant_id, topic_bare).is_some();
    if !is_stream && !is_topic {
        return Err(sqlstate_error(
            "42704",
            &format!("change stream or topic '{stream_name}' does not exist"),
        ));
    }

    // Check for duplicate group.
    if state
        .group_registry
        .get(tenant_id, &stream_name, &group_name)
        .is_some()
    {
        return Err(sqlstate_error(
            "42710",
            &format!("consumer group '{group_name}' already exists on stream '{stream_name}'"),
        ));
    }

    let now = std::time::SystemTime::now()
        .duration_since(std::time::UNIX_EPOCH)
        .map_err(|_| sqlstate_error("XX000", "system clock error"))?
        .as_secs();

    let def = ConsumerGroupDef {
        tenant_id,
        name: group_name.clone(),
        stream_name: stream_name.clone(),
        owner: identity.username.clone(),
        created_at: now,
    };

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

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

    state.group_registry.register(def);

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

    Ok(vec![Response::Execution(Tag::new("CREATE CONSUMER GROUP"))])
}