Clean up temporary snapshots when there are errors

Guaderxx · Jul 10, 2017 · 5f378a6 · 5f378a6
1 parent 37d2320
commit 5f378a6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/file_snapshot.go b/file_snapshot.go
@@ -369,6 +369,10 @@ func (s *FileSnapshotSink) Close() error {
 	// Close the open handles
 	if err := s.finalize(); err != nil {
 		s.logger.Printf("[ERR] snapshot: Failed to finalize snapshot: %v", err)
+		if delErr := os.RemoveAll(s.dir); delErr != nil {
+			s.logger.Printf("[ERR] snapshot: Failed to delete temporary snapshot at path %v: %v", s.dir, delErr)
+			return delErr
+		}
 		return err
 	}