Merge branch 'master' into rocket

.drone.yml

@@ -1,9 +1,10 @@
 kind: pipeline
+type: docker
 name: default
 
 steps:
 - name: build
-  image: rustlang/rust:nightly-alpine
+  image: rust:1-alpine
   commands:
   - apk add --no-cache musl-dev
   - cargo fetch

Cargo.toml

@@ -11,18 +11,14 @@ readme = "README.md"
 
 [dependencies]
 rand="0.8"
-serde="1.0"
-serde_json="1.0"
-rocket="0.4"
-chgk_ledb_lib = {git = "https://gitea.b4tman.ru/b4tman/chgk_ledb.git", rev="8120a996a3", package="chgk_ledb_lib"}
-
-[dependencies.rocket_contrib]
-version = "0.4"
-default-features = false
-features = ["serve", "tera_templates"]
+rocket = { version = "=0.5.0-rc.3", features = ["json"] }
+rocket_dyn_templates = { version = "=0.1.0-rc.3", features = ["tera"] }
+chgk_ledb_lib = {git = "https://gitea.b4tman.ru/b4tman/chgk_ledb.git", rev="699478f85e", package="chgk_ledb_lib", features=["async"]}
+mini-moka = "0.10.0"
 
 [profile.release]
 opt-level = 3
 debug = false
 lto = true
 strip = true
+

src/main.rs

@@ -1,25 +1,24 @@
-#![feature(proc_macro_hygiene, decl_macro)]
-
-extern crate serde;
-extern crate serde_json;
-
 #[macro_use]
 extern crate rocket;
-extern crate rocket_contrib;
 
+use rocket::fs::FileServer;
 use rocket::response::Redirect;
-use rocket::{Rocket, State};
-use rocket_contrib::serve::StaticFiles;
-use rocket_contrib::templates::Template;
+use rocket::State;
+use rocket_dyn_templates::tera;
+use rocket_dyn_templates::Template;
 
 use rand::distributions::Uniform;
 use rand::Rng;
 
+use std::ops::Deref;
 use std::sync::Arc;
 
-use chgk_ledb_lib::db;
+use chgk_ledb_lib::async_db;
 use chgk_ledb_lib::questions::Question;
 
+use mini_moka::sync::Cache;
+use std::time::Duration;
+
 const DB_FILENAME: &str = "db.dat";
 
 trait ErrorEmpty {
@@ -34,7 +33,25 @@ impl<T, E> ErrorEmpty for Result<T, E> {
     }
 }
 
-type DataBaseInner = db::Reader<Question>;
+#[derive(Clone)]
+struct ArcTemplateData {
+    value: Arc<tera::Value>,
+}
+
+impl ArcTemplateData {
+    fn new(value: tera::Value) -> ArcTemplateData {
+        ArcTemplateData {
+            value: Arc::new(value),
+        }
+    }
+    fn render(&self, name: &'static str) -> Template {
+        Template::render(name, self.value.deref())
+    }
+}
+
+type TemplateCache = mini_moka::sync::Cache<usize, ArcTemplateData>;
+
+type DataBaseInner = async_db::Reader<Question>;
 type DataBase = Arc<DataBaseInner>;
 struct AppState {
     db: DataBase,
@@ -49,7 +66,7 @@ impl From<DataBaseInner> for AppState {
 
         Self {
             db,
-            database_distribution
+            database_distribution,
         }
     }
 }
@@ -59,19 +76,33 @@ fn random_question_id(database_distribution: &Uniform<usize>) -> usize {
     rng.sample(database_distribution)
 }
 
-fn get_question(db: &DataBase, id: usize) -> Result<Question, ()> {
-    db.get(id - 1).err_empty()
+async fn get_question(db: &DataBase, id: usize) -> Result<Question, ()> {
+    db.get(id - 1).await.err_empty()
 }
 
-fn show_question_details(template_name: &'static str, data: &AppState, id: usize) -> Template {
-    match get_question(&data.db, id) {
+async fn show_question_details(
+    template_name: &'static str,
+    data: &AppState,
+    cache: &TemplateCache,
+    id: usize,
+) -> Template {
+    if let Some(value) = cache.get(&id) {
+        return value.render(template_name);
+    }
+
+    match get_question(&data.db, id).await {
         Ok(question) => {
-            let mut context = serde_json::to_value(question).expect("question serialize");
+            let mut context = tera::to_value(question).expect("question serialize");
             if context.is_object() {
                 let next_id = random_question_id(&data.database_distribution);
-                context["next"] = serde_json::to_value(next_id).expect("question id serialize");
+                context["next"] = tera::to_value(next_id).expect("question id serialize");
             }
-            Template::render(template_name, &context)
+
+            let value = ArcTemplateData::new(context);
+            let result = value.render(template_name);
+            cache.insert(id, value);
+
+            result
         }
         Err(_) => {
             use std::collections::HashMap;
@@ -82,13 +113,17 @@ fn show_question_details(template_name: &'static str, data: &AppState, id: usize
 }
 
 #[get("/q/<id>")]
-fn show_question(data: State<AppState>, id: usize) -> Template {
-    show_question_details("question", data.inner(), id)
+async fn show_question(
+    data: &State<AppState>,
+    cache: &State<TemplateCache>,
+    id: usize,
+) -> Template {
+    show_question_details("question", data.inner(), cache.inner(), id).await
 }
 
 #[get("/q/<id>/a")]
-fn show_answer(data: State<AppState>, id: usize) -> Template {
-    show_question_details("answer", data.inner(), id)
+async fn show_answer(data: &State<AppState>, cache: &State<TemplateCache>, id: usize) -> Template {
+    show_question_details("answer", data.inner(), cache.inner(), id).await
 }
 
 #[get("/q/0")]
@@ -102,7 +137,7 @@ fn answer0() -> Redirect {
 }
 
 #[get("/")]
-fn index(data: State<AppState>) -> Redirect {
+fn index(data: &State<AppState>) -> Redirect {
     let id = random_question_id(&data.database_distribution);
     Redirect::temporary(format!("/q/{}", id))
 }
@@ -114,21 +149,26 @@ fn not_found(_req: &rocket::Request) -> Template {
     Template::render("404", context)
 }
 
-fn rocket() -> Rocket {
-    let state: AppState = db::Reader::new(DB_FILENAME, 2048).expect("open db").into();
+#[launch]
+async fn rocket() -> _ {
+    let state: AppState = async_db::Reader::new(DB_FILENAME)
+        .await
+        .expect("open db")
+        .into();
+    let cache: TemplateCache = Cache::builder()
+        .time_to_idle(Duration::from_secs(15 * 60))
+        .max_capacity(300)
+        .build();
 
-    rocket::ignite()
+    rocket::build()
         .manage(state)
-        .register(catchers![not_found])
+        .manage(cache)
+        .register("/", catchers![not_found])
         .mount(
             "/",
             routes![index, show_question, show_answer, question0, answer0],
         )
         .mount("/q", routes![index])
-        .mount("/q/static", StaticFiles::from("static/"))
+        .mount("/q/static", FileServer::from("static/"))
         .attach(Template::fairing())
 }
-
-fn main() {
-    rocket().launch();
-}