risingwave_meta_model/
user_privilege.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use risingwave_pb::user::grant_privilege::PbAction;
16use sea_orm::entity::prelude::*;
17use serde::{Deserialize, Serialize};
18
19use crate::{ObjectId, PrivilegeId, UserId};
20
21#[derive(
22    Clone, Copy, Debug, Hash, PartialEq, Eq, EnumIter, DeriveActiveEnum, Serialize, Deserialize,
23)]
24#[sea_orm(rs_type = "String", db_type = "string(None)")]
25pub enum Action {
26    #[sea_orm(string_value = "INSERT")]
27    Insert,
28    #[sea_orm(string_value = "SELECT")]
29    Select,
30    #[sea_orm(string_value = "UPDATE")]
31    Update,
32    #[sea_orm(string_value = "DELETE")]
33    Delete,
34    #[sea_orm(string_value = "USAGE")]
35    Usage,
36    #[sea_orm(string_value = "CREATE")]
37    Create,
38    #[sea_orm(string_value = "CONNECT")]
39    Connect,
40    #[sea_orm(string_value = "EXECUTE")]
41    Execute,
42}
43
44impl From<PbAction> for Action {
45    fn from(action: PbAction) -> Self {
46        match action {
47            PbAction::Unspecified => unreachable!("unspecified action"),
48            PbAction::Insert => Self::Insert,
49            PbAction::Select => Self::Select,
50            PbAction::Update => Self::Update,
51            PbAction::Delete => Self::Delete,
52            PbAction::Usage => Self::Usage,
53            PbAction::Create => Self::Create,
54            PbAction::Connect => Self::Connect,
55            PbAction::Execute => Self::Execute,
56        }
57    }
58}
59
60impl From<Action> for PbAction {
61    fn from(action: Action) -> Self {
62        match action {
63            Action::Insert => Self::Insert,
64            Action::Select => Self::Select,
65            Action::Update => Self::Update,
66            Action::Delete => Self::Delete,
67            Action::Usage => Self::Usage,
68            Action::Create => Self::Create,
69            Action::Connect => Self::Connect,
70            Action::Execute => Self::Execute,
71        }
72    }
73}
74
75#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
76#[sea_orm(table_name = "user_privilege")]
77pub struct Model {
78    #[sea_orm(primary_key)]
79    pub id: PrivilegeId,
80    pub dependent_id: Option<PrivilegeId>,
81    pub user_id: UserId,
82    pub oid: ObjectId,
83    pub granted_by: UserId,
84    pub action: Action,
85    pub with_grant_option: bool,
86}
87
88#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
89pub enum Relation {
90    #[sea_orm(
91        belongs_to = "super::object::Entity",
92        from = "Column::Oid",
93        to = "super::object::Column::Oid",
94        on_update = "NoAction",
95        on_delete = "NoAction"
96    )]
97    Object,
98    #[sea_orm(
99        belongs_to = "super::user::Entity",
100        from = "Column::GrantedBy",
101        to = "super::user::Column::UserId",
102        on_update = "NoAction",
103        on_delete = "NoAction"
104    )]
105    User2,
106    #[sea_orm(
107        belongs_to = "super::user::Entity",
108        from = "Column::UserId",
109        to = "super::user::Column::UserId",
110        on_update = "NoAction",
111        on_delete = "Cascade"
112    )]
113    User1,
114    #[sea_orm(
115        belongs_to = "Entity",
116        from = "Column::DependentId",
117        to = "Column::Id",
118        on_update = "NoAction",
119        on_delete = "Cascade"
120    )]
121    SelfRef,
122}
123
124impl Related<super::object::Entity> for Entity {
125    fn to() -> RelationDef {
126        Relation::Object.def()
127    }
128}
129
130impl ActiveModelBehavior for ActiveModel {}
risingwave_meta_model/user_privilege.rs

risingwave_meta_model/
user_privilege.rs
