from actix to rocket

This commit is contained in:
Dmitry Belyaev 2019-08-05 17:06:27 +03:00
parent a7b1233a0c
commit 2a6cd616be
Signed by: b4tman
GPG Key ID: 41A00BF15EA7E5F3
8 changed files with 359 additions and 1299 deletions

1473
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@ -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

View File

@ -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();
}

View File

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

View File

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

View File

@ -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>

View File

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