diff --git a/emgauwa-core/src/app_state.rs b/emgauwa-core/src/app_state.rs index a30712a..ab97265 100644 --- a/emgauwa-core/src/app_state.rs +++ b/emgauwa-core/src/app_state.rs @@ -27,25 +27,25 @@ pub struct Action { pub action: ControllerWsAction, } -pub struct AppServer { +pub struct AppState { pub pool: Pool, pub connected_controllers: HashMap)>, } -impl AppServer { - pub fn new(pool: Pool) -> AppServer { - AppServer { +impl AppState { + pub fn new(pool: Pool) -> AppState { + AppState { pool, connected_controllers: HashMap::new(), } } } -impl Actor for AppServer { +impl Actor for AppState { type Context = Context; } -impl Handler for AppServer { +impl Handler for AppState { type Result = Result<(), EmgauwaError>; fn handle(&mut self, msg: DisconnectController, _ctx: &mut Self::Context) -> Self::Result { @@ -66,7 +66,7 @@ impl Handler for AppServer { } } -impl Handler for AppServer { +impl Handler for AppState { type Result = Result<(), EmgauwaError>; fn handle(&mut self, msg: ConnectController, _ctx: &mut Self::Context) -> Self::Result { @@ -77,7 +77,7 @@ impl Handler for AppServer { } } -impl Handler for AppServer { +impl Handler for AppState { type Result = Result<(), EmgauwaError>; fn handle(&mut self, msg: Action, _ctx: &mut Self::Context) -> Self::Result { diff --git a/emgauwa-core/src/handlers/v1/controllers.rs b/emgauwa-core/src/handlers/v1/controllers.rs index bfccf4a..171cb5b 100644 --- a/emgauwa-core/src/handlers/v1/controllers.rs +++ b/emgauwa-core/src/handlers/v1/controllers.rs @@ -7,7 +7,7 @@ use emgauwa_lib::types::{ControllerUid, ControllerWsAction, RequestUpdateControl use sqlx::{Pool, Sqlite}; use crate::app_state; -use crate::app_state::AppServer; +use crate::app_state::AppState; #[get("/api/v1/controllers")] pub async fn index(pool: web::Data>) -> Result { @@ -41,7 +41,7 @@ pub async fn show( #[put("/api/v1/controllers/{controller_id}")] pub async fn update( pool: web::Data>, - app_server: web::Data>, + app_server: web::Data>, path: web::Path<(String,)>, data: web::Json, ) -> Result { @@ -73,7 +73,7 @@ pub async fn update( #[delete("/api/v1/controllers/{controller_id}")] pub async fn delete( pool: web::Data>, - app_server: web::Data>, + app_server: web::Data>, path: web::Path<(String,)>, ) -> Result { let mut pool_conn = pool.acquire().await?; diff --git a/emgauwa-core/src/handlers/v1/relays.rs b/emgauwa-core/src/handlers/v1/relays.rs index d536e97..08c728b 100644 --- a/emgauwa-core/src/handlers/v1/relays.rs +++ b/emgauwa-core/src/handlers/v1/relays.rs @@ -8,7 +8,7 @@ use emgauwa_lib::utils; use sqlx::{Pool, Sqlite}; use crate::app_state; -use crate::app_state::AppServer; +use crate::app_state::AppState; #[get("/api/v1/relays")] pub async fn index(pool: web::Data>) -> Result { @@ -84,7 +84,7 @@ pub async fn show_for_controller( #[put("/api/v1/controllers/{controller_id}/relays/{relay_num}")] pub async fn update_for_controller( pool: web::Data>, - app_server: web::Data>, + app_server: web::Data>, path: web::Path<(String, i64)>, data: web::Json, ) -> Result { diff --git a/emgauwa-core/src/handlers/v1/ws/controllers/mod.rs b/emgauwa-core/src/handlers/v1/ws/controllers/mod.rs index e48030b..0a131d6 100644 --- a/emgauwa-core/src/handlers/v1/ws/controllers/mod.rs +++ b/emgauwa-core/src/handlers/v1/ws/controllers/mod.rs @@ -13,13 +13,13 @@ use sqlx::pool::PoolConnection; use sqlx::{Pool, Sqlite}; use ws::Message; -use crate::app_state::{AppServer, DisconnectController}; +use crate::app_state::{AppState, DisconnectController}; use crate::utils::flatten_result; pub struct ControllerWs { pub pool: Pool, pub controller_uid: Option, - pub app_server: Addr, + pub app_server: Addr, pub hb: Instant, } diff --git a/emgauwa-core/src/handlers/v1/ws/mod.rs b/emgauwa-core/src/handlers/v1/ws/mod.rs index 80c3b58..dd66ad5 100644 --- a/emgauwa-core/src/handlers/v1/ws/mod.rs +++ b/emgauwa-core/src/handlers/v1/ws/mod.rs @@ -6,7 +6,7 @@ use actix_web_actors::ws; use emgauwa_lib::errors::EmgauwaError; use sqlx::{Pool, Sqlite}; -use crate::app_state::AppServer; +use crate::app_state::AppState; use crate::handlers::v1::ws::controllers::ControllerWs; pub mod controllers; @@ -14,7 +14,7 @@ pub mod controllers; #[get("/api/v1/ws/controllers")] pub async fn ws_controllers( pool: web::Data>, - app_server: web::Data>, + app_server: web::Data>, req: HttpRequest, stream: web::Payload, ) -> Result { diff --git a/emgauwa-core/src/main.rs b/emgauwa-core/src/main.rs index 45c1214..00f2561 100644 --- a/emgauwa-core/src/main.rs +++ b/emgauwa-core/src/main.rs @@ -8,7 +8,7 @@ use emgauwa_lib::db::DbController; use emgauwa_lib::errors::EmgauwaError; use emgauwa_lib::utils::init_logging; -use crate::app_state::AppServer; +use crate::app_state::AppState; use crate::utils::drop_privileges; mod app_state; @@ -33,7 +33,7 @@ async fn main() -> Result<(), std::io::Error> { .map_err(EmgauwaError::from)?; conn.close().await.map_err(EmgauwaError::from)?; - let app_server = AppServer::new(pool.clone()).start(); + let app_server = AppState::new(pool.clone()).start(); log::info!("Starting server on {}:{}", settings.host, settings.port);