tatara-core 0.2.101

Shared domain types, config, and cluster types for tatara
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

use serde::{Deserialize, Serialize};
use tracing::info;

use super::types::NodeRoles;

/// Role configuration loaded from config or pinned explicitly.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RoleConfig {
    #[serde(default = "default_true")]
    pub voter: bool,
    #[serde(default = "default_true")]
    pub worker: bool,
    #[serde(default = "default_priority")]
    pub voter_priority: u32,
}

impl Default for RoleConfig {
    fn default() -> Self {
        Self {
            voter: true,
            worker: true,
            voter_priority: 10,
        }
    }
}

impl From<RoleConfig> for NodeRoles {
    fn from(cfg: RoleConfig) -> Self {
        NodeRoles {
            voter: cfg.voter,
            worker: cfg.worker,
            voter_priority: cfg.voter_priority,
        }
    }
}

/// Determines effective roles for this node.
///
/// If roles are explicitly pinned in config, use those.
/// Otherwise, default to both voter + worker (every node can do everything).
pub fn resolve_roles(config: &RoleConfig) -> NodeRoles {
    let roles = NodeRoles::from(config.clone());

    info!(
        voter = roles.voter,
        worker = roles.worker,
        priority = roles.voter_priority,
        "Node roles resolved"
    );

    roles
}

fn default_true() -> bool {
    true
}

fn default_priority() -> u32 {
    10
}