This commit is contained in:
Dmitry Belyaev 2023-01-04 01:12:03 +03:00
parent 747a611a67
commit f56ccf471f
Signed by: b4tman
GPG Key ID: 41A00BF15EA7E5F3
4 changed files with 24 additions and 43 deletions

4
.gitignore vendored
View File

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

1
Cargo.lock generated
View File

@ -1279,7 +1279,6 @@ name = "qchgk_web"
version = "0.1.1" version = "0.1.1"
dependencies = [ dependencies = [
"chgk_ledb_lib", "chgk_ledb_lib",
"ledb",
"rand", "rand",
"rocket", "rocket",
"rocket_contrib", "rocket_contrib",

View File

@ -13,12 +13,9 @@ readme = "README.md"
rand="0.8" rand="0.8"
serde="1.0" serde="1.0"
serde_json="1.0" serde_json="1.0"
ledb = {git = "https://github.com/b4tman/ledb.git", rev="a646b90e", package="ledb"}
rocket="0.4" rocket="0.4"
chgk_ledb_lib = {git = "https://gitea.b4tman.ru/b4tman/chgk_ledb.git", rev="e521e39f5e", package="chgk_ledb_lib"} chgk_ledb_lib = {git = "https://gitea.b4tman.ru/b4tman/chgk_ledb.git", rev="e521e39f5e", package="chgk_ledb_lib"}
# lmdb-zero="0.4"
[dependencies.rocket_contrib] [dependencies.rocket_contrib]
version = "0.4" version = "0.4"
default-features = false default-features = false

View File

@ -1,6 +1,5 @@
#![feature(proc_macro_hygiene, decl_macro)] #![feature(proc_macro_hygiene, decl_macro)]
extern crate ledb;
extern crate serde; extern crate serde;
extern crate serde_json; extern crate serde_json;
@ -16,10 +15,11 @@ use rocket_contrib::templates::Template;
use rand::distributions::Uniform; use rand::distributions::Uniform;
use rand::Rng; use rand::Rng;
use ledb::{Options, Storage}; use chgk_ledb_lib::db;
use chgk_ledb_lib::questions::Question; use chgk_ledb_lib::questions::Question;
const DB_FILENAME: &str = "db.dat";
trait ErrorEmpty { trait ErrorEmpty {
type Output; type Output;
fn err_empty(self) -> Result<Self::Output, ()>; fn err_empty(self) -> Result<Self::Output, ()>;
@ -32,36 +32,31 @@ impl<T, E> ErrorEmpty for Result<T, E> {
} }
} }
struct AppState { struct AppState{
storage: Storage, database_distribution: Uniform<usize>,
database_distribution: Uniform<u32>,
} }
fn get_database_distribution(storage: &Storage) -> Uniform<u32> { fn open_db() -> db::Reader<Question> {
let collection = storage db::Reader::new(DB_FILENAME, 2048).expect("new db reader")
.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 { fn get_database_distribution() -> Uniform<usize> {
let last_id = open_db().len();
rand::distributions::Uniform::new_inclusive(1usize, last_id)
}
fn random_question_id(database_distribution: &Uniform<usize>) -> usize {
let mut rng = rand::thread_rng(); let mut rng = rand::thread_rng();
rng.sample(database_distribution) rng.sample(database_distribution)
} }
fn get_question(storage: &Storage, id: u32) -> Result<Question, ()> { fn get_question(id: usize) -> Result<Question, ()> {
storage open_db().get(id - 1).err_empty()
.collection("questions")
.err_empty()?
.get::<Question>(id)
.err_empty()?
.ok_or(())
} }
fn show_question_details(template_name: &'static str, data: &AppState, id: u32) -> Template { fn show_question_details(template_name: &'static str, data: &AppState, id: usize) -> Template {
match get_question(&data.storage, id) { match get_question(id) {
Ok(question) => { Ok(question) => {
let mut context = serde_json::to_value(question).expect("question serialize"); let mut context = serde_json::to_value(question).expect("question serialize");
if context.is_object() { if context.is_object() {
@ -79,12 +74,12 @@ fn show_question_details(template_name: &'static str, data: &AppState, id: u32)
} }
#[get("/q/<id>")] #[get("/q/<id>")]
fn show_question(data: State<AppState>, id: u32) -> Template { fn show_question(data: State<AppState>, id: usize) -> Template {
show_question_details("question", data.inner(), id) show_question_details("question", data.inner(), id)
} }
#[get("/q/<id>/a")] #[get("/q/<id>/a")]
fn show_answer(data: State<AppState>, id: u32) -> Template { fn show_answer(data: State<AppState>, id: usize) -> Template {
show_question_details("answer", data.inner(), id) show_question_details("answer", data.inner(), id)
} }
@ -112,19 +107,9 @@ fn not_found(_req: &rocket::Request) -> Template {
} }
fn rocket() -> Rocket { fn rocket() -> Rocket {
let options: Options = serde_json::from_value(serde_json::json!({ let state = AppState {
"read_only": true, database_distribution: get_database_distribution(),
"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() rocket::ignite()
.manage(state) .manage(state)