数据库

Diesel

目前,Diesel 1.0不支持异步操作,但是可以将actix synchronous actor系统用作数据库接口api。

从技术上讲,sync actors是worker风格的actors。多个sync actors可以并行运行,并处理来自同一队列的消息。Sync actors在mpsc模式下工作。

让我们创建一个简单的数据库api,可以将新的user行插入SQLite表。我们必须定义一个sync actor和该actor将使用的连接。相同的方法可以用于其他数据库。

use actix::prelude::*;

struct DbExecutor(SqliteConnection);

impl Actor for DbExecutor {
    type Context = SyncContext<Self>;
}

这是我们actor的定义。现在,我们必须定义*create user*消息和响应。

struct CreateUser {
    name: String,
}

impl Message for CreateUser {
    type Result = Result<User, Error>;
}

我们可以将CreateUser消息发送给DbExecutor actor,结果,我们将收到一个User模型实例。接下来,我们必须为此消息定义handler实现。

impl Handler<CreateUser> for DbExecutor {
    type Result = Result<User, Error>;

    fn handle(&mut self, msg: CreateUser, _: &mut Self::Context) -> Self::Result {
        use self::schema::users::dsl::*;

        // Create insertion model
        let uuid = format!("{}", uuid::Uuid::new_v4());
        let new_user = models::NewUser {
            id: &uuid,
            name: &msg.name,
        };

        // normal diesel operations
        diesel::insert_into(users)
            .values(&new_user)
            .execute(&self.0)
            .expect("Error inserting person");

        let mut items = users
            .filter(id.eq(&uuid))
            .load::<models::User>(&self.0)
            .expect("Error loading person");

        Ok(items.pop().unwrap())
    }
}

现在,我们可以从任何http handler或中间件中使用DbExecutor actor。我们需要做的就是启动DbExecutor actor并将地址存储在http handler可以访问它的状态下。

/// This is state where we will store *DbExecutor* address.
struct State {
    db: Addr<DbExecutor>,
}

fn main() {
    let sys = actix::System::new("diesel-example");

    // Start 3 parallel db executors
    let addr = SyncArbiter::start(3, || {
        DbExecutor(SqliteConnection::establish("test.db").unwrap())
    });

    // Start http server
    HttpServer::new(move || {
        App::with_state(State { db: addr.clone() })
            .resource("/{name}", |r| r.method(Method::GET).a(index))
    })
    .bind("127.0.0.1:8080")
    .unwrap()
    .start()
    .unwrap();

    println!("Started http server: 127.0.0.1:8080");
    let _ = sys.run();
}

我们将在请求handler中使用该地址。handle(句柄)返回一个future对象;因此,我们异步接收消息响应。 Route::a()必须用于 async handler注册。

/// Async handler
fn index(req: &HttpRequest<State>) -> Box<Future<Item = HttpResponse, Error = Error>> {
    let name = &req.match_info()["name"];

    // Send message to `DbExecutor` actor
    req.state()
        .db
        .send(CreateUser {
            name: name.to_owned(),
        })
        .from_err()
        .and_then(|res| match res {
            Ok(user) => Ok(HttpResponse::Ok().json(user)),
            Err(_) => Ok(HttpResponse::InternalServerError().into()),
        })
        .responder()
}

示例目录中提供了完整的示例。

可以在actix文档中找到有关sync actors的更多信息。