akita 0.7.0

Akita - Mini orm for rust.
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

/*
 *
 *  *
 *  *      Copyright (c) 2018-2025, SnackCloud All rights reserved.
 *  *
 *  *   Redistribution and use in source and binary forms, with or without
 *  *   modification, are permitted provided that the following conditions are met:
 *  *
 *  *   Redistributions of source code must retain the above copyright notice,
 *  *   this list of conditions and the following disclaimer.
 *  *   Redistributions in binary form must reproduce the above copyright
 *  *   notice, this list of conditions and the following disclaimer in the
 *  *   documentation and/or other materials provided with the distribution.
 *  *   Neither the name of the www.snackcloud.cn developer nor the names of its
 *  *   contributors may be used to endorse or promote products derived from
 *  *   this software without specific prior written permission.
 *  *   Author: SnackCloud
 *  *
 *
 */
use crate::database_err;
use crate::driver::non_blocking::get_tokio_context;
use crate::errors::AkitaError;
use async_trait::async_trait;
use deadpool::managed::{Metrics, Object, Pool, RecycleResult};
use deadpool::Runtime;
use mysql_async::prelude::Queryable;
use mysql_async::{Conn, Opts};
use tokio::runtime::Handle;

/// MySQL Asynchronous connection pool type
pub type MysqlAsyncPool = Pool<MysqlAsyncConnectionManager>;
/// MySQL Asynchronous connection type
pub type MysqlAsyncConnection = Object<MysqlAsyncConnectionManager>;

/// MySQL Asynchronous connection manager
#[derive(Clone)]
pub struct MysqlAsyncConnectionManager {
    opts: Opts,
    config: crate::config::AkitaConfig,
}

impl MysqlAsyncConnectionManager {
    /// Create a new connection manager
    pub fn new(config: &crate::config::AkitaConfig) -> Result<Self, AkitaError> {
        let connection_string = config.get_connection_string()?;

        let opts = Opts::from_url(&connection_string)
            .map_err(|e| database_err!(format!("Invalid MySQL URL: {}", e)))?;

        Ok(Self {
            opts,
            config: config.clone(),
        })
    }
}

#[async_trait]
impl deadpool::managed::Manager for MysqlAsyncConnectionManager {
    type Type = Conn;
    type Error = mysql_async::Error;

    async fn create(&self) -> Result<Self::Type, Self::Error> {
        let mut conn = Conn::new(self.opts.clone()).await?;

        // Setting connection options
        let _ = conn
            .query_drop(format!(
                "SET SESSION max_execution_time = {}",
                self.config.get_connection_timeout().as_millis()
            ))
            .await;
        Ok(conn)
    }

    async fn recycle(
        &self,
        obj: &mut Self::Type,
        _metrics: &Metrics,
    ) -> RecycleResult<Self::Error> {
        // Ping Check that the connection is valid
        obj.ping().await?;
        Ok(())
    }
}

/// Initialize MySQL asynchronous connection pool
pub async fn init_mysql_async_pool(
    config: crate::config::AkitaConfig,
) -> Result<MysqlAsyncPool, AkitaError> {
    let manager = MysqlAsyncConnectionManager::new(&config)?;
    // Check the Tokio context
    let _handle = get_tokio_context()?;
    let pool_config = deadpool::managed::PoolConfig {
        max_size: config.get_max_size() as usize,
        timeouts: deadpool::managed::Timeouts {
            wait: Some(config.get_connection_timeout()),
            create: Some(config.get_connection_timeout()),
            recycle: Some(config.get_idle_timeout()),
        },
        ..Default::default()
    };

    let pool = Pool::builder(manager)
        .runtime(Runtime::Tokio1)
        .config(pool_config)
        .build()?;

    // Testing connections
    let mut conn: MysqlAsyncConnection = pool
        .get()
        .await
        .map_err(|e| database_err!(format!("Failed to get connection from pool: {}", e)))?;
    conn.query_drop("SELECT 1")
        .await
        .map_err(|e| database_err!(format!("MySQL async connection test failed: {}", e)))?;

    tracing::info!("MySQL async connection pool initialized successfully");

    Ok(pool)
}