leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Ready to test session limits

Browse source
This commit is contained in:
Igor Katson 2024-11-07 22:00:55 +00:00
parent 37810443df
commit 3924197461
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
1 changed files with 16 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

18
crates/librqbit/src/http_api.rs
View file

@ -21,9 +21,10 @@ use tokio::net::TcpListener;
use tower_http::trace::{DefaultOnFailure, DefaultOnResponse, OnFailure}; use tower_http::trace::{DefaultOnFailure, DefaultOnResponse, OnFailure};
use tracing::{debug, error_span, trace, Span}; use tracing::{debug, error_span, trace, Span};
use axum::Router; use axum::{Json, Router};
use crate::api::{Api, ApiTorrentListOpts, TorrentIdOrHash}; use crate::api::{Api, ApiTorrentListOpts, EmptyJsonResponse, TorrentIdOrHash};
use crate::limits::LimitsConfig;
use crate::peer_connection::PeerConnectionOptions; use crate::peer_connection::PeerConnectionOptions;
use crate::session::{AddTorrent, AddTorrentOptions, SUPPORTED_SCHEMES}; use crate::session::{AddTorrent, AddTorrentOptions, SUPPORTED_SCHEMES};
use crate::torrent_state::peer::stats::snapshot::PeerStatsFilter; use crate::torrent_state::peer::stats::snapshot::PeerStatsFilter;
@ -490,6 +491,18 @@ impl HttpApi {
Ok(axum::body::Body::from_stream(s)) Ok(axum::body::Body::from_stream(s))
} }
async fn update_session_ratelimits(
State(state): State<ApiState>,
Json(limits): Json<LimitsConfig>,
) -> Result<impl IntoResponse> {
state.session().ratelimits.set_upload_bps(limits.upload_bps);
state
.session()
.ratelimits
.set_download_bps(limits.download_bps);
Ok(Json(EmptyJsonResponse {}))
}
let mut app = Router::new() let mut app = Router::new()
.route("/", get(api_root)) .route("/", get(api_root))
.route("/stream_logs", get(stream_logs)) .route("/stream_logs", get(stream_logs))
@ -515,6 +528,7 @@ impl HttpApi {
if !self.opts.read_only { if !self.opts.read_only {
app = app app = app
.route("/torrents", post(torrents_post)) .route("/torrents", post(torrents_post))
.route("/torrents/limits", post(update_session_ratelimits))
.route("/torrents/:id/pause", post(torrent_action_pause)) .route("/torrents/:id/pause", post(torrent_action_pause))
.route("/torrents/:id/start", post(torrent_action_start)) .route("/torrents/:id/start", post(torrent_action_start))
.route("/torrents/:id/forget", post(torrent_action_forget)) .route("/torrents/:id/forget", post(torrent_action_forget))