diesel rust orm 框架试用

Danialzhou 2020-02-03

diesel 是一个不错的rust orm框架，提供了cli ，可以方便的进行migration 操作，以及帮助代码生成
以下是一个简单的试用

安装cli

注意需要最新的stable 版本，数据库使用docker-compose 运行

cargo install diesel_cli

项目准备

  • docker-compose 服务
    关于rust 项目构建的，后边说明
 
version: "3"
services: 
  app:
    build: ./
  pg:
    image: postgres:12.1-alpine
    environment: 
    - "POSTGRES_PASSWORD:dalong"
    ports:
    - "5432:5432"
  • 初始化项目
cargo new --lib diesel_demo
cd diesel_demo
  • 添加diesel 依赖
    Cargo.toml
 
[dependencies]
diesel = { version = "1.0.0", features = ["postgres"] }
dotenv = "0.9.0"
  • 添加数据库环境变量
    项目使用dotenv管理环境变量,12 factor 模式
 
.env 本地开发模式使用
DATABASE_URL=postgres://postgres:da/diesel_demo
.env-prod： docker 生产使用
DATABASE_URL=postgres://postgres:/diesel_demo
  • 启动pg 服务
docker-compose up -d pg
  • 创建数据库
diesel setup
  • 创建一个实体
diesel migration generate create_posts
  • 添加migration 脚本
    up.sql
 
CREATE TABLE posts (
  id SERIAL PRIMARY KEY,
  title VARCHAR NOT NULL,
  body TEXT NOT NULL,
  published BOOLEAN NOT NULL DEFAULT ‘f‘
)

down.sql

DROP TABLE posts
  • schema migration
diesel migration run

代码使用

  • 创建连接
    src/lib.rs
 
#[macro_use]
extern crate diesel;
extern crate dotenv;
?
use diesel::prelude::*;
use diesel::pg::PgConnection;
use dotenv::dotenv;
use std::env;
pub mod schema;
pub mod models;
pub fn establish_connection() -> PgConnection {
    dotenv().ok();
?
    let database_url = env::var("DATABASE_URL")
        .expect("DATABASE_URL must be set");
    PgConnection::establish(&database_url)
        .expect(&format!("Error connecting to {}", database_url))
}
  • model
    src/models.rs
 
#[derive(Queryable)]
pub struct Post {
    pub id: i32,
    pub title: String,
    pub body: String,
    pub published: bool,
}
  • 生成的schema
table! {
    posts (id) {
        id -> Int4,
        title -> Varchar,
        body -> Text,
        published -> Bool,
    }
}
  • 代码入口
    src/bin/show_posts.rs
 
extern crate diesel_demo;
extern crate diesel;
?
use self::diesel_demo::*;
use self::models::*;
use self::diesel::prelude::*;
?
fn main() {
    use diesel_demo::schema::posts::dsl::*;
?
    let connection = establish_connection();
    let results = posts.filter(published.eq(true))
        .limit(5)
        .load::<Post>(&connection)
        .expect("Error loading posts");
?
    println!("Displaying {} posts", results.len());
    for post in results {
        println!("{}", post.title);
        println!("----------\n");
        println!("{}", post.body);
    }
}
  • insert demo 数据
INSERT INTO "public"."posts"("id","title","body","published")
VALUES
(1,E‘dalong‘,E‘demo‘,TRUE);
  • 本地运行
cargo run --bin show_posts

diesel rust orm 框架试用

  • docker运行
    dockerfile,使用多阶段构建(制作小的镜像)，同时为了提高构建的速度，使用了中科大的cargo镜像
FROM rustlang/rust:nightly-slim as build
WORKDIR /app
RUN mkdir -p $HOME/.cargo
COPY diesel_demo/ /app
# cargo 加速
COPY diesel_demo/config $HOME/.cargo/
RUN apt-get update && apt-get install -y libpq-dev
RUN rustup --version
RUN rustc --version && \
    rustup --version && \
    cargo --version
RUN cargo build --bin show_posts --release
?
FROM debian:stretch-slim
WORKDIR /app
RUN apt-get update && apt-get install -y libpq5
COPY diesel_demo/.env-prod /app/.env
COPY --from=build /app/target/release/show_posts /usr/local/bin/show_posts
CMD [ "show_posts" ]

加速配置
config

 
[source.crates-io]
registry = "https://github.com/rust-lang/crates.io-index"
replace-with = ‘ustc‘
[source.ustc]
registry = "git://mirrors.ustc.edu.cn/crates.io-index"

说明

从目前来说rust 以及python 还有php 等语言，使用alpine 做为基础镜像并不太好（依赖，libc 的不同，还有依赖）
使用debian，ubuntu，还有centos 作为这类语言的基础镜像会很不错，有时表象的小，可能并不是最优的方案

参考资料

http://diesel.rs/guides/getting-started/
https://github.com/rongfengliang/diesel-docker-running
https://www.cnblogs.com/rongfengliang/p/11337157.html
https://lug.ustc.edu.cn/wiki/mirrors/help/rust-crates

Danialzhou

Danialzhou