mirror of
https://github.com/perstarkse/minne.git
synced 2026-03-25 02:41:27 +01:00
moved entry points to bin to clarify
This commit is contained in:
58
src/bin/server.rs
Normal file
58
src/bin/server.rs
Normal file
@@ -0,0 +1,58 @@
|
||||
use axum::{
|
||||
extract::DefaultBodyLimit,
|
||||
routing::{delete, get, post, put},
|
||||
Extension, Router,
|
||||
};
|
||||
use std::sync::Arc;
|
||||
use tracing_subscriber::{fmt, prelude::*, EnvFilter};
|
||||
use zettle_db::{
|
||||
rabbitmq::{publisher::RabbitMQProducer, RabbitMQConfig},
|
||||
routes::{
|
||||
file::{delete_file_handler, get_file_handler, update_file_handler, upload_handler},
|
||||
ingress::ingress_handler,
|
||||
queue_length::queue_length_handler,
|
||||
},
|
||||
storage::db::SurrealDbClient,
|
||||
};
|
||||
|
||||
#[tokio::main(flavor = "multi_thread", worker_threads = 2)]
|
||||
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
// Set up tracing
|
||||
tracing_subscriber::registry()
|
||||
.with(fmt::layer())
|
||||
.with(EnvFilter::from_default_env())
|
||||
.try_init()
|
||||
.ok();
|
||||
|
||||
// Set up RabbitMQ
|
||||
let config = RabbitMQConfig {
|
||||
amqp_addr: "amqp://localhost".to_string(),
|
||||
exchange: "my_exchange".to_string(),
|
||||
queue: "my_queue".to_string(),
|
||||
routing_key: "my_key".to_string(),
|
||||
};
|
||||
|
||||
// Set up producer
|
||||
let producer = Arc::new(RabbitMQProducer::new(&config).await?);
|
||||
|
||||
// Set up database client
|
||||
let db_client = Arc::new(SurrealDbClient::new().await?);
|
||||
|
||||
// Create Axum router
|
||||
let app = Router::new()
|
||||
.route("/ingress", post(ingress_handler))
|
||||
.route("/message_count", get(queue_length_handler))
|
||||
.layer(Extension(producer))
|
||||
.route("/file", post(upload_handler))
|
||||
.layer(DefaultBodyLimit::max(1024 * 1024 * 1024))
|
||||
.route("/file/:uuid", get(get_file_handler))
|
||||
.route("/file/:uuid", put(update_file_handler))
|
||||
.route("/file/:uuid", delete(delete_file_handler))
|
||||
.layer(Extension(db_client));
|
||||
|
||||
tracing::info!("Listening on 0.0.0.0:3000");
|
||||
let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await?;
|
||||
axum::serve(listener, app).await?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
Reference in New Issue
Block a user