YGGverse/ggemini
1
0
Fork 0
mirror of https://github.com/YGGverse/ggemini.git synced 2026-03-31 17:15:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make Size struct for tuple argument

Browse source
This commit is contained in:
yggverse 2025-03-27 21:14:14 +02:00
parent b6ea830545
commit 4dddbd5f8a
1 changed files with 28 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

45
src/gio/file_output_stream.rs
View file

@ -7,6 +7,14 @@ use gio::{
}; };
use glib::{Bytes, Priority, object::IsA}; use glib::{Bytes, Priority, object::IsA};
/// Mutable bytes count
pub struct Size {
pub chunk: usize,
/// `None` for unlimited
pub limit: Option<usize>,
pub total: usize,
}
/// Asynchronously move all bytes from [IOStream](https://docs.gtk.org/gio/class.IOStream.html) /// Asynchronously move all bytes from [IOStream](https://docs.gtk.org/gio/class.IOStream.html)
/// to [FileOutputStream](https://docs.gtk.org/gio/class.FileOutputStream.html) /// to [FileOutputStream](https://docs.gtk.org/gio/class.FileOutputStream.html)
/// * require `IOStream` reference to keep `Connection` active in async thread /// * require `IOStream` reference to keep `Connection` active in async thread
@ -15,59 +23,50 @@ pub fn from_stream_async(
file_output_stream: FileOutputStream, file_output_stream: FileOutputStream,
cancellable: Cancellable, cancellable: Cancellable,
priority: Priority, priority: Priority,
(chunk, limit, mut total): ( mut size: Size,
usize, // bytes_in_chunk
Option<usize>, // bytes_total_limit, `None` to unlimited
usize, // bytes_total
),
(on_chunk, on_complete): ( (on_chunk, on_complete): (
impl Fn(Bytes, usize) + 'static, // on_chunk impl Fn(Bytes, usize) + 'static, // on_chunk
impl FnOnce(Result<(FileOutputStream, usize), Error>) + 'static, // on_complete impl FnOnce(Result<(FileOutputStream, usize), Error>) + 'static, // on_complete
), ),
) { ) {
io_stream.input_stream().read_bytes_async( io_stream.input_stream().read_bytes_async(
chunk, size.chunk,
priority, priority,
Some(&cancellable.clone()), Some(&cancellable.clone()),
move |result| match result { move |result| match result {
Ok(bytes) => { Ok(bytes) => {
total += bytes.len(); size.total += bytes.len();
on_chunk(bytes.clone(), total); on_chunk(bytes.clone(), size.total);
if let Some(limit) = limit { if let Some(limit) = size.limit {
if total > limit { if size.total > limit {
return on_complete(Err(Error::BytesTotal(total, limit))); return on_complete(Err(Error::BytesTotal(size.total, limit)));
} }
} }
if bytes.len() == 0 { if bytes.len() == 0 {
return on_complete(Ok((file_output_stream, total))); return on_complete(Ok((file_output_stream, size.total)));
} }
// Make sure **all bytes** sent to the destination // Make sure **all bytes** sent to the destination
// > A partial write is performed with the size of a message block, which is 16kB // > A partial write is performed with the size of a message block, which is 16kB
// > https://docs.openssl.org/3.0/man3/SSL_write/#notes // > https://docs.openssl.org/3.0/man3/SSL_write/#notes
file_output_stream.clone().write_all_async( file_output_stream.clone().write_all_async(
bytes.clone(), bytes,
priority, priority,
Some(&cancellable.clone()), Some(&cancellable.clone()),
move |result| { move |result| match result {
match result { Ok(_) => from_stream_async(
Ok(_) => {
// continue read..
from_stream_async(
io_stream, io_stream,
file_output_stream, file_output_stream,
cancellable, cancellable,
priority, priority,
(chunk, limit, total), size,
(on_chunk, on_complete), (on_chunk, on_complete),
); ),
}
Err((b, e)) => on_complete(Err(Error::OutputStream(b, e))), Err((b, e)) => on_complete(Err(Error::OutputStream(b, e))),
}
}, },
); )
} }
Err(e) => on_complete(Err(Error::InputStream(e))), Err(e) => on_complete(Err(Error::InputStream(e))),
}, },