dieselrustorm框架试用

原作者: [db:作者] 来自: [db:来源] 收藏邀请

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"