From 67694f908f7fc76f78566e6f02d89db2d7a19c1d Mon Sep 17 00:00:00 2001 From: metamuffin Date: Tue, 27 Dec 2022 16:20:36 +0100 Subject: rift: handle control messages --- client-native-rift/src/main.rs | 38 +++++++++++++++++++++++--------------- 1 file changed, 23 insertions(+), 15 deletions(-) (limited to 'client-native-rift/src') diff --git a/client-native-rift/src/main.rs b/client-native-rift/src/main.rs index 5d0c7d6..45f720a 100644 --- a/client-native-rift/src/main.rs +++ b/client-native-rift/src/main.rs @@ -192,21 +192,29 @@ impl EventHandler for Handler { let writer = writer.clone(); let pos = pos.clone(); Box::pin(async move { - let pos = pos.fetch_add(mesg.data.len(), Ordering::Relaxed); - info!( - "recv {:?} ({} of {})", - mesg.data.len(), - humansize::format_size(pos, DECIMAL), - humansize::format_size(resource.size.unwrap_or(0), DECIMAL), - ); - writer - .write() - .await - .as_mut() - .unwrap() - .write_all(&mesg.data) - .await - .unwrap(); + // TODO + if mesg.is_string { + let s = String::from_utf8((&mesg.data).to_vec()).unwrap(); + if &s == "end" { + info!("EOF reached") + } + } else { + let pos = pos.fetch_add(mesg.data.len(), Ordering::Relaxed); + info!( + "recv {:?} ({} of {})", + mesg.data.len(), + humansize::format_size(pos, DECIMAL), + humansize::format_size(resource.size.unwrap_or(0), DECIMAL), + ); + writer + .write() + .await + .as_mut() + .unwrap() + .write_all(&mesg.data) + .await + .unwrap(); + } }) }) } -- cgit v1.2.3-70-g09d2