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

Fix

Browse Source
This commit is contained in:
Bulat Kurbanov
2024-01-08 02:09:01 +01:00
parent 801f9664cf
commit b21f70cb8f
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/bots_manager/axum_server.rs
View File

@@ -1,9 +1,11 @@
use axum::extract::State;
use axum::response::IntoResponse; use axum::response::IntoResponse;
use axum::routing::post; use axum::routing::post;
use axum::{extract::Path, routing::get}; use axum::{extract::Path, routing::get};
use axum_prometheus::PrometheusMetricLayer; use axum_prometheus::PrometheusMetricLayer;
use reqwest::StatusCode; use reqwest::StatusCode;
use tokio::sync::Mutex;
use tokio::time; use tokio::time;
use std::time::Duration; use std::time::Duration;
@@ -26,7 +28,7 @@ use crate::bots_manager::{internal::start_bot, BOTS_DATA, BOTS_ROUTES, SERVER_PO
pub async fn start_axum_server(stop_signal: Arc<AtomicBool>) { pub async fn start_axum_server(stop_signal: Arc<AtomicBool>) {
async fn telegram_request( async fn telegram_request(
// State(start_bot_mutex): State<Arc<Mutex<()>>>, State(start_bot_mutex): State<Arc<Mutex<()>>>,
Path(token): Path<String>, Path(token): Path<String>,
input: String, input: String,
) -> impl IntoResponse { ) -> impl IntoResponse {
@@ -40,7 +42,7 @@ pub async fn start_axum_server(stop_signal: Arc<AtomicBool>) {
} }
'creator: { 'creator: {
// let _guard = start_bot_mutex.lock().await; let _guard = start_bot_mutex.lock().await;
if BOTS_ROUTES.contains_key(&token) { if BOTS_ROUTES.contains_key(&token) {
break 'creator; break 'creator;
@@ -91,11 +93,12 @@ pub async fn start_axum_server(stop_signal: Arc<AtomicBool>) {
let (prometheus_layer, metric_handle) = PrometheusMetricLayer::pair(); let (prometheus_layer, metric_handle) = PrometheusMetricLayer::pair();
// let start_bot_mutex = Arc::new(Mutex::new(())); let start_bot_mutex = Arc::new(Mutex::new(()));
let app_router = axum::Router::new() let app_router = axum::Router::new()
.route("/:token/", post(telegram_request)) .route("/:token/", post(telegram_request))
.layer(prometheus_layer); .layer(prometheus_layer)
.with_state(start_bot_mutex);
let metric_router = let metric_router =
axum::Router::new().route("/metrics", get(|| async move { metric_handle.render() })); axum::Router::new().route("/metrics", get(|| async move { metric_handle.render() }));