Add RabbitMQ messaging

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

backend/assets/src/main.rs

@@ -2,26 +2,46 @@ use axum::{
     routing::{delete, get, post},
     Extension, Router, Server,
 };
-use common::alpaca::create_alpaca_pool;
+use common::pool::{
+    create_alpaca_pool_from_env, create_database_pool_from_env, create_rabbitmq_pool_from_env,
+};
+use deadpool::managed::{Hook, HookError, HookErrorCause};
 use dotenv::dotenv;
+use lapin::ExchangeKind;
 use log::info;
-use sqlx::PgPool;
-use std::{env, error::Error, net::SocketAddr};
+use std::{error::Error, net::SocketAddr};
 
 mod routes;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
     dotenv().ok();
+    log4rs::init_file("log4rs.yaml", Default::default()).unwrap();
 
-    let database_pool = PgPool::connect(&env::var("DATABASE_URL").unwrap()).await?;
-
-    let alpaca_pool = create_alpaca_pool(
-        &env::var("APCA_API_BASE_URL").unwrap(),
-        &env::var("APCA_API_KEY_ID").unwrap(),
-        &env::var("APCA_API_SECRET_KEY").unwrap(),
-        10,
-    )?;
+    let num_clients = 10;
+    let database_pool = create_database_pool_from_env(num_clients).await?;
+    let alpaca_pool = create_alpaca_pool_from_env(num_clients).await?;
+    let rabbitmq_pool = create_rabbitmq_pool_from_env(
+        num_clients,
+        Hook::async_fn(|connection: &mut lapin::Connection, _| {
+            Box::pin(async move {
+                connection
+                    .create_channel()
+                    .await
+                    .map_err(|e| HookError::Abort(HookErrorCause::Backend(e)))?
+                    .exchange_declare(
+                        "assets",
+                        ExchangeKind::Topic,
+                        Default::default(),
+                        Default::default(),
+                    )
+                    .await
+                    .map_err(|e| HookError::Abort(HookErrorCause::Backend(e)))?;
+                Ok(())
+            })
+        }),
+    )
+    .await?;
 
     let app = Router::new()
         .route("/assets", get(routes::get_assets))
@@ -30,7 +50,8 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
         .route("/assets/:symbol", post(routes::update_asset))
         .route("/assets/:symbol", delete(routes::delete_asset))
         .layer(Extension(database_pool))
-        .layer(Extension(alpaca_pool));
+        .layer(Extension(alpaca_pool))
+        .layer(Extension(rabbitmq_pool));
 
     let addr = SocketAddr::from(([0, 0, 0, 0], 7878));
     info!("Listening on {}...", addr);