diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/main.rs | 82 |
1 files changed, 66 insertions, 16 deletions
diff --git a/src/main.rs b/src/main.rs index 6e98fe8b1d73..559db8cc9662 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,33 +1,83 @@ #[macro_use] extern crate diesel; + +#[macro_use] +extern crate log; + extern crate chrono; +extern crate actix; +extern crate actix_web; +extern crate env_logger; +extern crate r2d2; +extern crate futures; pub mod schema; pub mod models; +pub mod db; + +use actix::prelude::*; +use actix_web::*; -use diesel::prelude::*; use diesel::pg::PgConnection; +use diesel::r2d2::{ConnectionManager, Pool}; use std::env; +use db::*; +use futures::Future; +use models::Thread; -fn connect_db() -> PgConnection { - let db_url = env::var("DATABASE_URL") - .expect("DATABASE_URL must be set"); +/// Represents the state carried by the web server actors. +struct AppState { + db: Addr<Syn, DbExecutor>, +} + +/// Really inefficient renderer example! +fn render_threads(threads: Vec<Thread>) -> String { + let mut res = String::new(); + + for thread in threads { + res.push_str(&format!("Subject: {}\n", thread.title)); + res.push_str(&format!("Posted at: {}\n\n", thread.posted)); + res.push_str(&format!("{}\n", thread.body)); + res.push_str("-------------------------------"); + } - PgConnection::establish(&db_url) - .expect(&format!("Error connecting to {}", db_url)) + res +} + +fn forum_index(req: HttpRequest<AppState>) -> FutureResponse<HttpResponse> { + req.state().db.send(ListThreads) + .from_err() + .and_then(|res| match res { + Ok(threads) => Ok(HttpResponse::from(render_threads(threads))), + Err(err) => { + error!("Error loading threads: {}", err); + Ok(HttpResponse::InternalServerError().into()) + } + }) + .responder() } fn main() { - use schema::threads::dsl::*; - use models::*; + env_logger::init(); - let conn = connect_db(); - let result: Vec<Thread> = threads - .load::<Thread>(&conn) - .expect("Error loading threads"); + info!("Welcome to Converse! Hold on tight while we're getting ready."); + let sys = actix::System::new("converse"); - for thread in result { - println!("Subject: {}\nPosted: {}\n", thread.title, thread.posted); - println!("{}", thread.body); - } + info!("Initialising database connection pool ..."); + let db_url = env::var("DATABASE_URL") + .expect("DATABASE_URL must be set"); + + let manager = ConnectionManager::<PgConnection>::new(db_url); + let pool = Pool::builder().build(manager).expect("Failed to initialise DB pool"); + + let db_addr = SyncArbiter::start(2, move || DbExecutor(pool.clone())); + + info!("Initialising HTTP server ..."); + server::new(move || { + App::with_state(AppState { db: db_addr.clone() }) + .middleware(middleware::Logger::default()) + .route("/", http::Method::GET, &forum_index) + }).bind("127.0.0.1:4567").unwrap().start(); + + let _ = sys.run(); } |