53 lines
1.2 KiB
Rust
53 lines
1.2 KiB
Rust
use std::time::Instant;
|
|
|
|
use actix::Addr;
|
|
use actix_web::{get, web, HttpRequest, HttpResponse};
|
|
use actix_web_actors::ws;
|
|
use emgauwa_lib::errors::EmgauwaError;
|
|
use sqlx::{Pool, Sqlite};
|
|
|
|
use crate::app_state::AppState;
|
|
use crate::handlers::v1::ws::controllers::ControllersWs;
|
|
use crate::handlers::v1::ws::relays::RelaysWs;
|
|
|
|
pub mod controllers;
|
|
pub mod relays;
|
|
|
|
#[get("/ws/controllers")]
|
|
pub async fn ws_controllers(
|
|
pool: web::Data<Pool<Sqlite>>,
|
|
app_state: web::Data<Addr<AppState>>,
|
|
req: HttpRequest,
|
|
stream: web::Payload,
|
|
) -> Result<HttpResponse, EmgauwaError> {
|
|
let resp = ws::start(
|
|
ControllersWs {
|
|
pool: pool.get_ref().clone(),
|
|
controller_uid: None,
|
|
app_state: app_state.get_ref().clone(),
|
|
hb: Instant::now(),
|
|
},
|
|
&req,
|
|
stream,
|
|
)
|
|
.map_err(|_| EmgauwaError::Internal(String::from("error starting websocket")));
|
|
resp
|
|
}
|
|
|
|
#[get("/ws/relays")]
|
|
pub async fn ws_relays(
|
|
app_state: web::Data<Addr<AppState>>,
|
|
req: HttpRequest,
|
|
stream: web::Payload,
|
|
) -> Result<HttpResponse, EmgauwaError> {
|
|
let resp = ws::start(
|
|
RelaysWs {
|
|
app_state: app_state.get_ref().clone(),
|
|
hb: Instant::now(),
|
|
},
|
|
&req,
|
|
stream,
|
|
)
|
|
.map_err(|_| EmgauwaError::Internal(String::from("error starting websocket")));
|
|
resp
|
|
}
|