diesel rust orm 框架试用

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:dalong@localhost/diesel_demo

.env-prod： docker 生产使用

DATABASE_URL=postgres://postgres:dalong@pg/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

 

 

• 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