fix: handle nil *badger.DB in *Engines.Close (#398)

Co-authored-by: Connor <[email protected]>
talent-plan · Jul 6, 2022 · d975005 · d975005
1 parent dae8650
commit d975005
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/kv/util/engine_util/engines.go b/kv/util/engine_util/engines.go
@@ -38,11 +38,14 @@ func (en *Engines) WriteRaft(wb *WriteBatch) error {
 }
 
 func (en *Engines) Close() error {
-	if err := en.Kv.Close(); err != nil {
-		return err
-	}
-	if err := en.Raft.Close(); err != nil {
-		return err
+	dbs := []*badger.DB{en.Kv, en.Raft}
+	for _, db := range dbs {
+		if db == nil {
+			continue
+		}
+		if err := db.Close(); err != nil {
+			return err
+		}
 	}
 	return nil
 }