reuse client object

This commit is contained in:
Aravinth Manivannan 2021-10-31 22:26:22 +05:30
parent a1395234e6
commit 63bd10688c
Signed by untrusted user: realaravinth
GPG key ID: AD9F0F08E855ED88
2 changed files with 39 additions and 0 deletions

33
src/data.rs Normal file
View file

@ -0,0 +1,33 @@
/*
* Copyright (C) 2021 Aravinth Manivannan <realaravinth@batsense.net>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
use actix_web::web;
use reqwest::Client;
#[derive(Clone)]
pub struct Data {
pub client: Client,
}
pub type AppData = web::Data<Data>;
impl Data {
pub fn new() -> AppData {
AppData::new(Self {
client: Client::new(),
})
}
}

View file

@ -23,11 +23,14 @@ use actix_web::{
use lazy_static::lazy_static; use lazy_static::lazy_static;
use log::info; use log::info;
mod data;
mod meta; mod meta;
mod proxy; mod proxy;
mod routes; mod routes;
mod settings; mod settings;
pub use data::AppData;
pub use data::Data;
pub use routes::ROUTES as V1_API_ROUTES; pub use routes::ROUTES as V1_API_ROUTES;
pub use settings::Settings; pub use settings::Settings;
@ -57,6 +60,8 @@ async fn main() -> std::io::Result<()> {
println!("Starting server on: http://{}", SETTINGS.server.get_ip()); println!("Starting server on: http://{}", SETTINGS.server.get_ip());
let data = Data::new();
HttpServer::new(move || { HttpServer::new(move || {
App::new() App::new()
.wrap(actix_middleware::Logger::default()) .wrap(actix_middleware::Logger::default())
@ -69,6 +74,7 @@ async fn main() -> std::io::Result<()> {
.wrap(actix_middleware::NormalizePath::new( .wrap(actix_middleware::NormalizePath::new(
actix_middleware::TrailingSlash::Trim, actix_middleware::TrailingSlash::Trim,
)) ))
.app_data(data.clone())
.configure(services) .configure(services)
}) })
.workers(SETTINGS.server.workers.unwrap_or_else(num_cpus::get)) .workers(SETTINGS.server.workers.unwrap_or_else(num_cpus::get))