deadpool-postgres 0.3.0

Dead simple async pool for tokio-postgres
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

use std::env;
use std::path::Path;

use deadpool_postgres::{Manager, Pool};

fn pg_config_from_env() -> tokio_postgres::config::Config {
    let mut config = tokio_postgres::Config::new();
    if let Ok(host) = env::var("PG_HOST") {
        config.host(host.as_str());
    } else if Path::new("/run/postgresql").exists() {
        config.host("/run/postgresql");
    } else {
        config.host("/tmp");
    }
    if let Ok(port) = env::var("PG_PORT") {
        match u16::from_str_radix(port.as_str(), 10) {
            Ok(port) => { config.port(port); }
            Err(_) => { panic!(format!("Invalid port: {}", port)); }
        }
    }
    if let Ok(user) = env::var("PG_USER") {
        config.user(user.as_str());
    } else if let Ok(user) = env::var("USER") {
        config.user(user.as_str());
    } else {
        panic!("PG_USER missing in environment; fallback to USER failed as well.");
    }
    if let Ok(password) = env::var("PG_PASSWORD") {
        config.password(password.as_str());
    }
    if let Ok(dbname) = env::var("PG_DBNAME") {
        config.dbname(dbname.as_str());
    } else {
        config.dbname("deadpool");
    }
    config
}

fn create_pool() -> Pool {
    let cfg = pg_config_from_env();
    let mgr = Manager::new(cfg, tokio_postgres::NoTls);
    Pool::new(mgr, 16)
}

#[tokio::main]
#[test]
async fn test_basic() {
    let pool = create_pool();
    let mut client = pool.get().await.unwrap();
    let stmt = client.prepare("SELECT 1 + 2").await.unwrap();
    let rows = client.query(&stmt, &[]).await.unwrap();
    let value: i32 = rows[0].get(0);
    assert_eq!(value, 3);
    assert_eq!(client.statement_cache.size(), 1);
}

#[tokio::main]
#[test]
async fn test_transaction_1() {
    let pool = create_pool();
    let mut client = pool.get().await.unwrap();
    {
        let mut txn = client.transaction().await.unwrap();
        let stmt = txn.prepare("SELECT 1 + 2").await.unwrap();
        let rows = txn.query(&stmt, &[]).await.unwrap();
        let value: i32 = rows[0].get(0);
        txn.commit().await.unwrap();
        assert_eq!(value, 3);
        assert_eq!(client.statement_cache.size(), 1);
    }
}

#[tokio::main]
#[test]
async fn test_transaction_2() {
    let pool = create_pool();
    let mut client = pool.get().await.unwrap();
    let stmt = client.prepare("SELECT 1 + 2").await.unwrap();
    {
        let txn = client.transaction().await.unwrap();
        let rows = txn.query(&stmt, &[]).await.unwrap();
        let value: i32 = rows[0].get(0);
        txn.commit().await.unwrap();
        assert_eq!(value, 3);
        assert_eq!(client.statement_cache.size(), 1);
    }
}