initial websocket implementation

src/main.rs

@@ -1,15 +1,16 @@
-use std::sync::Arc;
+use std::{net::SocketAddr, sync::Arc, time::Duration};
 
 use axum::{extract::State, http::HeaderMap, routing::get};
 use reqwest::{header::USER_AGENT};
 use sea_orm::{ConnectionTrait, Database, DatabaseConnection, EntityTrait, PaginatorTrait, Schema};
+use tokio::{sync::broadcast::{self, Sender}, task, time};
 use utoipa::openapi::{ContactBuilder, InfoBuilder, LicenseBuilder};
 use utoipa_axum::router::OpenApiRouter;
 use utoipa_redoc::{Redoc, Servable};
 use utoipa_swagger_ui::{Config, SwaggerUi};
 use utoipa_axum::routes;
 
-use crate::entities::prelude::BookInstance;
+use crate::{entities::{owner, prelude::BookInstance}, utils::events::Event};
 
 pub mod entities;
 pub mod utils;
@@ -18,6 +19,7 @@ pub mod routes;
 pub struct AppState {
     app_name: String,
     db_conn: Arc<DatabaseConnection>,
+    event_bus: Sender<Event>,
     web_client: reqwest::Client
 }
 
@@ -53,11 +55,31 @@ async fn main() {
         return;
     }
 
+    let (event_bus, _) = broadcast::channel(16);
+    
+    let ntx = event_bus.clone();
+    let _forever = task::spawn(async move {
+        let mut interval = time::interval(Duration::from_secs(5));
+
+        let mut id = 1;
+        loop {
+            interval.tick().await;
+            let _ = ntx.send(Event::WebsocketBroadcast(utils::events::WebsocketMessage::NewOwner(Arc::new(owner::Model {
+                id,
+                first_name: "Avril".to_string(),
+                last_name: "Papillon".to_string(),
+                contact: "avril.papillon@proton.me".to_string()
+            }))));
+            id += 1;
+        }
+    });
+
     let mut default_headers = HeaderMap::new();
     default_headers.append(USER_AGENT, "Alexandria/1.0 (unionetudianteauvergne@gmail.com)".parse().unwrap());
     let shared_state = Arc::new(AppState {
         app_name: "Alexandria".to_string(),
         db_conn: db,
+        event_bus,
         web_client: reqwest::Client::builder().default_headers(default_headers).build().expect("creating the reqwest client failed")
     });
 
@@ -68,6 +90,7 @@ async fn main() {
         .routes(routes!(routes::book_instance::create_book_instance))
         .routes(routes!(routes::owner::get_owner_by_id))
         .routes(routes!(routes::owner::create_owner))
+        .route("/ws", get(routes::websocket::ws_handler))
         .route("/", get(index))
         .with_state(shared_state)
         .split_for_parts();
@@ -95,6 +118,10 @@ async fn main() {
     let router = router.merge(swagger);
     
     let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
-    axum::serve(listener, router).await.unwrap();
+    println!("Running on http://{}", listener.local_addr().unwrap());
+    axum::serve(
+        listener,
+        router.into_make_service_with_connect_info::<SocketAddr>()
+    ).await.unwrap();
 }