refactor(main): Move handlers into separate handler module
This commit is contained in:
parent
67a0642f8c
commit
c8cee945c5
2 changed files with 79 additions and 61 deletions
76
src/handlers.rs
Normal file
76
src/handlers.rs
Normal file
|
@ -0,0 +1,76 @@
|
||||||
|
//! This module contains the implementation of converse's actix-web
|
||||||
|
//! HTTP handlers.
|
||||||
|
//!
|
||||||
|
//! Most handlers have an associated rendering function using one of
|
||||||
|
//! the tera templates stored in the `/templates` directory in the
|
||||||
|
//! project root.
|
||||||
|
|
||||||
|
use tera;
|
||||||
|
use actix_web::*;
|
||||||
|
use models::*;
|
||||||
|
use db::*;
|
||||||
|
use actix::prelude::{Addr, Syn};
|
||||||
|
use futures::Future;
|
||||||
|
use errors::{Result, ConverseError};
|
||||||
|
|
||||||
|
type ConverseResponse = Box<Future<Item=HttpResponse, Error=ConverseError>>;
|
||||||
|
|
||||||
|
/// Represents the state carried by the web server actors.
|
||||||
|
pub struct AppState {
|
||||||
|
/// Address of the database actor
|
||||||
|
pub db: Addr<Syn, DbExecutor>,
|
||||||
|
|
||||||
|
/// Compiled templates
|
||||||
|
pub tera: tera::Tera,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This function renders an overview of threads into the default
|
||||||
|
/// thread list template.
|
||||||
|
fn render_threads(tpl: &tera::Tera, threads: Vec<Thread>) -> Result<HttpResponse> {
|
||||||
|
let mut ctx = tera::Context::new();
|
||||||
|
ctx.add("threads", &threads);
|
||||||
|
let body = tpl.render("index.html", &ctx)?;
|
||||||
|
Ok(HttpResponse::Ok().content_type("text/html").body(body))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn forum_index(state: State<AppState>) -> ConverseResponse {
|
||||||
|
state.db.send(ListThreads)
|
||||||
|
.from_err()
|
||||||
|
.and_then(move |res| match res {
|
||||||
|
Ok(threads) => Ok(render_threads(&state.tera, threads)?),
|
||||||
|
Err(err) => {
|
||||||
|
error!("Error loading threads: {}", err);
|
||||||
|
Ok(HttpResponse::InternalServerError().into())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This function renders a single forum thread into the default
|
||||||
|
/// thread view.
|
||||||
|
fn render_thread(tpl: &tera::Tera, thread: Thread, posts: Vec<Post>)
|
||||||
|
-> Result<HttpResponse> {
|
||||||
|
let mut ctx = tera::Context::new();
|
||||||
|
ctx.add("thread", &thread);
|
||||||
|
ctx.add("posts", &posts);
|
||||||
|
|
||||||
|
let body = tpl.render("thread.html", &ctx)?;
|
||||||
|
Ok(HttpResponse::Ok()
|
||||||
|
.content_type("text/html")
|
||||||
|
.body(body))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This handler retrieves and displays a single forum thread.
|
||||||
|
pub fn forum_thread(state: State<AppState>, thread_id: Path<i32>) -> ConverseResponse {
|
||||||
|
let id = thread_id.into_inner();
|
||||||
|
state.db.send(GetThread(id))
|
||||||
|
.from_err()
|
||||||
|
.and_then(move |res| match res {
|
||||||
|
Ok((thread, posts)) => Ok(render_thread(&state.tera, thread, posts)?),
|
||||||
|
Err(err) => {
|
||||||
|
error!("Error loading thread {}: {}", id, err);
|
||||||
|
Ok(HttpResponse::InternalServerError().into())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
64
src/main.rs
64
src/main.rs
|
@ -21,74 +21,16 @@ extern crate serde;
|
||||||
pub mod schema;
|
pub mod schema;
|
||||||
pub mod models;
|
pub mod models;
|
||||||
pub mod db;
|
pub mod db;
|
||||||
|
pub mod handlers;
|
||||||
|
|
||||||
use actix::prelude::*;
|
use actix::prelude::*;
|
||||||
use actix_web::*;
|
use actix_web::*;
|
||||||
use actix_web::http::Method;
|
use actix_web::http::Method;
|
||||||
|
use db::*;
|
||||||
use diesel::pg::PgConnection;
|
use diesel::pg::PgConnection;
|
||||||
use diesel::r2d2::{ConnectionManager, Pool};
|
use diesel::r2d2::{ConnectionManager, Pool};
|
||||||
use std::env;
|
use std::env;
|
||||||
use db::*;
|
use handlers::*;
|
||||||
use futures::Future;
|
|
||||||
use models::*;
|
|
||||||
|
|
||||||
/// Represents the state carried by the web server actors.
|
|
||||||
struct AppState {
|
|
||||||
/// Address of the database actor
|
|
||||||
db: Addr<Syn, DbExecutor>,
|
|
||||||
|
|
||||||
/// Compiled templates
|
|
||||||
tera: tera::Tera,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Really inefficient renderer example!
|
|
||||||
fn render_threads(tpl: &tera::Tera, threads: Vec<Thread>) -> String {
|
|
||||||
let mut ctx = tera::Context::new();
|
|
||||||
ctx.add("threads", &threads);
|
|
||||||
tpl.render("index.html", &ctx).expect("Oh no")
|
|
||||||
}
|
|
||||||
|
|
||||||
fn forum_index(state: State<AppState>) -> FutureResponse<HttpResponse> {
|
|
||||||
state.db.send(ListThreads)
|
|
||||||
.from_err()
|
|
||||||
.and_then(move |res| match res {
|
|
||||||
Ok(threads) => Ok(HttpResponse::Ok()
|
|
||||||
.content_type("text/html")
|
|
||||||
.body(render_threads(&state.tera, threads))),
|
|
||||||
Err(err) => {
|
|
||||||
error!("Error loading threads: {}", err);
|
|
||||||
Ok(HttpResponse::InternalServerError().into())
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.responder()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn render_thread(tpl: &tera::Tera, thread: Thread, posts: Vec<Post>) -> HttpResponse {
|
|
||||||
let mut ctx = tera::Context::new();
|
|
||||||
ctx.add("thread", &thread);
|
|
||||||
ctx.add("posts", &posts);
|
|
||||||
|
|
||||||
let body = tpl.render("thread.html", &ctx).expect("Oh no");
|
|
||||||
HttpResponse::Ok()
|
|
||||||
.content_type("text/html")
|
|
||||||
.body(body)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn forum_thread(state: State<AppState>, thread_id: Path<i32>)
|
|
||||||
-> FutureResponse<HttpResponse> {
|
|
||||||
let id = thread_id.into_inner();
|
|
||||||
state.db.send(GetThread(id))
|
|
||||||
.from_err()
|
|
||||||
.and_then(move |res| match res {
|
|
||||||
Ok((thread, posts)) => Ok(render_thread(&state.tera, thread, posts)),
|
|
||||||
Err(err) => {
|
|
||||||
error!("Error loading thread {}: {}", id, err);
|
|
||||||
Ok(HttpResponse::InternalServerError().into())
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.responder()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
env_logger::init();
|
env_logger::init();
|
||||||
|
|
Loading…
Reference in a new issue