remdb 0.3.1

嵌入式内存数据库
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

//! Role module
//!
//! This module defines the Role struct for RBAC.

use std::string::String;
use std::vec::Vec;

use crate::rbac::permission::Permission;

/// Role struct for RBAC
#[derive(Debug, Clone)]
pub struct Role {
    /// Role name
    pub name: String,
    /// Permissions associated with the role
    pub permissions: Vec<(Permission, Option<String>, Option<String>)>, // (permission, table_name, column_name)
}

impl Role {
    /// Create a new role
    pub fn new(name: String) -> Self {
        Self {
            name,
            permissions: Vec::new(),
        }
    }

    /// Add a permission to the role
    pub fn add_permission(
        &mut self,
        permission: Permission,
        table_name: Option<String>,
        column_name: Option<String>,
    ) {
        self.permissions.push((permission, table_name, column_name));
    }

    /// Remove a permission from the role
    pub fn remove_permission(
        &mut self,
        permission: &Permission,
        table_name: &Option<String>,
        column_name: &Option<String>,
    ) {
        self.permissions
            .retain(|(p, t, c)| p != permission || t != table_name || c != column_name);
    }

    /// Check if the role has a specific permission
    pub fn has_permission(
        &self,
        permission: &Permission,
        table_name: &Option<String>,
        column_name: &Option<String>,
    ) -> bool {
        self.permissions.iter().any(|(p, t, c)| {
            p == permission
                && (t.is_none() || table_name.is_none() || t == table_name)
                && (c.is_none() || column_name.is_none() || c == column_name)
        })
    }
}