db-derive

db-derive 是一个使用 r2d2、postgres 和 rusqlite 包装 PostgreSQL 和 SQLite 的库，用于 stry 故事托管服务器。

MSRV: 1.40.0

警告

直到 db-derive 升级到 v0.1.x，API 可能会发生变化。这将会发生。

示例

模式

#[derive(db_derive::Table)]
#[table(exists, schema)]
struct Tag {
    #[table(rename = "Id", primary, length = 10)]
    id: String,

    #[table(rename = "Name", unique, length = 256)]
    name: Option<String>,
}

fn main() -> Result<(), db_derive::Error> {
    let pool = Pool::sqlite("example.db")?;

    pool.transaction(|mut trans| {
        trans.schema::<Tag>()?;

        Ok(())
    })?;

    Ok(())
}

执行

#[derive(db_derive::Execute)]
#[execute(sql = "INSERT INTO Tag (Id, Name) VALUES ($id, $name);")]
struct AddTag<'a> {
    id: &'a str,
    name: Option<&'a str>,
}

fn main() -> Result<(), db_derive::Error> {
    let pool = Pool::sqlite("example.db")?;

    pool.transaction(|trans| {
        let add = AddTag {
            id: "example",
            name: Some("Example"),
        };

        add.execute(trans)?;

        Ok(())
    })?;

    Ok(())
}

查询

#[derive(db_derive::Query)]
#[query(sql = "SELECT Id, Name FROM Tag WHERE Id = $id;")]
struct GetTagById<'a> {
    id: &'a str,
}

fn main() -> Result<(), db_derive::Error> {
    let pool = Pool::sqlite("example.db")?;

    let get = GetTagById { id: "example" };

    let tag = get.query_row::<_, Tag>(&pool)?;

    println!("Tag Name: {}", tag.name);

    Ok(())
}

表

#[derive(db_derive::Execute)]
#[execute(sql = "SELECT Id, Name FROM Tag WHERE Id = $id;")]
struct GetTagById<'a> {
    id: &'a str,
}

fn main() -> Result<(), db_derive::Error> {
    let pool = Pool::sqlite("example.db")?;

    let get = GetTagById { id: "example" };

    let tag = get.query_row::<_, Tag>(&pool)?;

    println!("Tag Name: {}", tag.name);

    Ok(())
}

类型

#[derive(db_derive::Kind)]
enum State {
    #[kind(rename = "running")]
    Running,

    #[kind(rename = "stopped")]
    Stopped,
}