55 lines
2.2 KiB
Rust
55 lines
2.2 KiB
Rust
//! Run against a disposable database: creates and deletes a demonstration post.
|
|
use chusql_driver::{Client, Query};
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let address = std::env::var("CHUSQL_ADDRESS").unwrap_or_else(|_| "127.0.0.1:7777".into());
|
|
let db = Client::connect(
|
|
address,
|
|
&std::env::var("CHUSQL_USER")?,
|
|
&std::env::var("CHUSQL_PASSWORD")?,
|
|
)
|
|
.await?;
|
|
// Use a dedicated session so USE and subsequent operations cannot interleave with other handlers.
|
|
let database = std::env::var("CHUSQL_DATABASE")?;
|
|
if database.is_empty()
|
|
|| !database
|
|
.chars()
|
|
.all(|c| c.is_ascii_alphanumeric() || c == '_')
|
|
|| database.starts_with(|c: char| c.is_ascii_digit())
|
|
{
|
|
return Err("CHUSQL_DATABASE must be a plain identifier".into());
|
|
}
|
|
db.query(&format!("USE {database}")).await?;
|
|
// Run once in a disposable database; table must not already exist.
|
|
db.query("CREATE TABLE driver_blog_posts (id INT PRIMARY KEY, title TEXT, body TEXT, published BOOL)").await?;
|
|
let insert = Query::new("INSERT INTO driver_blog_posts (id, title, body, published) VALUES (")
|
|
.bind(1_i64)
|
|
.push(", ")
|
|
.bind("ChuSQL 博客")
|
|
.push(", ")
|
|
.bind("第一篇文章：It's simple.\n正文支持换行。")
|
|
.push(", ")
|
|
.bind(true)
|
|
.push(")");
|
|
db.query(insert.as_str()).await?;
|
|
let update = Query::new("UPDATE driver_blog_posts SET title = ")
|
|
.bind("更新后的标题")
|
|
.push(" WHERE id = ")
|
|
.bind(1_i64);
|
|
db.query(update.as_str()).await?;
|
|
let select = Query::new("SELECT id, title, body FROM driver_blog_posts WHERE published = ")
|
|
.bind(true)
|
|
.push(" ORDER BY id DESC LIMIT 10");
|
|
let result = db.query(select.as_str()).await?;
|
|
assert!(!result.truncated);
|
|
assert_eq!(result.rows.len(), 1);
|
|
assert_eq!(result.rows[0][1].as_str(), Some("更新后的标题"));
|
|
println!("{:?}", result.rows);
|
|
let delete = Query::new("DELETE FROM driver_blog_posts WHERE id = ").bind(1_i64);
|
|
db.query(delete.as_str()).await?;
|
|
db.query("DROP TABLE driver_blog_posts").await?;
|
|
db.close().await;
|
|
Ok(())
|
|
}
|