Trying to fix rust compat

This commit is contained in:
Igor Katson 2024-11-20 16:13:37 +00:00
parent 69a638fb81
commit e5c11e778a
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
2 changed files with 14 additions and 15 deletions

View file

@ -46,7 +46,7 @@ pub mod stats;
use std::{
collections::{HashMap, HashSet},
net::SocketAddr,
num::NonZero,
num::NonZeroU32,
sync::{
atomic::{AtomicBool, AtomicU64, Ordering},
Arc,
@ -418,12 +418,12 @@ impl TorrentStateLive {
) -> anyhow::Result<()> {
while let Some((tx, ci)) = rx.recv().await {
self.ratelimits
.prepare_for_upload(NonZero::new(ci.size).unwrap())
.prepare_for_upload(NonZeroU32::new(ci.size).unwrap())
.await?;
if let Some(session) = self.torrent.session.upgrade() {
session
.ratelimits
.prepare_for_upload(NonZero::new(ci.size).unwrap())
.prepare_for_upload(NonZeroU32::new(ci.size).unwrap())
.await?;
}
let _ = tx.send(WriterRequest::ReadChunkRequest(ci));
@ -1449,13 +1449,13 @@ impl PeerHandler {
self.state
.ratelimits
.prepare_for_download(NonZero::new(request.length).unwrap())
.prepare_for_download(NonZeroU32::new(request.length).unwrap())
.await?;
if let Some(session) = self.state.torrent().session.upgrade() {
session
.ratelimits
.prepare_for_download(NonZero::new(request.length).unwrap())
.prepare_for_download(NonZeroU32::new(request.length).unwrap())
.await?;
}