deadpool-aykroyd 0.1.0

Aykroyd micro-ORM for the deadpool connection pool.
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

//! Aykroyd support for the `deadpool` connection pool.

use async_trait::async_trait;
use aykroyd::async_client::Client;
use tokio_postgres::tls::{MakeTlsConnect, TlsConnect};
use tokio_postgres::Socket;

type RecycleResult = deadpool::managed::RecycleResult<tokio_postgres::Error>;
type RecycleError = deadpool::managed::RecycleError<tokio_postgres::Error>;

pub use deadpool_postgres::{ManagerConfig, RecyclingMethod};

pub type Object<T> = deadpool::managed::Object<Manager<T>>;
pub type Pool<T> = deadpool::managed::Pool<Manager<T>, deadpool::managed::Object<Manager<T>>>;
pub type PoolBuilder<T> = deadpool::managed::PoolBuilder<Manager<T>>;
pub type PoolError = deadpool::managed::PoolError<tokio_postgres::Error>;

#[derive(Debug)]
pub struct Manager<T> {
    config: ManagerConfig,
    pg_config: tokio_postgres::Config,
    tls: T,
}

impl<T> Manager<T> {
    pub fn new(pg_config: tokio_postgres::Config, tls: T) -> Self {
        Self::from_config(pg_config, tls, ManagerConfig::default())
    }

    pub fn from_config(pg_config: tokio_postgres::Config, tls: T, config: ManagerConfig) -> Self {
        Manager { config, pg_config, tls }
    }
}

#[async_trait]
impl<T> deadpool::managed::Manager for Manager<T>
where
    T: MakeTlsConnect<Socket> + Clone + Sync + Send + 'static,
    T::Stream: Sync + Send,
    T::TlsConnect: Sync + Send,
    <T::TlsConnect as TlsConnect<Socket>>::Future: Send,
{
    type Type = Client;
    type Error = tokio_postgres::Error;

    async fn create(&self) -> Result<Client, tokio_postgres::Error> {
        let (client, connection) = self.pg_config.connect(self.tls.clone()).await?;
        tokio::spawn(async move {
            if let Err(e) = connection.await {
                //log::warn!(target: "deadpool.postgres", "Connection error: {}", e);
                panic!("Error in deadpool-aykroyd: connection error: {e}");
            }
        });
        Ok(Client::new(client))
    }

    async fn recycle(&self, client: &mut Client) -> RecycleResult {
        if client.as_ref().is_closed() {
            //log::info!(target: "deadpool.postgres", "Connection could not be recycled: Connection closed");
            return Err(RecycleError::StaticMessage("Connection closed"));
        }
        match self.config.recycling_method.query() {
            Some(sql) => match client.as_ref().simple_query(sql).await {
                Ok(_) => Ok(()),
                Err(e) => {
                    //log::info!(target: "deadpool.postgres", "Connection could not be recycled: {}", e);
                    Err(e.into())
                }
            },
            None => Ok(()),
        }
    }
}