From: sigoden <sigoden@gmail.com>
Date: Fri, 3 Nov 2023 12:58:53 +0000 (+0800)
Subject: fix: auto delete half-uploaded files (#280)
X-Git-Url: https://git.ozva.co.uk/?a=commitdiff_plain;h=8b4cab1e691f46c0c0cbbcae8dc0b4e033114706;p=ozva-cloud

fix: auto delete half-uploaded files (#280)
---

diff --git a/src/server.rs b/src/server.rs
index 0cd0989..85cbce1 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -427,7 +427,12 @@ impl Server {
 
         futures::pin_mut!(body_reader);
 
-        io::copy(&mut body_reader, &mut file).await?;
+        let ret = io::copy(&mut body_reader, &mut file).await;
+        if ret.is_err() {
+            tokio::fs::remove_file(&path).await?;
+
+            ret?;
+        }
 
         *res.status_mut() = StatusCode::CREATED;
         Ok(())