from actix to rocket

Cargo.toml

@@ -6,19 +6,26 @@ edition = "2018"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
+
 [dependencies]
-actix-files="0.1"
-actix-web = "1.0"
+rand="0.7"
 serde="1.0"
 serde_derive="1.0"
 serde_json="1.0"
 ledb="0.2"
 ledb-derive="0.2"
 ledb-types="0.2"
-lmdb-zero="0.4"
-rand="0.7"
-env_logger = "0.6"
 tera = "0.11"
-# actix="0.7"
-# tokio="0.1"
-# futures="0.1"
+rocket="0.4"
+
+# lmdb-zero="0.4"
+
+[dependencies.rocket_contrib]
+version = "0.4"
+default-features = false
+features = ["serve", "tera_templates"]
+
+[profile.release]
+opt-level = 3
+debug = false
+lto = true

src/main.rs

@@ -1,45 +1,28 @@
-// extern crate actix;
-extern crate actix_files;
-extern crate actix_web;
+#![feature(proc_macro_hygiene, decl_macro)]
+
 extern crate serde;
 #[macro_use]
 extern crate serde_derive;
-#[macro_use]
 extern crate serde_json;
-#[macro_use]
 extern crate ledb;
 #[macro_use]
 extern crate ledb_derive;
-extern crate env_logger;
 extern crate ledb_types;
-#[macro_use]
-extern crate tera;
 
-use tera::Context;
+//#[macro_use]
+//extern crate tera;
 
-// extern crate futures;
+//use tera::Context;
 
-// extern crate tokio;
+#[macro_use] extern crate rocket;
+extern crate rocket_contrib;
 
-use actix_web::{
-    error, guard, http::header, http::Method, middleware::Logger, middleware::NormalizePath, web,
-    App, Error, HttpRequest, HttpResponse, HttpServer, Responder, Result,
-};
-use std::cell::Cell;
+use rocket::{Rocket, State};
+use rocket::response::Redirect;
+use rocket_contrib::serve::StaticFiles;
+use rocket_contrib::templates::Template;
 
 use rand::seq::IteratorRandom;
-use std::time::Instant;
-
-use std::{fs, io};
-// use tokio::spawn;
-// use futures::{Future};
-
-// use actix::Actor;
-// use actix::System;
-
-//use crate::tokio::prelude::Future;
-
-//use ledb_actix::{Document, Options, Storage, StorageAddrExt};
 
 use ledb::{Options, Storage};
 
@@ -118,32 +101,29 @@ struct Question {
 
 struct AppState {
     storage: Storage,
-    template: tera::Tera,
 }
 
-fn get_question(storage: &Storage, id: u32) -> Result<Option<Question>, Error> {
+fn get_question(storage: &Storage, id: u32) -> Result<Option<Question>, ()> {
     if 0 == id {
         return Ok(None);
     }
 
-    let collection = storage.collection("questions").unwrap();
-    let last_id = collection.last_id().unwrap();
+    let collection = storage.collection("questions").expect("collection \"questions\"");
+    let last_id = collection.last_id().expect("\"questions\" last_id");
 
     if id > last_id {
-        Err(Error::from(()))
+        Err(())
     } else {
         let question = collection.get::<Question>(id);
         if question.is_err() {
-            Err(Error::from(()))
+            Err(())
         } else {
             Ok(question.unwrap())
         }
     }
 }
 
-fn show_question_details(template_file: &str, data: web::Data<AppState>, id: web::Path<u32>) -> Result<HttpResponse, Error> {
-    let id = id.into_inner();
-
+fn show_question_details(template_name: &'static str, data: &AppState, id: u32) -> Result<Template, Redirect>  {
     let question = get_question(&data.storage, id);
 
     if question.is_ok() {
@@ -151,85 +131,67 @@ fn show_question_details(template_file: &str, data: web::Data<AppState>, id: web
 
         if question.is_some() {
             let question = question.unwrap();
-            let body = data.template.render(template_file, &question).unwrap();
-            Ok(HttpResponse::Ok().content_type("text/html").body(body))
+            Ok(Template::render(template_name, &question))
         } else {
-            Ok(HttpResponse::Found()
-                .header(header::LOCATION, "/q/")
-                .finish())
+            Err(Redirect::permanent("/q/"))
         }
     } else {
-        let context = Context::new();
-        Ok(HttpResponse::with_body(
-            actix_web::http::StatusCode::NOT_FOUND,
-            actix_web::dev::Body::from(data.template.render("404.html", &context).unwrap()),
-        ))
+        use std::collections::HashMap;
+        let context: HashMap<String, String> = HashMap::new();
+        Ok(Template::render("404", &context))
     }
 }
 
-fn show_question(data: web::Data<AppState>, id: web::Path<u32>) -> Result<HttpResponse, Error> {
-    show_question_details("question.html", data, id)
+#[get("/q/<id>")]
+fn show_question(data: State<AppState>, id: u32) -> Result<Template, Redirect> {
+    show_question_details("question", data.inner(), id)
 }
 
-fn show_answer(data: web::Data<AppState>, id: web::Path<u32>) -> Result<HttpResponse, Error> {
-    show_question_details("answer.html", data, id)
+#[get("/q/<id>/a")]
+fn show_answer(data: State<AppState>, id: u32) -> Result<Template, Redirect> {
+    show_question_details("answer", data.inner(), id)
 }
 
-fn index(data: web::Data<AppState>, req: HttpRequest) -> Result<HttpResponse, Error> {
-    let collection = data.storage.collection("questions").unwrap();
+#[get("/")]
+fn index(data: State<AppState>) -> Redirect {
     let mut rng = rand::thread_rng();
-    let last_id = collection.last_id().unwrap();
+    let collection = data.storage.collection("questions").expect("collection \"questions\"");
+    let last_id = collection.last_id().expect("\"questions\" last_id");
     let id = (1..(last_id + 1)).choose(&mut rng).unwrap();
 
-    let url = req.url_for("question", &[format!("{}", id)])?;
-
-    Ok(HttpResponse::Found()
-        .header(header::LOCATION, url.as_str())
-        .finish())
+    Redirect::temporary(format!("/q/{}", id))
 }
 
-fn main() {
-    std::env::set_var("RUST_LOG", "actix_web=info");
-    env_logger::init();
+#[catch(404)]
+fn not_found(_req: &rocket::Request) -> Template {
+    use std::collections::HashMap;
+    let context: HashMap<String, String> = HashMap::new();
+    Template::render("404", &context)
+}
 
-    let options: Options = serde_json::from_value(json!({
+fn rocket() -> Rocket {
+    let options: Options = serde_json::from_value(serde_json::json!({
     "read_only": true,
     "no_lock": true,
     }))
     .unwrap();
 
-    let storage = Storage::new("db", options).unwrap();
-
-    HttpServer::new(move || {
-        let data = AppState {
-            storage: storage.clone(),
-            template: compile_templates!("./templates/**/*"),
+    let state = AppState {
+        storage: Storage::new("db", options).expect("open db")
     };
-        App::new()
-            .wrap(Logger::default())
-            .data(data)
-            .route("/q", web::to(index))
-            .service(
-                web::scope("/q")
-                    .service(actix_files::Files::new("/static", "./static"))
-                    .service(
-                        web::resource("/{id}")
-                            .name("question") // <- set resource name, then it could be used in `url_for`
-                            .guard(guard::Get())
-                            .to(show_question),
-                    )
-                    .service(
-                        web::resource("/{id}/a/")
-                            .name("answer") // <- set resource name, then it could be used in `url_for`
-                            .guard(guard::Get())
-                            .to(show_answer),
-                    )
-                    .route("/", web::to(index))
-            )
-            .route("/", web::to(index))
-    })
-    .bind("127.0.0.1:8088")
-    .unwrap()
-    .run()
-    .unwrap();
+
+    let collection = state.storage.collection("questions").expect("collection \"questions\"");
+    collection.last_id().expect("\"questions\" last_id");
+
+    rocket::ignite()
+        .manage(state)
+        .register(catchers![not_found])
+        .mount("/", routes![index, show_question, show_answer])
+        .mount("/q", routes![index])
+        .mount("/q/static", StaticFiles::from("static/"))
+        .attach(Template::fairing())
+}
+
+fn main() {
+    rocket().launch();
 }

templates/404.html.tera

@@ -1,4 +1,4 @@
-{% extends "base.html" %}
+{% extends "base" %}
 {% block title %}404{% endblock title %}
 {% block content %}
   <h1>404 - Could not find that page</h1>

templates/answer.html.tera

@@ -1,4 +1,4 @@
-{% extends "base.html" %} {% block title %} Ответ {% endblock title %}
+{% extends "base" %} {% block title %} Ответ {% endblock title %}
 
 {% block content %}
 <!-- <h1>{{ id }}</h1> -->

templates/base.html.tera

@@ -7,7 +7,7 @@
   <title>{% block title %}{% endblock title %}</title>
 </head>
 <body>
-    {% include "nav.html" %}
+    {% include "nav" %}
     <div id="content" class="container">{% block content %}{% endblock content %}</div>
 </body>
 </html>

templates/question.html.tera

@@ -1,4 +1,4 @@
-{% extends "base.html" %} {% block title %} Вопрос {% endblock title %} 
+{% extends "base" %} {% block title %} Вопрос {% endblock title %} 
 
 {% block content %}
 <!-- <h1>{{ id }}</h1> -->