(overly) properly flush zstd in replaytool

1 files changed, 9 insertions, 3 deletions

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...");