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

sigoden · Nov 3, 2023 · 8b4cab1 · 8b4cab1
1 parent 70300b1
commit 8b4cab1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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(())