1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#![deny(clippy::all)]
mod config;
mod handlers;
mod stats;
use std::env;
use ::config::Config;
use actix_governor::{Governor, GovernorConfigBuilder};
use actix_web::{middleware::Logger, web, App, HttpServer};
use actix_web_lab::web as web_lab;
use dotenv::dotenv;
use handlers::index_page;
use tokio_postgres::NoTls;
use crate::config::ServerConfig;
use crate::handlers::status_get_api;
use stats::Stats;
fn scoped_config(cfg: &mut web::ServiceConfig) {
cfg.service(index_page);
}
fn api_scoped_config(cfg: &mut web::ServiceConfig) {
cfg.service(status_get_api);
}
#[actix_web::main]
async fn main() -> std::io::Result<()> {
std::env::set_var("RUST_LOG", "debug");
std::env::set_var("RUST_BACKTRACE", "1");
env_logger::init();
dotenv().ok();
let config_ = Config::builder()
.add_source(::config::Environment::default().separator("__"))
.build()
.unwrap();
let config: ServerConfig = config_.try_deserialize().unwrap();
let pool = config.pg.create_pool(None, NoTls).unwrap();
let governor_conf = GovernorConfigBuilder::default()
.per_second(2)
.burst_size(10)
.finish()
.unwrap();
let base_addr = env::var("BASE_ADDR").expect("BASE_ADDR not set");
let server = HttpServer::new(move || {
let logger = Logger::default();
App::new()
.wrap(logger)
.wrap(Governor::new(&governor_conf))
.app_data(web::Data::new(pool.clone()))
.service(web::scope("/api").configure(api_scoped_config))
.service(web::scope("/stats").configure(scoped_config))
.service(web_lab::Redirect::new(
"/",
format!("{}{}", base_addr, "/stats"),
))
})
.bind(config.server_addr.clone())?
.run();
println!("Server running at http://{}/", config.server_addr);
server.await
}