chimes_rust/entity/

chimes_permission_info.rs

use rbatis::crud::{Skip, CRUD};
use rbatis::crud_table;
use rbatis::error::Error;
use rbatis::rbatis::Rbatis;
use rbatis::Page;
use rbatis::PageRequest;
use rbson::Bson;
use serde_derive::{Deserialize, Serialize};
/**
 * Generate the file for chimes_permission_info.rs,
 */
use std::fmt::Debug;

#[crud_table(table_name:"chimes_permission"|table_columns:"id,alias,create_time,name,pid,api_pattern,service_id,api_method,api_bypass")]
#[derive(Debug, Clone, Default, Deserialize, Serialize)]
pub struct ChimesPermissionInfo {
    pub id: Option<i64>,
    pub alias: Option<String>,
    pub create_time: Option<rbatis::DateTimeNative>,
    pub name: Option<String>,
    pub pid: Option<i64>,
    pub api_pattern: Option<String>,
    pub service_id: Option<String>,
    pub api_method: Option<String>,
    pub api_bypass: Option<String>,
}

impl ChimesPermissionInfo {
    #[allow(dead_code)]
    pub async fn from_id(rb: &Rbatis, id: &i64) -> Result<Option<Self>, Error> {
        let wp = rb.new_wrapper().eq("id", id);
        rb.fetch_by_wrapper::<Option<Self>>(wp).await
    }

    #[allow(dead_code)]
    pub async fn save(&mut self, rb: &Rbatis) -> Result<u64, Error> {
        match rb.save(self, &[Skip::Column("id")]).await {
            Ok(ds) => {
                self.id = ds.last_insert_id;
                Ok(ds.rows_affected)
            }
            Err(err) => Err(err),
        }
    }

    #[allow(dead_code)]
    pub async fn update(&self, rb: &Rbatis) -> Result<u64, Error> {
        let wp = rb.new_wrapper().eq("id", self.id);
        rb.update_by_wrapper(self, wp, &[Skip::Column("id")]).await
    }

    #[allow(dead_code)]
    pub async fn update_selective(&self, rb: &Rbatis) -> Result<u64, Error> {
        let wp = rb.new_wrapper().eq("id", self.id);
        rb.update_by_wrapper(self, wp, &[Skip::Value(Bson::Null)])
            .await
    }

    #[allow(dead_code)]
    pub async fn remove_batch(&self, rb: &Rbatis) -> Result<u64, Error> {
        let wp = rb
            .new_wrapper()
            .r#if(self.id.clone().is_some(), |w| {
                w.and().eq("id", self.id.unwrap())
            })
            .r#if(self.alias.clone().is_some(), |w| {
                w.and().eq("alias", self.alias.clone().unwrap())
            })
            .r#if(self.create_time.clone().is_some(), |w| {
                w.and().eq("create_time", self.create_time.unwrap())
            })
            .r#if(self.name.clone().is_some(), |w| {
                w.and().eq("name", self.name.clone().unwrap())
            })
            .r#if(self.pid.clone().is_some(), |w| {
                w.and().eq("pid", self.pid.unwrap())
            })
            .r#if(self.api_pattern.clone().is_some(), |w| {
                w.and().eq("api_pattern", self.api_pattern.clone().unwrap())
            })
            .r#if(self.service_id.clone().is_some(), |w| {
                w.and().eq("service_id", self.service_id.clone().unwrap())
            })
            .r#if(self.api_method.clone().is_some(), |w| {
                w.and().eq("api_method", self.api_method.clone().unwrap())
            })
            .r#if(self.api_bypass.clone().is_some(), |w| {
                w.and().eq("api_bypass", self.api_bypass.clone().unwrap())
            });
        rb.remove_by_wrapper::<Self>(wp).await
    }

    #[allow(dead_code)]
    pub async fn remove(&mut self, rb: &Rbatis) -> Result<u64, Error> {
        let wp = rb.new_wrapper().eq("id", self.id);
        rb.remove_by_wrapper::<Self>(wp).await
    }

    #[allow(dead_code)]
    pub async fn remove_ids(rb: &Rbatis, ids: &[i64]) -> Result<u64, Error> {
        let wp = rb.new_wrapper().r#in("id", ids);
        rb.remove_by_wrapper::<Self>(wp).await
    }

    #[allow(dead_code)]
    pub async fn query_paged(&self, rb: &Rbatis, curr: u64, ps: u64) -> Result<Page<Self>, Error> {
        let wp = rb
            .new_wrapper()
            .r#if(self.id.clone().is_some(), |w| {
                w.and().eq("id", self.id.unwrap())
            })
            .r#if(self.alias.clone().is_some(), |w| {
                w.and().eq("alias", self.alias.clone().unwrap())
            })
            .r#if(self.create_time.clone().is_some(), |w| {
                w.and().eq("create_time", self.create_time.unwrap())
            })
            .r#if(self.name.clone().is_some(), |w| {
                w.and().eq("name", self.name.clone().unwrap())
            })
            .r#if(self.pid.clone().is_some(), |w| {
                w.and().eq("pid", self.pid.unwrap())
            })
            .r#if(self.pid.clone().is_none(), |w| w.and().eq("pid", Some(0)))
            .r#if(self.api_pattern.clone().is_some(), |w| {
                w.and().eq("api_pattern", self.api_pattern.clone().unwrap())
            })
            .r#if(self.service_id.clone().is_some(), |w| {
                w.and().eq("service_id", self.service_id.clone().unwrap())
            })
            .r#if(self.api_method.clone().is_some(), |w| {
                w.and().eq("api_method", self.api_method.clone().unwrap())
            })
            .r#if(self.api_bypass.clone().is_some(), |w| {
                w.and().eq("api_bypass", self.api_bypass.clone().unwrap())
            });
        rb.fetch_page_by_wrapper::<Self>(wp, &PageRequest::new(curr, ps))
            .await
    }

    #[allow(dead_code)]
    pub async fn query_list(&self, rb: &Rbatis) -> Result<Vec<Self>, Error> {
        let wp = rb
            .new_wrapper()
            .r#if(self.id.clone().is_some(), |w| {
                w.and().eq("id", self.id.unwrap())
            })
            .r#if(self.alias.clone().is_some(), |w| {
                w.and().eq("alias", self.alias.clone().unwrap())
            })
            .r#if(self.create_time.clone().is_some(), |w| {
                w.and().eq("create_time", self.create_time.unwrap())
            })
            .r#if(self.name.clone().is_some(), |w| {
                w.and().eq("name", self.name.clone().unwrap())
            })
            .r#if(self.pid.clone().is_some(), |w| {
                w.and().eq("pid", self.pid.unwrap())
            })
            .r#if(self.pid.clone().is_none(), |w| w.and().eq("pid", Some(0)))
            .r#if(self.api_pattern.clone().is_some(), |w| {
                w.and().eq("api_pattern", self.api_pattern.clone().unwrap())
            })
            .r#if(self.service_id.clone().is_some(), |w| {
                w.and().eq("service_id", self.service_id.clone().unwrap())
            })
            .r#if(self.api_method.clone().is_some(), |w| {
                w.and().eq("api_method", self.api_method.clone().unwrap())
            })
            .r#if(self.api_bypass.clone().is_some(), |w| {
                w.and().eq("api_bypass", self.api_bypass.clone().unwrap())
            });
        rb.fetch_list_by_wrapper::<Self>(wp).await
    }

    #[allow(dead_code)]
    pub async fn query_all(rb: &Rbatis) -> Result<Vec<Self>, Error> {
        let wp = rb.new_wrapper();
        rb.fetch_list_by_wrapper::<Self>(wp).await
    }

    #[allow(dead_code)]
    pub async fn query_tree(rb: &Rbatis, pid: &Option<i64>) -> Result<Vec<Self>, Error> {
        let wp = rb
            .new_wrapper()
            .r#if(pid.clone().is_some(), |w| w.and().eq("pid", pid.unwrap()))
            .r#if(pid.clone().is_none(), |w| w.and().eq("pid", 0));
        rb.fetch_list_by_wrapper::<Self>(wp).await
    }

    fn find_children(list: &Vec<Self>, pid: &i64) -> Vec<Self> {
        let mut found = vec![];
        for it in list.clone() {
            if it.pid == Some(*pid) {
                found.push(it.clone());
                let nid = it.id.unwrap_or_default();
                let mut mt = Self::find_children(list, &nid);
                if !mt.is_empty() {
                    found.append(&mut mt);
                }
            }
        }
        found
    }

    pub async fn query_children(rb: &Rbatis, pid: &i64) -> Result<Vec<Self>, Error> {
        match Self::query_all(rb).await {
            Ok(rs) => {
                let mut found = Self::find_children(&rs, pid);
                for it in rs.clone() {
                    if it.id == Some(*pid) {
                        found.insert(0, it);
                    }
                }
                Ok(found)
            }
            Err(err) => Err(err),
        }
    }
}

#[derive(Debug, Clone, Default, Deserialize, Serialize)]
pub struct ChimesPermissionInfoValue {
    pub id: Option<i64>,
    pub alias: Option<String>,
    pub create_time: Option<rbatis::DateTimeNative>,
    pub name: Option<String>,
    pub pid: Option<i64>,
    pub api_pattern: Option<String>,
    pub service_id: Option<String>,
    pub api_method: Option<String>,
    pub api_bypass: Option<String>,
    pub leaf: bool,
    pub label: Option<String>,
    pub has_children: bool,
    #[serde(default)]
    pub children: Vec<ChimesPermissionInfoValue>,
}

impl ChimesPermissionInfoValue {
    #[allow(dead_code)]
    pub fn from_entity(param: &ChimesPermissionInfo) -> Self {
        Self {
            id: param.id,
            alias: param.alias.clone(),
            create_time: param.create_time,
            name: param.name.clone(),
            pid: param.pid,
            api_pattern: param.api_pattern.clone(),
            service_id: param.service_id.clone(),
            api_method: param.api_method.clone(),
            api_bypass: param.api_bypass.clone(),
            has_children: false,
            leaf: false,
            children: vec![],
            label: param.name.clone(),
        }
    }

    #[allow(dead_code)]
    pub fn from_entity_with(
        param: &ChimesPermissionInfo,
        haschild: bool,
        children: &[Self],
    ) -> Self {
        Self {
            id: param.id,
            alias: param.alias.clone(),
            create_time: param.create_time,
            name: param.name.clone(),
            pid: param.pid,
            api_pattern: param.api_pattern.clone(),
            service_id: param.service_id.clone(),
            api_method: param.api_method.clone(),
            api_bypass: param.api_bypass.clone(),
            has_children: haschild,
            leaf: !haschild,
            children: children.to_vec(),
            label: param.name.clone(),
        }
    }

    #[allow(dead_code)]
    pub fn to_entity(&self) -> ChimesPermissionInfo {
        ChimesPermissionInfo {
            id: self.id,
            alias: self.alias.clone(),
            create_time: self.create_time,
            name: self.name.clone(),
            pid: self.pid,
            api_pattern: self.api_pattern.clone(),
            service_id: self.service_id.clone(),
            api_method: self.api_method.clone(),
            api_bypass: self.api_bypass.clone(),
        }
    }

    fn recurse_build_tree(items: &Vec<Self>, parent_item: &mut Self) {
        for xip in items.clone() {
            if xip.pid == parent_item.id {
                let mut mip = xip;
                Self::recurse_build_tree(items, &mut mip);
                if mip.children.is_empty() {
                    mip.leaf = true;
                    mip.has_children = false;
                }
                parent_item.children.push(mip);
            }
        }
    }

    #[allow(dead_code)]
    pub fn build_tree(items: &Vec<Self>) -> Vec<Self> {
        let mut tmptree = vec![];
        for xip in items.clone() {
            if xip.pid.is_none() || xip.pid == Some(0) {
                tmptree.push(xip.clone());
            }
        }
        let mut tree = vec![];
        for mut it in tmptree {
            Self::recurse_build_tree(items, &mut it);
            tree.push(it);
        }
        tree
    }
}