r2d2-sqlite3 0.1.0

Sqlite support for the r2d2 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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

extern crate sqlite;
extern crate r2d2;
extern crate r2d2_sqlite3;
extern crate tempdir;

use std::sync::mpsc;
use std::thread;

use r2d2::ManageConnection;
use tempdir::TempDir;

use r2d2_sqlite3::SqliteConnectionManager;
use sqlite::Connection;

#[test]
fn test_basic() {
    let manager = SqliteConnectionManager::file("file.db");
    let config = r2d2::Config::builder().pool_size(2).build();
    let pool = r2d2::Pool::new(config, manager).unwrap();

    let (s1, r1) = mpsc::channel();
    let (s2, r2) = mpsc::channel();

    let pool1 = pool.clone();
    let t1 = thread::spawn(move || {
                               let conn = pool1.get().unwrap();
                               s1.send(()).unwrap();
                               r2.recv().unwrap();
                               drop(conn);
                           });

    let pool2 = pool.clone();
    let t2 = thread::spawn(move || {
                               let conn = pool2.get().unwrap();
                               s2.send(()).unwrap();
                               r1.recv().unwrap();
                               drop(conn);
                           });

    t1.join().unwrap();
    t2.join().unwrap();

    pool.get().unwrap();
}

#[test]
fn test_file() {
    let manager = SqliteConnectionManager::file("file.db");
    let config = r2d2::Config::builder().pool_size(2).build();
    let pool = r2d2::Pool::new(config, manager).unwrap();

    let (s1, r1) = mpsc::channel();
    let (s2, r2) = mpsc::channel();

    let pool1 = pool.clone();
    let t1 = thread::spawn(move || {
                               let conn = pool1.get().unwrap();
                               let conn1: &Connection = &*conn;
                               s1.send(()).unwrap();
                               r2.recv().unwrap();
                               drop(conn1);
                           });

    let pool2 = pool.clone();
    let t2 = thread::spawn(move || {
                               let conn = pool2.get().unwrap();
                               s2.send(()).unwrap();
                               r1.recv().unwrap();
                               drop(conn);
                           });

    t1.join().unwrap();
    t2.join().unwrap();

    pool.get().unwrap();
}

#[test]
fn test_is_valid() {
    let manager = SqliteConnectionManager::file("file.db");
    let config = r2d2::Config::builder()
        .pool_size(1)
        .test_on_check_out(true)
        .build();
    let pool = r2d2::Pool::new(config, manager).unwrap();

    pool.get().unwrap();
}

#[test]
fn test_error_handling() {
    //! We specify a directory as a database. This is bound to fail.
    let dir = TempDir::new("r2d2-sqlite").expect("Could not create temporary directory");
    let dirpath = dir.path().to_str().unwrap();
    let manager = SqliteConnectionManager::file(dirpath);
    assert!(manager.connect().is_err());
}