八、Rust 集成 Sqlx

八、Rust 集成 Sqlx
　　
　　Rust 语言级异步，是 xxx 版添加进来的特性。如果只是需要一款支持异步的连接池工具，推荐 tokio-postgres、deadpool-postgres。而用于生产工序，还是需要 ORM 的加入。本节我们来学习一下 sqlx。
　　
　　本节需要 PostgreSQL 数据库，需要读者提前准备，并假定实例中已包含名为 postgres 的库，且账号为：postgres:postgres。
 

1、目录结构

.
├── Cargo.lock
├── Cargo.toml
├── README.md
├── config.yaml
└── src
    ├── boot
    │   ├── db.rs
    │   └── mod.rs
    ├── main.rs
    └── module
        ├── mod.rs
        └── user
            ├── api.rs
            ├── mod.rs
            └── model.rs

2、开始集成

1）追加依赖：Cargo.toml

serde_yaml = "0.8.13"
sqlx = {version = "0.4.0-beta.1", default-features = false, features = ["runtime-tokio", "postgres", "macros", "chrono"]}
once_cell = "1.3.1"
chrono = { version = "0.4.15",features = ["serde"] }

2）配置文件：config.yaml

server:
  port: 8080
  log: data/logs/app/web.log
  env: dev
postgres:
  dsn: postgres://postgres:postgres@192.168.24.251:5432/postgres
  min: 5
  max: 15

3）代码：src/boot/mod.rs

use std::sync::Arc;

use once_cell::sync::OnceCell;
use serde::Deserialize;
use crate::boot;

pub mod db;

#[derive(Debug, Deserialize)]
pub struct Conf {
    pub server: Server,
    pub postgres: Option<Postgres>,
}

#[derive(Debug, Deserialize)]
pub struct Server {
    pub port: u32,
    pub log: String,
    pub env: String,
}

#[derive(Debug, Deserialize)]
pub struct Postgres {
    pub dsn: String,
    pub min: u32,
    pub max: u32,
}

#[allow(dead_code)]
pub fn global() -> &'static Arc<Conf> {
    static CONFIG: OnceCell<Arc<Conf>> = OnceCell::new();
    CONFIG.get_or_init(|| {
        let s = std::fs::read_to_string(&"config.yaml").unwrap();
        Arc::new(serde_yaml::from_str(&s).unwrap())
    })
}

pub fn start() {
    boot::db::init_db_pool()
}

impl Conf {
    #[allow(dead_code)]
    pub fn get_env(&self) -> String {
        self.server.env.clone()
    }

    #[allow(dead_code)]
    pub fn addr(&self) -> String {
        format!("0.0.0.0:{}", self.server.port)
    }
}

4）代码：src/boot/db.rs

use once_cell::sync::OnceCell;
use sqlx::{Pool, Postgres};
use sqlx::postgres::PgPoolOptions;

pub static POSTGRES_POOL: OnceCell<Pool<Postgres>> = OnceCell::new();

pub fn init_db_pool() {
    if let Some(pg) = &crate::boot::global().postgres {
        let pool = PgPoolOptions::new()
            .min_connections(pg.min)
            .max_connections(pg.max)
            .connect_lazy(&pg.dsn).unwrap();
        assert!(POSTGRES_POOL.set(pool).is_ok());
        println!("datasource: {}\n       min: {}\n       max: {}", pg.dsn, pg.min, pg.max)
    }
}

#[allow(dead_code)]
pub fn get_pool() -> Option<&'static Pool<Postgres>> {
    POSTGRES_POOL.get()
}

5）代码：src/main.rs

#[macro_use]
extern crate validator_derive;

use actix_web::{App, HttpServer};

mod boot;			// 增加了这里
mod module;

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    boot::start();	// 增加了这里
    HttpServer::new(move || App::new()
        .service(module::handler::api_routes())
    ).bind(boot::global().addr())?.run().await
}

 　　至此，便使我们的项目集成了 Sqlx，这款 Rust 异步 ORM 框架。现在，可以尝试编译，和启动我们的服务。

 $ cargo run
   Compiling rust-demo v0.1.0 (~/Rust-Study/rust-demo)
    Finished dev [unoptimized + debuginfo] target(s) in 6.52s
     Running `target/debug/rust-demo`
datasource: postgres://postgres:postgres@192.168.24.251:5432/postgres
       min: 5
       max: 15

 　　或

 $ ./target/debug/rust-demo
datasource: postgres://postgres:postgres@192.168.24.251:5432/postgres
       min: 5
       max: 15

 　　
　　为使每一小节，都尽可能简单/轻巧，本节就先这样。下一节，我们具体来看，使用 Sqlx 访问数据库 ！～