add user test db

Author: Jzow

dao/src/db.rs

@@ -0,0 +1,28 @@
+use sqlx::{mysql::MySqlPool, Pool, MySql};
+use once_cell::sync::Lazy;
+use std::sync::Mutex;
+use utils::config::load_config;
+
+static DB_POOL: Lazy<Mutex<Option<Pool<MySql>>>> = Lazy::new(|| Mutex::new(None));
+
+pub async fn get_db_pool() -> Result<Pool<MySql>, sqlx::Error> {
+    // load config
+    let config = load_config();
+    let mut pool_lock = DB_POOL.lock().unwrap();
+
+    // check if the pool has already been initialized
+    if let Some(pool) = &*pool_lock {
+        return Ok(pool.clone());
+    }
+
+    // load database url from config file
+    let database_url = config.database.url.clone();
+
+    // create a new pool
+    let pool = MySqlPool::connect(&database_url).await?;
+
+    // set the pool in the static variable
+    *pool_lock = Some(pool.clone());
+
+    Ok(pool)
+}

dao/src/user/mod.rs

@@ -0,0 +1 @@
+pub mod user_repository;

dao/src/user/user_repository.rs

@@ -0,0 +1,10 @@
+use sqlx::MySqlPool;
+use domain::entity::user::user::SysUser;
+
+pub async fn get_all_users(pool: &MySqlPool) -> Result<Vec<SysUser>, sqlx::Error> {
+    let users = sqlx::query_as::<_, SysUser>("SELECT * FROM sys_user")
+        .fetch_all(pool)
+        .await?;
+
+    Ok(users)
+}

domain/src/entity/mod.rs

@@ -0,0 +1 @@
+pub mod user;

domain/src/entity/user/mod.rs

@@ -0,0 +1 @@
+pub mod user;

domain/src/entity/user/user.rs

@@ -0,0 +1,88 @@
+use chrono::{NaiveDateTime};
+use serde::{Deserialize, Serialize};
+use sqlx::FromRow;
+use utils::naive_datetime_format;
+
+#[derive(Debug, Clone, Serialize, Deserialize, FromRow)]
+pub struct SysUser {
+    pub id: Option<i64>,                 // 对应 Long 类型
+    pub name: Option<String>,            // 用户姓名
+    pub user_name: Option<String>,       // 登录用户名
+    pub avatar: Option<String>,          // 头像
+    pub password: Option<String>,        // 登录密码
+    pub leader_flag: Option<bool>,       // 是否经理
+    pub position: Option<String>,        // 职位
+    pub email: Option<String>,           // 电子邮件
+    pub phone_number: Option<String>,    // 手机号码
+    pub system_language: Option<String>, // 系统语言
+    pub is_manager: Option<i32>,         // 是否为管理者 0==管理者 1==员工
+    pub is_system: Option<i32>,          // 是否系统自带数据
+    pub status: Option<i32>,             // 状态 0：正常，1：删除，2：封禁
+    pub description: Option<String>,     // 用户描述
+    pub remark: Option<String>,          // 备注
+    pub wechat_open_id: Option<String>,  // 微信绑定
+    pub tenant_id: Option<i64>,          // 租户id
+
+    // NaiveDateTime添加序列化和反序列化处理
+    #[serde(with = "naive_datetime_format")]
+    pub create_time: Option<NaiveDateTime>, // 创建时间
+
+    #[serde(with = "naive_datetime_format")]
+    pub update_time: Option<NaiveDateTime>, // 更新时间
+
+    pub create_by: Option<i64>,          // 创建人
+    pub update_by: Option<i64>,          // 修改人
+    pub delete_flag: Option<i32>,        // 删除标记
+}
+
+impl SysUser {
+    pub fn new(
+        id: Option<i64>,
+        name: Option<String>,
+        user_name: Option<String>,
+        avatar: Option<String>,
+        password: Option<String>,
+        leader_flag: Option<bool>,
+        position: Option<String>,
+        email: Option<String>,
+        phone_number: Option<String>,
+        system_language: Option<String>,
+        is_manager: Option<i32>,
+        is_system: Option<i32>,
+        status: Option<i32>,
+        description: Option<String>,
+        remark: Option<String>,
+        wechat_open_id: Option<String>,
+        tenant_id: Option<i64>,
+        create_time: Option<NaiveDateTime>,
+        update_time: Option<NaiveDateTime>,
+        create_by: Option<i64>,
+        update_by: Option<i64>,
+        delete_flag: Option<i32>,
+    ) -> Self {
+        SysUser {
+            id,
+            name,
+            user_name,
+            avatar,
+            password,
+            leader_flag,
+            position,
+            email,
+            phone_number,
+            system_language,
+            is_manager,
+            is_system,
+            status,
+            description,
+            remark,
+            wechat_open_id,
+            tenant_id,
+            create_time,
+            update_time,
+            create_by,
+            update_by,
+            delete_flag,
+        }
+    }
+}

utils/src/naive_datetime_format.rs

@@ -0,0 +1,28 @@
+use chrono::NaiveDateTime;
+use serde::{self, Deserialize, Deserializer, Serialize, Serializer};
+
+// custom format for NaiveDateTime to string
+const FORMAT: &str = "%Y-%m-%d %H:%M:%S";
+
+pub fn serialize<S>(dt: &Option<NaiveDateTime>, serializer: S) -> Result<S::Ok, S::Error>
+where
+    S: Serializer,
+{
+    match *dt {
+        Some(ref dt) => dt.format(FORMAT).to_string().serialize(serializer),
+        None => "".serialize(serializer),  // for None, serialize as empty string
+    }
+}
+
+pub fn deserialize<'de, D>(deserializer: D) -> Result<Option<NaiveDateTime>, D::Error>
+where
+    D: Deserializer<'de>,
+{
+    let opt_str: Option<String> = Option::deserialize(deserializer)?;
+    match opt_str {
+        Some(s) => NaiveDateTime::parse_from_str(&s, FORMAT)
+            .map(Some)
+            .map_err(serde::de::Error::custom),
+        None => Ok(None),
+    }
+}