flibusta-apps/fb2converter_server
1
0
Fork 0
mirror of https://github.com/flibusta-apps/fb2converter_server.git synced 2025-12-06 15:05:37 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add prometheus

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-08-10 00:05:00 +02:00
parent a73ba00075
commit dce9811cbd
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.toml
View File

@@ -20,3 +20,5 @@ sentry = "0.31.5"
async-tempfile = "0.4.0" async-tempfile = "0.4.0"
uuid = "1.4.1" uuid = "1.4.1"
axum-prometheus = "0.4.0"

6
src/main.rs
View File

@@ -1,5 +1,6 @@
use std::{net::SocketAddr, time::SystemTime}; use std::{net::SocketAddr, time::SystemTime};
use axum::{Router, routing::post, extract::Multipart, response::{IntoResponse, AppendHeaders}, http::{StatusCode, header}, body::StreamBody}; use axum::{Router, routing::{post, get}, extract::Multipart, response::{IntoResponse, AppendHeaders}, http::{StatusCode, header}, body::StreamBody};
use axum_prometheus::PrometheusMetricLayer;
use tokio::{fs::{remove_file, read_dir, remove_dir, File}, io::{AsyncWriteExt, copy}, process::Command}; use tokio::{fs::{remove_file, read_dir, remove_dir, File}, io::{AsyncWriteExt, copy}, process::Command};
use tracing::{info, log}; use tracing::{info, log};
use async_tempfile::TempFile; use async_tempfile::TempFile;
@@ -160,8 +161,11 @@ async fn convert_file(
fn get_router() -> Router { fn get_router() -> Router {
let (prometheus_layer, metric_handle) = PrometheusMetricLayer::pair();
Router::new() Router::new()
.route("/", post(convert_file)) .route("/", post(convert_file))
.route("/metrics", get(|| async move { metric_handle.render() }))
} }