git/caesar-transfer
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

fix(shuttle): fix imports after code refactoring

Browse source
This commit is contained in:
Patryk Hegenberg 2024-04-28 10:03:22 +02:00
parent bec368325c
commit 26f09218f4
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/shuttle.rs
View file

@ -1,6 +1,7 @@
use crate::relay::appstate::AppState;
use crate::relay::server::ws_handler;
use axum::{routing::get, Router}; use axum::{routing::get, Router};
use axum_client_ip::SecureClientIpSource; use axum_client_ip::SecureClientIpSource;
use relay::server;
use shuttle_axum::ShuttleAxum; use shuttle_axum::ShuttleAxum;
pub mod receiver; pub mod receiver;
@ -11,11 +12,11 @@ pub mod shared;
#[shuttle_runtime::main] #[shuttle_runtime::main]
async fn axum() -> ShuttleAxum { async fn axum() -> ShuttleAxum {
// Create a new server data structure. // Create a new server data structure.
let appstate = server::AppState::new(); let appstate = AppState::new();
// Set up the application routes. // Set up the application routes.
let app = Router::new() let app = Router::new()
.route("/ws", get(relay::ws_handler)) .route("/ws", get(ws_handler))
.with_state(appstate) .with_state(appstate)
.layer(SecureClientIpSource::ConnectInfo.into_extension()); .layer(SecureClientIpSource::ConnectInfo.into_extension());