share database reader instance

.gitignore

@@ -1,4 +1,4 @@
 /target
 **/*.rs.bk
-
+db.dat
-/db
+/db

Cargo.toml

@@ -1,23 +1,20 @@
 [package]
 name = "qchgk_web"
-version = "0.1.0"
+version = "0.1.1"
 authors = ["Dmitry <b4tm4n@mail.ru>"]
-edition = "2018"
+edition = "2021"
+license = "MIT"
+readme = "README.md"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 
 [dependencies]
-rand="0.7"
+rand="0.8"
 serde="1.0"
-serde_derive="1.0"
 serde_json="1.0"
-ledb="0.3"
-ledb-derive="0.3"
-ledb-types="0.3"
 rocket="0.4"
-
+chgk_ledb_lib = {git = "https://gitea.b4tman.ru/b4tman/chgk_ledb.git", rev="8120a996a3", package="chgk_ledb_lib"}
-# lmdb-zero="0.4"
 
 [dependencies.rocket_contrib]
 version = "0.4"
@@ -28,3 +25,4 @@ features = ["serve", "tera_templates"]
 opt-level = 3
 debug = false
 lto = true
+strip = true

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2022 Dmitry Belyaev
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -0,0 +1,10 @@
+# qchgk_web
+
+Простой веб интерфейс к базе вопросов ЧГК
+
+## Ссылки
+
+- Источник вопросов: http://db.chgk.info
+- Копия файлов базы вопросов: https://gitlab.com/b4tman/db_chgk
+- Конвертор в JSON: https://gitea.b4tman.ru/b4tman/chgk_txt2json
+- Загрузчик в формат базы для этой штуки: https://gitea.b4tman.ru/b4tman/chgk_ledb

src/main.rs

@@ -1,18 +1,7 @@
 #![feature(proc_macro_hygiene, decl_macro)]
 
 extern crate serde;
-#[macro_use]
-extern crate serde_derive;
-extern crate ledb;
 extern crate serde_json;
-#[macro_use]
-extern crate ledb_derive;
-extern crate ledb_types;
-
-//#[macro_use]
-//extern crate tera;
-
-//use tera::Context;
 
 #[macro_use]
 extern crate rocket;
@@ -26,160 +15,92 @@ use rocket_contrib::templates::Template;
 use rand::distributions::Uniform;
 use rand::Rng;
 
-use ledb::{Options, Storage};
+use std::sync::Arc;
 
-#[derive(Debug, Default, Clone, Serialize, Deserialize, Document)]
+use chgk_ledb_lib::db;
-struct BatchInfo {
+use chgk_ledb_lib::questions::Question;
-    #[document(primary)]
+
-    #[serde(default)]
+const DB_FILENAME: &str = "db.dat";
-    filename: String,
+
-    #[serde(default)]
+trait ErrorEmpty {
-    description: String,
+    type Output;
-    #[serde(default)]
+    fn err_empty(self) -> Result<Self::Output, ()>;
-    author: String,
-    #[serde(default)]
-    comment: String,
-    #[serde(default)]
-    url: String,
-    #[serde(default)]
-    date: String,
-    #[serde(default)]
-    processed_by: String,
-    #[serde(default)]
-    redacted_by: String,
-    #[serde(default)]
-    copyright: String,
-    #[serde(default)]
-    theme: String,
-    #[serde(default)]
-    kind: String,
-    #[serde(default)]
-    source: String,
-    #[serde(default)]
-    rating: String,
 }
 
-#[derive(Debug, Default, Clone, Serialize, Deserialize, Document)]
+impl<T, E> ErrorEmpty for Result<T, E> {
-struct Question {
+    type Output = T;
-    #[document(primary)]
+    fn err_empty(self) -> Result<Self::Output, ()> {
-    #[serde(default)]
+        self.map_err(|_| ())
-    num: u32,
-    #[document(index)]
-    id: String,
-
-    description: String,
-    answer: String,
-
-    #[serde(default)]
-    author: String,
-    #[serde(default)]
-    comment: String,
-    #[serde(default)]
-    comment1: String,
-    #[serde(default)]
-    tour: String,
-    #[serde(default)]
-    url: String,
-    #[serde(default)]
-    date: String,
-    #[serde(default)]
-    processed_by: String,
-    #[serde(default)]
-    redacted_by: String,
-    #[serde(default)]
-    copyright: String,
-    #[serde(default)]
-    theme: String,
-    #[serde(default)]
-    kind: String,
-    #[serde(default)]
-    source: String,
-    #[serde(default)]
-    rating: String,
-    #[document(nested)]
-    #[serde(default)]
-    batch_info: BatchInfo,
-}
-
-struct AppState {
-    storage: Storage,
-    database_distribution: Uniform<u32>,
-}
-
-fn get_database_distribution(storage: &Storage) -> Uniform<u32> {
-    let collection = storage
-        .collection("questions")
-        .expect("collection \"questions\"");
-    let last_id = collection.last_id().expect("\"questions\" last_id");
-
-    rand::distributions::Uniform::new_inclusive(1u32, last_id)
-}
-
-fn random_question_id(database_distribution: &Uniform<u32>) -> u32 {
-    let mut rng = rand::thread_rng();
-    rng.sample(database_distribution)
-}
-
-fn get_question(storage: &Storage, id: u32) -> Result<Option<Question>, ()> {
-    if 0 == id {
-        return Ok(None);
     }
+}
 
-    let collection = storage
+type DataBaseInner = db::Reader<Question>;
-        .collection("questions")
+type DataBase = Arc<DataBaseInner>;
-        .expect("collection \"questions\"");
+struct AppState{
-    let last_id = collection.last_id().expect("\"questions\" last_id");
+    db: DataBase,
+    database_distribution: Uniform<usize>,
+}
 
-    if id > last_id {
+impl From<DataBaseInner> for AppState {
-        Err(())
+    fn from(db: DataBaseInner) -> Self {
-    } else {
+        let last_id = db.len();
-        let question = collection.get::<Question>(id);
+        let database_distribution = rand::distributions::Uniform::new_inclusive(1usize, last_id);
-        if question.is_err() {
+        let db = Arc::new(db);
-            Err(())
+
-        } else {
+        Self {
-            Ok(question.unwrap())
+            db,
+            database_distribution
         }
     }
 }
 
-fn show_question_details(
+fn random_question_id(database_distribution: &Uniform<usize>) -> usize {
-    template_name: &'static str,
+    let mut rng = rand::thread_rng();
-    data: &AppState,
+    rng.sample(database_distribution)
-    id: u32,
+}
-) -> Result<Template, Redirect> {
-    let question = get_question(&data.storage, id);
 
-    if question.is_ok() {
+fn get_question(db: &DataBase, id: usize) -> Result<Question, ()> {
-        let question = question.unwrap();
+    db.get(id - 1).err_empty()
+}
 
-        if question.is_some() {
+fn show_question_details(template_name: &'static str, data: &AppState, id: usize) -> Template {
-            let question = question.unwrap();
+    match get_question(&data.db, id) {
+        Ok(question) => {
             let mut context = serde_json::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");
             }
-            Ok(Template::render(template_name, &context))
+            Template::render(template_name, &context)
-        } else {
+        }
-            Err(Redirect::permanent("/q/"))
+        Err(_) => {
+            use std::collections::HashMap;
+            let context: HashMap<String, String> = HashMap::new();
+            Template::render("404", context)
         }
-    } else {
-        use std::collections::HashMap;
-        let context: HashMap<String, String> = HashMap::new();
-        Ok(Template::render("404", &context))
     }
 }
 
 #[get("/q/<id>")]
-fn show_question(data: State<AppState>, id: u32) -> Result<Template, Redirect> {
+fn show_question(data: State<AppState>, id: usize) -> Template {
     show_question_details("question", data.inner(), id)
 }
 
 #[get("/q/<id>/a")]
-fn show_answer(data: State<AppState>, id: u32) -> Result<Template, Redirect> {
+fn show_answer(data: State<AppState>, id: usize) -> Template {
     show_question_details("answer", data.inner(), id)
 }
 
+#[get("/q/0")]
+fn question0() -> Redirect {
+    Redirect::to("/")
+}
+
+#[get("/q/0/a")]
+fn answer0() -> Redirect {
+    Redirect::to("/")
+}
+
 #[get("/")]
 fn index(data: State<AppState>) -> Redirect {
     let id = random_question_id(&data.database_distribution);
@@ -190,28 +111,19 @@ fn index(data: State<AppState>) -> Redirect {
 fn not_found(_req: &rocket::Request) -> Template {
     use std::collections::HashMap;
     let context: HashMap<String, String> = HashMap::new();
-    Template::render("404", &context)
+    Template::render("404", context)
 }
 
 fn rocket() -> Rocket {
-    let options: Options = serde_json::from_value(serde_json::json!({
+    let state: AppState = db::Reader::new(DB_FILENAME, 2048).expect("open db").into();
-    "read_only": true,
-    "no_lock": true,
-    }))
-    .unwrap();
-
-    let mut state = AppState {
-        storage: Storage::new("db", options).expect("open db"),
-        database_distribution: Uniform::new(1, 3),
-    };
-    state.database_distribution = get_database_distribution(&state.storage);
-
-    let state = state;
 
     rocket::ignite()
         .manage(state)
         .register(catchers![not_found])
-        .mount("/", routes![index, show_question, show_answer])
+        .mount(
+            "/",
+            routes![index, show_question, show_answer, question0, answer0],
+        )
         .mount("/q", routes![index])
         .mount("/q/static", StaticFiles::from("static/"))
         .attach(Template::fairing())

templates/question.html.tera

@@ -11,9 +11,8 @@
 </div></div>
 <br/>
 <br/>
-<br/><details>
+<br/>
-    <summary>Ответ</summary><br/>
+<div class="justify-content-center text-center">
-    <a class="btn btn-dark shadow" href="/q/{{ num }}/a/">Показать ответ</a>
+<a class="btn btn-dark shadow" href="/q/{{ num }}/a/">Показать ответ</a>
-</details>
+</div>
-
 {% endblock content %}