starred/minne
1
0
Fork 0
mirror of https://github.com/perstarkse/minne.git synced 2026-03-24 18:31:45 +01:00
Code Issues Packages Projects Releases 10 Wiki Activity

improved rabbitmq structs

Browse Source
This commit is contained in:
Per Stark
2024-09-20 12:40:40 +02:00
parent d45ec5b503
commit 9be73c30d4
6 changed files with 424 additions and 203 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
src/server.rs
View File

@@ -53,6 +53,77 @@
// let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
// axum::serve(listener, app).await.unwrap();
// Ok(())
// }
// use axum::{
// routing::post,
// Router,
// response::{IntoResponse, Response},
// Error, Extension, Json,
// };
// use serde::Deserialize;
// use tracing_subscriber::{fmt, prelude::*, EnvFilter};
// use zettle_db::rabbitmq::{RabbitMQProducer};
// use std::sync::Arc;
// #[derive(Deserialize)]
// struct IngressPayload {
// payload: String,
// }
// use tracing::{info, error};
// async fn ingress_handler(
// // Extension(rabbitmq): Extension<Arc<RabbitMQClient>>,
// Extension(producer): Extension<Arc<RabbitMQProducer>>,
// Json(payload): Json<IngressPayload>
// ) -> Response {
// info!("Received payload: {:?}", payload.payload);
// match producer.publish(&payload.payload).await {
// Ok(_) => {
// info!("Message published successfully");
// "thank you".to_string().into_response()
// },
// Err(e) => {
// error!("Failed to publish message: {:?}", e);
// (axum::http::StatusCode::INTERNAL_SERVER_ERROR, "Failed to publish message").into_response()
// }
// }
// // match rabbitmq.publish("hello", payload.payload.as_bytes()).await {
// // Ok(_) => {
// // info!("Message published successfully");
// // "thank you".to_string().into_response()
// // },
// // Err(e) => {
// // error!("Failed to publish message: {:?}", e);
// // (axum::http::StatusCode::INTERNAL_SERVER_ERROR, "Failed to publish message").into_response()
// // }
// // }
// }
// #[tokio::main(flavor = "multi_thread", worker_threads = 2)]
// async fn main() -> Result<(), Error> {
// // Set up tracing
// tracing_subscriber::registry()
// .with(fmt::layer())
// .with(EnvFilter::from_default_env())
// .try_init()
// .ok();
// // Set up RabbitMQ
// // let rabbitmq = Arc::new(RabbitMQClient::new("amqp://guest:guest@localhost:5672").await.expect("Failed to connect to RabbitMQ"));
// let producer = Arc::new(RabbitMQProducer::new("amqp://localhost", "my_exchange", "my_routing_key").await);
// // Create Axum router
// let app = Router::new()
// .route("/ingress", post(ingress_handler))
// .layer(Extension(producer));
// tracing::info!("Listening on 0.0.0.0:3000");
// let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
// axum::serve(listener, app).await.unwrap();
// Ok(())
// }
use axum::{
@@ -63,7 +134,7 @@ use axum::{
};
use serde::Deserialize;
use tracing_subscriber::{fmt, prelude::*, EnvFilter};
use zettle_db::rabbitmq::RabbitMQClient;
use zettle_db::rabbitmq::{RabbitMQProducer, RabbitMQError};
use std::sync::Arc;
#[derive(Deserialize)]
@@ -74,11 +145,11 @@ struct IngressPayload {
use tracing::{info, error};
async fn ingress_handler(
Extension(rabbitmq): Extension<Arc<RabbitMQClient>>,
Extension(producer): Extension<Arc<RabbitMQProducer>>,
Json(payload): Json<IngressPayload>
) -> Response {
info!("Received payload: {:?}", payload.payload);
match rabbitmq.publish("hello", payload.payload.as_bytes()).await {
match producer.publish(&payload.payload).await {
Ok(_) => {
info!("Message published successfully");
"thank you".to_string().into_response()
@@ -91,7 +162,7 @@ async fn ingress_handler(
}
#[tokio::main(flavor = "multi_thread", worker_threads = 2)]
async fn main() -> Result<(), Error> {
async fn main() -> Result<(), Box<dyn std::error::Error>> {
// Set up tracing
tracing_subscriber::registry()
.with(fmt::layer())
@@ -100,17 +171,18 @@ async fn main() -> Result<(), Error> {
.ok();
// Set up RabbitMQ
let rabbitmq = Arc::new(RabbitMQClient::new("amqp://guest:guest@localhost:5672").await.expect("Failed to connect to RabbitMQ"));
let producer = Arc::new(RabbitMQProducer::new("amqp://localhost", "my_exchange", "my_routing_key").await?);
// Create Axum router
let app = Router::new()
.route("/ingress", post(ingress_handler))
.layer(Extension(rabbitmq));
.layer(Extension(producer));
tracing::info!("Listening on 0.0.0.0:3000");
let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
axum::serve(listener, app).await.unwrap();
let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await?;
axum::serve(listener, app).await?;
Ok(())
}