Apply suggestions from code review

crypto-org-chain · Jul 12, 2023 · 9765f8b · 9765f8b
1 parent 58e6a5e
commit 9765f8b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/memiavl/db.go b/memiavl/db.go
@@ -206,7 +206,7 @@ func Load(dir string, opts Options) (*DB, error) {
 	snapshotDir := snapshotName(db.lastCommitInfo.Version)
 	tmpDir := snapshotDir + "-tmp"
 	path = filepath.Join(db.dir, tmpDir)
-	if err := os.Remove(path); err != nil {
+	if err := os.RemoveAll(path); err != nil {
 		db.logger.Error("failed to remove tmp directories", "err", err)
 	}
 
@@ -471,7 +471,7 @@ func (db *DB) RewriteSnapshot() error {
 		return err
 	}
 	if err := db.MultiTree.WriteSnapshot(path); err != nil {
-		if err := os.Remove(path); err != nil {
+		if err := os.RemoveAll(path); err != nil {
 			db.logger.Error("failed to remove tmp directories", "err", err)
 		}
 		return err