pgbatis 0.1.50

pgbaits 用于操作数据库数据的增删改查
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

/*
 * @Author: BuddyCoder
 * @Date: 2021-12-17 16:36:38
 * @LastEditTime: 2023-03-16 10:28:36
 * @LastEditors: BuddyCoder
 * @Description:
 * @FilePath: /pgbatis/src/plugin/logic_delete.rs
 * MIT
 */

use std::fmt::Debug;
pub use tokio_postgres::types::ToSql;

pub trait LogicDelete: Send + Sync + Debug {
    ///the name
    // fn name(&self) -> &str {
    //     std::any::type_name::<Self>()
    // }
    fn is_exclusion_table(&self, table_name: &str) -> bool;
    /// database column
    fn column(&self) -> &str;
    /// deleted data,must be i16
    fn deleted(&self) -> i16;
    /// un deleted data,must be i16
    fn un_deleted(&self) -> i16;
}

#[derive(Debug)]
pub struct LogicDeletePlugin {
    pub exclusion_table: Vec<String>,
    pub column: String,
    pub deleted: i16,
    pub un_deleted: i16,
}

impl LogicDeletePlugin {
    pub fn new(column: &str) -> Self {
        Self {
            exclusion_table: vec![],
            column: column.to_string(),
            deleted: 1,
            un_deleted: 0,
        }
    }
    pub fn new_opt(
        column: &str,
        deleted: i16,
        un_deleted: i16,
        exclusion_table: Vec<String>,
    ) -> Self {
        if deleted == un_deleted {
            panic!("[Dbaits] deleted can not equal to un_deleted on RbatisLogicDeletePlugin::new_opt(column: &str, deleted: i32, un_deleted: i32)")
        }
        Self {
            exclusion_table,
            column: column.to_string(),
            deleted,
            un_deleted,
        }
    }
}

impl LogicDelete for LogicDeletePlugin {
    fn is_exclusion_table(&self, table_name: &str) -> bool {
        for i in self.exclusion_table.iter() {
            if table_name.eq(i) {
                return true;
            }
        }
        return false;
    }

    fn column(&self) -> &str {
        self.column.as_str()
    }

    fn deleted(&self) -> i16 {
        self.deleted
    }

    fn un_deleted(&self) -> i16 {
        self.un_deleted
    }
}