feat: implement RabbitMQ module with server and consumer functionality

This commit is contained in:
Per Stark" (aider)
2024-09-18 12:23:56 +02:00
parent 00f32bc7f9
commit 0704a011b9
3 changed files with 95 additions and 23 deletions

View File

@@ -1,10 +1,42 @@
use axum::{
routing::post,
Router,
extract::Json,
};
use serde::Deserialize;
use zettle_db::rabbitmq::RabbitMQ;
use std::sync::Arc;
#[tokio::main(flavor = "multi_thread", worker_threads = 2)]
async fn main() {
let rabbitmq = RabbitMQ::new().await;
let queue_name = rabbitmq.declare_queue("amqprs.examples.basic").await;
rabbitmq.bind_queue(&queue_name.0, "amq.topic", "amqprs.example").await;
//...
#[derive(Deserialize)]
struct Message {
content: String,
}
async fn publish_message(
Json(message): Json<Message>,
rabbitmq: axum::extract::State<Arc<RabbitMQ>>,
) -> Result<(), String> {
rabbitmq
.publish_message("amq.topic", "amqprs.example", message.content)
.await
.map_err(|e| format!("Failed to publish message: {}", e))
}
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let rabbitmq = Arc::new(RabbitMQ::new().await?);
let queue_name = rabbitmq.declare_queue("amqprs.examples.basic").await?.0;
rabbitmq.bind_queue(&queue_name, "amq.topic", "amqprs.example").await?;
let app = Router::new()
.route("/publish", post(publish_message))
.with_state(rabbitmq);
println!("Server running on http://localhost:3000");
axum::Server::bind(&"0.0.0.0:3000".parse().unwrap())
.serve(app.into_make_service())
.await?;
Ok(())
}