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