49 lines
1 KiB
Rust
49 lines
1 KiB
Rust
|
use actix::{Actor, Context, Handler, Message};
|
||
|
|
use emgauwa_lib::errors::EmgauwaError;
|
||
|
|
use emgauwa_lib::models::Controller;
|
||
|
|
use futures::executor::block_on;
|
||
|
|
use sqlx::{Pool, Sqlite};
|
||
|
|
|
||
|
|
#[derive(Message)]
|
||
|
|
#[rtype(result = "Result<(), EmgauwaError>")]
|
||
|
|
pub struct Reload {}
|
||
|
|
|
||
|
|
#[derive(Message)]
|
||
|
|
#[rtype(result = "Controller")]
|
||
|
|
pub struct GetThis {}
|
||
|
|
|
||
|
|
pub struct AppState {
|
||
|
|
pub pool: Pool<Sqlite>,
|
||
|
|
pub this: Controller,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl AppState {
|
||
|
|
pub fn new(pool: Pool<Sqlite>, this: Controller) -> AppState {
|
||
|
|
AppState { pool, this }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Actor for AppState {
|
||
|
|
type Context = Context<Self>;
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Handler<Reload> for AppState {
|
||
|
|
type Result = Result<(), EmgauwaError>;
|
||
|
|
|
||
|
|
fn handle(&mut self, _msg: Reload, _ctx: &mut Self::Context) -> Self::Result {
|
||
|
|
let mut pool_conn = block_on(self.pool.acquire())?;
|
||
|
|
|
||
|
|
self.this.reload(&mut pool_conn)?;
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Handler<GetThis> for AppState {
|
||
|
|
type Result = Controller;
|
||
|
|
|
||
|
|
fn handle(&mut self, _msg: GetThis, _ctx: &mut Self::Context) -> Self::Result {
|
||
|
|
self.this.clone()
|
||
|
|
}
|
||
|
|
}
|