Add live data threads

Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>

backend/src/routes/assets.rs

@@ -1,12 +1,17 @@
+use crate::data::live::AssetMPSCMessage;
 use crate::database;
 use crate::database::assets::update_asset_trading;
 use crate::pool::alpaca::AlpacaPool;
 use crate::pool::postgres::PostgresPool;
 use crate::types::{Asset, Class, Exchange};
 use apca::api::v2::asset::{self, Symbol};
+use apca::RequestError;
 use axum::{extract::Path, http::StatusCode, Extension, Json};
+use log::info;
 use serde::Deserialize;
 use sqlx::types::time::OffsetDateTime;
+use std::sync::Arc;
+use tokio::sync::mpsc::Sender;
 
 pub async fn get_assets(
     Extension(postgres_pool): Extension<PostgresPool>,
@@ -42,6 +47,8 @@ pub struct AddAssetRequest {
 pub async fn add_asset(
     Extension(postgres_pool): Extension<PostgresPool>,
     Extension(alpaca_pool): Extension<AlpacaPool>,
+    Extension(stock_live_mpsc_sender): Extension<Arc<Sender<AssetMPSCMessage>>>,
+    Extension(crypto_live_mpsc_sender): Extension<Arc<Sender<AssetMPSCMessage>>>,
     Json(request): Json<AddAssetRequest>,
 ) -> Result<(StatusCode, Json<Asset>), StatusCode> {
     if database::assets::get_asset(&postgres_pool, &request.symbol)
@@ -58,7 +65,10 @@ pub async fn add_asset(
         .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?
         .issue::<asset::Get>(&Symbol::Sym(request.symbol))
         .await
-        .map_err(|_| StatusCode::NOT_FOUND)?;
+        .map_err(|e| match e {
+            RequestError::Endpoint(_) => StatusCode::NOT_FOUND,
+            _ => StatusCode::INTERNAL_SERVER_ERROR,
+        })?;
 
     let asset = Asset {
         symbol: asset.symbol,
@@ -72,6 +82,23 @@ pub async fn add_asset(
         .await
         .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
 
+    match asset.class {
+        Class(asset::Class::UsEquity) => {
+            stock_live_mpsc_sender
+                .send(AssetMPSCMessage::Added(asset.clone()))
+                .await
+                .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
+        }
+        Class(asset::Class::Crypto) => {
+            crypto_live_mpsc_sender
+                .send(AssetMPSCMessage::Added(asset.clone()))
+                .await
+                .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
+        }
+        _ => {}
+    }
+
+    info!("Added asset {}.", asset.symbol);
     Ok((StatusCode::CREATED, Json(asset)))
 }
 
@@ -91,21 +118,45 @@ pub async fn update_asset(
         .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
 
     match asset {
-        Some(asset) => Ok((StatusCode::OK, Json(asset))),
+        Some(asset) => {
+            info!("Updated asset {}.", symbol);
+            Ok((StatusCode::OK, Json(asset)))
+        }
         None => Err(StatusCode::NOT_FOUND),
     }
 }
 
 pub async fn delete_asset(
     Extension(postgres_pool): Extension<PostgresPool>,
+    Extension(stock_live_mpsc_sender): Extension<Arc<Sender<AssetMPSCMessage>>>,
+    Extension(crypto_live_mpsc_sender): Extension<Arc<Sender<AssetMPSCMessage>>>,
     Path(symbol): Path<String>,
-) -> Result<(StatusCode, Json<Asset>), StatusCode> {
+) -> Result<StatusCode, StatusCode> {
     let asset = database::assets::delete_asset(&postgres_pool, &symbol)
         .await
         .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
 
     match asset {
-        Some(asset) => Ok((StatusCode::NO_CONTENT, Json(asset))),
+        Some(asset) => {
+            match asset.class {
+                Class(asset::Class::UsEquity) => {
+                    stock_live_mpsc_sender
+                        .send(AssetMPSCMessage::Removed(asset.clone()))
+                        .await
+                        .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
+                }
+                Class(asset::Class::Crypto) => {
+                    crypto_live_mpsc_sender
+                        .send(AssetMPSCMessage::Removed(asset.clone()))
+                        .await
+                        .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
+                }
+                _ => {}
+            }
+
+            info!("Deleted asset {}.", symbol);
+            Ok(StatusCode::NO_CONTENT)
+        }
         None => Err(StatusCode::NOT_FOUND),
     }
 }