rbdc_sqlite/
driver.rs

1use crate::SqliteConnectOptions;
2use futures_core::future::BoxFuture;
3use rbdc::db::{ConnectOptions, Connection, Driver, Placeholder};
4use rbdc::Error;
5
6#[derive(Debug)]
7pub struct SqliteDriver {}
8
9impl Driver for SqliteDriver {
10    fn name(&self) -> &str {
11        "sqlite"
12    }
13
14    fn connect(&self, url: &str) -> BoxFuture<'_, Result<Box<dyn Connection>, Error>> {
15        let url = url.to_owned();
16        Box::pin(async move {
17            let mut opt = self.default_option();
18            opt.set_uri(&url)?;
19            if let Some(opt) = opt.downcast_ref::<SqliteConnectOptions>() {
20                let conn = opt.connect().await?;
21                Ok(Box::new(conn) as Box<dyn Connection>)
22            } else {
23                Err(Error::from("downcast_ref failure"))
24            }
25        })
26    }
27
28    fn connect_opt<'a>(
29        &'a self,
30        opt: &'a dyn ConnectOptions,
31    ) -> BoxFuture<'a, Result<Box<dyn Connection>, Error>> {
32        let opt: &SqliteConnectOptions = opt.downcast_ref().unwrap();
33        Box::pin(async move {
34            let conn = opt.connect().await?;
35            Ok(Box::new(conn) as Box<dyn Connection>)
36        })
37    }
38
39    fn default_option(&self) -> Box<dyn ConnectOptions> {
40        Box::new(SqliteConnectOptions::default())
41    }
42}
43
44impl Placeholder for SqliteDriver {
45    fn exchange(&self, sql: &str) -> String {
46        sql.to_string()
47    }
48}
49
50#[cfg(test)]
51mod test {
52    #[test]
53    fn test_default() {}
54}
55// #[cfg(test)]
56// mod test {
57//     use crate::driver::SqliteDriver;
58//     use crate::SqliteConnectOptions;
59//     use rbdc::block_on;
60//     use rbdc::db::{ConnectOptions, Driver};
61//     use rbdc::decimal::Decimal;
62//     use rbdc::pool::Pool;
63//     use rbs::Value;
64//     use std::fs::File;
65//
66//     #[test]
67//     fn test_sqlite_pool() {
68//         let f = File::create("../target/test.db");
69//         if f.is_err() {
70//             println!("{}", f.err().unwrap());
71//         } else {
72//             drop(f);
73//         }
74//         let f = async move {
75//             let pool = Pool::new_url(SqliteDriver {}, "sqlite://../target/test.db").unwrap();
76//             let mut conn = pool.get().await.unwrap();
77//             conn.exec(
78//                 "CREATE TABLE `biz_activity`
79// (
80//     `id`            TEXT PRIMARY KEY NOT NULL,
81//     `name`          TEXT     DEFAULT NULL,
82//     `pc_link`       TEXT     DEFAULT NULL,
83//     `h5_link`       TEXT     DEFAULT NULL,
84//     `sort`          TEXT     DEFAULT NULL,
85//     `status`        INT      DEFAULT NULL,
86//     `version`       INT      DEFAULT NULL,
87//     `remark`        TEXT     DEFAULT NULL,
88//     `create_time`   datetime DEFAULT NULL,
89//     `delete_flag`   INT(1)   DEFAULT NULL,
90//     `pc_banner_img` TEXT     DEFAULT NULL,
91//     `h5_banner_img` TEXT     DEFAULT NULL
92// );
93//
94// INSERT INTO `biz_activity`
95// VALUES ('1', '活动1', NULL, NULL, '1', 1, 1, 'fff', '2019-12-12 00:00:00', 0, NULL, NULL),
96//        ('178', 'test_insret', '', '', '1', 1, 0, '', '2020-06-17 20:08:13', 0, NULL, NULL),
97//        ('221', 'test', '', '', '0', 0, 0, '', '2020-06-17 20:10:23', 0, NULL, NULL),
98//        ('222', 'test', '', '', '0', 0, 0, '', '2020-06-17 20:10:23', 0, NULL, NULL),
99//        ('223', 'test', '', '', '0', 0, 0, '', '2020-06-17 20:10:23', 0, NULL, NULL);",
100//                 vec![],
101//             )
102//             .await;
103//
104//             let data = conn
105//                 .get_values("select * from biz_activity", vec![])
106//                 .await
107//                 .unwrap();
108//             for mut x in data {
109//                 println!("row: {}", x);
110//             }
111//         };
112//         block_on!(f);
113//     }
114//
115//     #[test]
116//     fn test_sqlite_param() {
117//         let task = async move {
118//             let mut d = SqliteDriver {};
119//             let mut c = d.connect("sqlite://../target/test.db").await.unwrap();
120//             let param = vec![
121//                 Decimal("1".to_string()).into(),
122//                 Value::String("1".to_string()),
123//             ];
124//             println!("param => {}", Value::Array(param.clone()));
125//             let data = c
126//                 .exec("update biz_activity set version = ? where id  = ?", param)
127//                 .await
128//                 .unwrap();
129//             println!("{}", data);
130//         };
131//         block_on!(task);
132//     }
133// }
rbdc_sqlite/driver.rs

rbdc_sqlite/
driver.rs
