nodedb-sql 0.3.0-beta.1

SQL parser, planner, and optimizer for NodeDB
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

// SPDX-License-Identifier: Apache-2.0

//! Parse cluster-admin: SHOW CLUSTER/NODES/RAFT/MIGRATIONS/RANGES/ROUTING,
//! REMOVE NODE, REBALANCE, ALTER RAFT GROUP.

use crate::ddl_ast::statement::{ClusterStmt, NodedbStatement};
use crate::error::SqlError;

pub(super) fn try_parse(
    upper: &str,
    parts: &[&str],
    _trimmed: &str,
) -> Option<Result<NodedbStatement, SqlError>> {
    (|| -> Option<NodedbStatement> {
        if upper.starts_with("SHOW CLUSTER") {
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowCluster));
        }
        if upper.starts_with("SHOW MIGRATIONS") {
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowMigrations));
        }
        if upper.starts_with("SHOW RANGES") {
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowRanges));
        }
        if upper.starts_with("SHOW ROUTING") {
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowRouting));
        }
        if upper.starts_with("SHOW SCHEMA VERSION") {
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowSchemaVersion));
        }
        if upper.starts_with("SHOW PEER HEALTH") {
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowPeerHealth));
        }
        if upper.starts_with("REBALANCE") {
            return Some(NodedbStatement::Cluster(ClusterStmt::Rebalance));
        }
        if upper.starts_with("SHOW RAFT GROUP ") {
            let id = parts.get(3)?.to_string();
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowRaftGroup {
                group_id: id,
            }));
        }
        if upper.starts_with("SHOW RAFT GROUPS") || upper.starts_with("SHOW RAFT") {
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowRaftGroups));
        }
        if upper.starts_with("ALTER RAFT GROUP ") {
            // ALTER RAFT GROUP <group_id> ADD|REMOVE NODE <node_id>
            let group_id = parts.get(3).map(|s| s.to_string()).unwrap_or_default();
            let action = parts.get(4).map(|s| s.to_uppercase()).unwrap_or_default();
            let node_id = parts.get(6).map(|s| s.to_string()).unwrap_or_default();
            return Some(NodedbStatement::Cluster(ClusterStmt::AlterRaftGroup {
                group_id,
                action,
                node_id,
            }));
        }
        if upper.starts_with("REMOVE NODE ") {
            let id = parts.get(2)?.to_string();
            return Some(NodedbStatement::Cluster(ClusterStmt::RemoveNode {
                node_id: id,
            }));
        }
        if upper.starts_with("SHOW NODE ") {
            let id = parts.get(2)?.to_string();
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowNode {
                node_id: id,
            }));
        }
        if upper.starts_with("SHOW NODES") {
            return Some(NodedbStatement::Cluster(ClusterStmt::ShowNodes));
        }
        None
    })()
    .map(Ok)
}