new db
This commit is contained in:
parent
747a611a67
commit
f56ccf471f
4
.gitignore
vendored
4
.gitignore
vendored
@ -1,4 +1,4 @@
|
|||||||
/target
|
/target
|
||||||
**/*.rs.bk
|
**/*.rs.bk
|
||||||
|
db.dat
|
||||||
/db
|
/db
|
||||||
|
1
Cargo.lock
generated
1
Cargo.lock
generated
@ -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",
|
||||||
|
@ -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
|
||||||
|
59
src/main.rs
59
src/main.rs
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user