Noratrieb/haesli
1
0
Fork 0
mirror of https://github.com/Noratrieb/haesli.git synced 2026-01-14 11:45:02 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

improve dashboard startup

Browse source
This commit is contained in:
nora 2022-03-07 15:24:46 +01:00
parent 08fa9163b8
commit 2fe3b4b77b
3 changed files with 27 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

14
amqp_dashboard/src/archive.rs
View file

@ -41,15 +41,17 @@ pub struct StaticFileService {
}
impl StaticFileService {
pub fn new(zip: &[u8]) -> Self {
let mut archive = ZipArchive::new(Cursor::new(zip)).unwrap();
/// Creates a new static file service from zip data. This is a blocking operation!
#[tracing::instrument(skip(zip))]
pub fn new(zip: &[u8]) -> anyhow::Result<Self> {
let mut archive = ZipArchive::new(Cursor::new(zip))?;
let mut files = HashMap::with_capacity(archive.len());
for i in 0..archive.len() {
let mut file = archive.by_index(i).unwrap();
let mut data = Vec::with_capacity(usize::try_from(file.size()).unwrap());
std::io::copy(&mut file, &mut data).unwrap();
let mut file = archive.by_index(i)?;
let mut data = Vec::with_capacity(usize::try_from(file.size())?);
std::io::copy(&mut file, &mut data)?;
trace!(name = %file.name(), size = %file.size(),"Unpacking dashboard frontend file");
@ -75,7 +77,7 @@ impl StaticFileService {
trace!(?files, "Created StaticFileService");
Self { files }
Ok(Self { files })
}
fn call_inner(&mut self, req: Request<Body>) -> Result<Response<Body>, anyhow::Error> {

24
amqp_dashboard/src/lib.rs
View file

@ -16,16 +16,26 @@ use tracing::{error, info};
const DATA_ZIP: &[u8] = include_bytes!(concat!(env!("OUT_DIR"), "/frontend.zip"));
pub async fn dashboard(global_data: GlobalData) {
pub async fn start_dashboard(global_data: GlobalData) {
match dashboard(global_data).await {
Ok(()) => {}
Err(err) => error!(%err, "Failed to start dashboard"),
}
}
#[tracing::instrument(skip(global_data))]
pub async fn dashboard(global_data: GlobalData) -> anyhow::Result<()> {
let cors = CorsLayer::new()
.allow_methods(vec![Method::GET])
.allow_origin(Any);
let static_file_service =
get_service(StaticFileService::new(DATA_ZIP)).handle_error(|error| async move {
error!(?error, "Error in static file service");
StatusCode::INTERNAL_SERVER_ERROR
});
tokio::task::spawn_blocking(|| StaticFileService::new(DATA_ZIP)).await??;
let static_file_service = get_service(static_file_service).handle_error(|error| async move {
error!(?error, "Error in static file service");
StatusCode::INTERNAL_SERVER_ERROR
});
let app = Router::new()
.route("/api/data", get(move || get_data(global_data)).layer(cors))
@ -37,8 +47,8 @@ pub async fn dashboard(global_data: GlobalData) {
axum::Server::bind(&socket_addr)
.serve(app.into_make_service())
.await
.unwrap();
.await?;
Ok(())
}
#[derive(Serialize)]

8
src/main.rs
View file

@ -3,7 +3,7 @@
use anyhow::Result;
use clap::Parser;
use std::str::FromStr;
use tracing::{info, info_span, Instrument};
use tracing::info;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt, EnvFilter, Registry};
/// An AMQP 0-9-1 broker implementation.
@ -28,11 +28,7 @@ async fn main() -> Result<()> {
if args.dashboard {
let global_data = global_data.clone();
tokio::spawn(async move {
amqp_dashboard::dashboard(global_data)
.instrument(info_span!("dashboard"))
.await
});
tokio::spawn(async move { amqp_dashboard::start_dashboard(global_data).await });
}
let res = amqp_transport::do_thing_i_guess(global_data, terminate()).await;