1
0
mirror of https://github.com/actix/actix-extras.git synced 2024-11-28 01:32:57 +01:00

added diesel example

This commit is contained in:
Nikolay Kim 2017-12-18 16:25:26 -08:00
parent 3e8a6c3988
commit fde94bfe95
9 changed files with 173 additions and 0 deletions

1
examples/diesel/.env Normal file
View File

@ -0,0 +1 @@
DATABASE_URL=file:test.db

View File

@ -0,0 +1,18 @@
[package]
name = "diesel-example"
version = "0.1.0"
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
[dependencies]
env_logger = "0.4"
actix = "^0.3.1"
actix-web = { git = "https://github.com/actix/actix-web.git" }
futures = "0.1"
uuid = { version = "0.5", features = ["serde", "v4"] }
serde = "1.0"
serde_json = "1.0"
serde_derive = "1.0"
diesel = { version = "1.0.0-beta1", features = ["sqlite"] }
dotenv = "0.10"

15
examples/diesel/README.md Normal file
View File

@ -0,0 +1,15 @@
Diesel's `Getting Started` guide using SQLite for Actix web
## Usage
install `diesel_cli`
```
cargo install diesel_cli --no-default-features --features sqlite
```
```
$ echo "DATABASE_URL=file:test.db" > .env
$ diesel migration run
```

View File

@ -0,0 +1 @@
DROP TABLE users

View File

@ -0,0 +1,4 @@
CREATE TABLE users (
id VARCHAR NOT NULL PRIMARY KEY,
name VARCHAR NOT NULL
)

114
examples/diesel/src/main.rs Normal file
View File

@ -0,0 +1,114 @@
//! Actix web diesel example
//!
//! Diesel does not support tokio, so we have to run it in separate threads.
//! Actix supports sync actors by default, so we going to create sync actor that will
//! use diesel. Technically sync actors are worker style actors, multiple of them
//! can run in parallele and process messages from same queue.
extern crate serde;
extern crate serde_json;
#[macro_use]
extern crate serde_derive;
#[macro_use]
extern crate diesel;
extern crate uuid;
extern crate futures;
extern crate actix;
extern crate actix_web;
extern crate env_logger;
use actix::prelude::*;
use actix_web::*;
use futures::future::{Future, ok};
use diesel::Connection;
use diesel::sqlite::SqliteConnection;
use diesel::prelude::*;
mod models;
mod schema;
struct State {
db: SyncAddress<DbExecutor>,
}
fn index(req: HttpRequest<State>) -> Box<Future<Item=HttpResponse, Error=Error>> {
let name = &req.match_info()["name"];
Box::new(
req.state().db.call_fut(CreatePerson{name: name.to_owned()})
.and_then(|res| {
match res {
Ok(person) => ok(httpcodes::HTTPOk.build().json(person).unwrap()),
Err(_) => ok(httpcodes::HTTPInternalServerError.response())
}
})
.map_err(|e| error::ErrorInternalServerError(e).into()))
}
/// This is db executor actor. We are going to run 3 of them in parallele.
struct DbExecutor(SqliteConnection);
/// This is only message that this actor can handle, but it is easy to extend number of
/// messages.
struct CreatePerson {
name: String,
}
impl ResponseType for CreatePerson {
type Item = models::User;
type Error = Error;
}
impl Actor for DbExecutor {
type Context = SyncContext<Self>;
}
impl Handler<CreatePerson> for DbExecutor {
fn handle(&mut self, msg: CreatePerson, _: &mut Self::Context)
-> Response<Self, CreatePerson>
{
use self::schema::users::dsl::*;
let uuid = format!("{}", uuid::Uuid::new_v4());
let new_user = models::NewUser {
id: &uuid,
name: &msg.name,
};
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");
Self::reply(items.pop().unwrap())
}
}
fn main() {
::std::env::set_var("RUST_LOG", "actix_web=info");
let _ = env_logger::init();
let sys = actix::System::new("diesel-example");
// Start db executor actors
let addr = SyncArbiter::start(3, || {
DbExecutor(SqliteConnection::establish("test.db").unwrap())
});
// Start http server
HttpServer::new(move || {
Application::with_state(State{db: addr.clone()})
// enable logger
.middleware(middlewares::Logger::default())
.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();
}

View File

@ -0,0 +1,14 @@
use super::schema::users;
#[derive(Serialize, Queryable)]
pub struct User {
pub id: String,
pub name: String,
}
#[derive(Insertable)]
#[table_name = "users"]
pub struct NewUser<'a> {
pub id: &'a str,
pub name: &'a str,
}

View File

@ -0,0 +1,6 @@
table! {
users (id) {
id -> Text,
name -> Text,
}
}

BIN
examples/diesel/test.db Normal file

Binary file not shown.