uxum-pools 0.1.0

Generic instrumented resource pools for use in uxum framework
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

use std::time::Duration;

use deadpool::managed::{Manager, Object, Pool, PoolError, Timeouts};

use crate::r#async::*;

#[async_trait::async_trait]
impl<'p, M, W> InstrumentablePool<'p> for Pool<M, W>
where
    M: Manager,
    M::Error: std::error::Error + 'static,
    W: From<Object<M>> + 'p,
{
    type Resource = W;
    type Error = M::Error;

    async fn get(&'p self) -> Result<Self::Resource, Error<Self::Error>> {
        Pool::get(self).await.map_err(|e| match e {
            // TODO: further specializee timeout types
            PoolError::Timeout(_) => Error::AcquireTimeout,
            PoolError::Backend(err) => Error::Pool(err),
            _ => Error::PoolExhausted,
        })
    }

    async fn get_timeout(
        &'p self,
        timeout: Duration,
    ) -> Result<Self::Resource, Error<Self::Error>> {
        let timeouts = Timeouts {
            wait: Some(timeout),
            create: Some(timeout),
            recycle: Some(timeout),
        };
        Pool::timeout_get(self, &timeouts)
            .await
            .map_err(|e| match e {
                // TODO: further specializee timeout types
                PoolError::Timeout(_) => Error::AcquireTimeout,
                PoolError::Backend(err) => Error::Pool(err),
                _ => Error::PoolExhausted,
            })
    }

    fn get_state(&'p self) -> Result<PoolState, Error<Self::Error>> {
        let inner = Pool::status(self);
        Ok(PoolState {
            max_size: Some(inner.max_size),
            size: Some(inner.size),
            idle: Some(inner.available),
            in_use: Some({
                if inner.size > inner.available {
                    inner.size - inner.available
                } else {
                    0
                }
            }),
            min_idle: None,
            max_idle: None,
        })
    }
}