fix(filestorage): sync and attempt delete file on error (#450)

cshum · Jul 9, 2024 · ef34a5f · ef34a5f
1 parent 716f298
commit ef34a5f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/storage/filestorage/filestorage.go b/storage/filestorage/filestorage.go
@@ -99,10 +99,16 @@ func (s *FileStorage) Put(_ context.Context, image string, blob *imagor.Blob) (e
 	}
 	defer func() {
 		_ = w.Close()
+		if err != nil {
+			_ = os.Remove(w.Name())
+		}
 	}()
 	if _, err = io.Copy(w, reader); err != nil {
 		return
 	}
+	if err = w.Sync(); err != nil {
+		return
+	}
 	return
 }