From a16fd3bca9e3d74916bcc473147e5f7dd4bb3040 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Thu, 11 Jul 2024 00:11:27 +0200
Subject: (overly) properly flush zstd in replaytool

---
 server/replaytool/src/main.rs | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'server/replaytool/src/main.rs')

diff --git a/server/replaytool/src/main.rs b/server/replaytool/src/main.rs
index 527c5abf..e8dae5ef 100644
--- a/server/replaytool/src/main.rs
+++ b/server/replaytool/src/main.rs
@@ -68,7 +68,7 @@ async fn main() -> anyhow::Result<()> {
             output,
             r#loop,
         } => loop {
-            let mut file = BufWriter::new(ZstdEncoder::new(
+            let mut file = BufWriter::new(ZstdEncoder::new(BufWriter::new(
                 File::create(&if r#loop {
                     output.join(format!(
                         "replay-{}",
@@ -81,7 +81,7 @@ async fn main() -> anyhow::Result<()> {
                     output.clone()
                 })
                 .await?,
-            ));
+            )));
             info!("connecting to {url:?}...");
             let (mut sock, _) = tokio_tungstenite::connect_async(&url).await?;
             info!("starting recording.");
@@ -123,7 +123,13 @@ async fn main() -> anyhow::Result<()> {
                     _ => (),
                 }
             }
-            drop(file);
+            file.flush().await?;
+            let mut file = file.into_inner();
+            file.flush().await?;
+            let mut file = file.into_inner();
+            file.flush().await?;
+            let mut file = file.into_inner();
+            file.flush().await?;
             info!("done");
             if r#loop {
                 info!("restarting...");
-- 
cgit v1.2.3-70-g09d2