fix: Panic when closing DB while other goroutines doing reads

Signed-off-by: fatelei <[email protected]>

db.go

@@ -546,6 +546,7 @@ func (db *DB) close() (err error) {
 	db.opt.Infof("Lifetime L0 stalled for: %s\n", time.Duration(db.lc.l0stallsMs.Load()))
 
 	db.blockWrites.Store(1)
+	db.isClosed.Store(1)
 
 	if !db.opt.InMemory {
 		// Stop value GC first.
@@ -631,7 +632,6 @@ func (db *DB) close() (err error) {
 	db.blockCache.Close()
 	db.indexCache.Close()
 
-	db.isClosed.Store(1)
 	db.threshold.close()
 
 	if db.opt.InMemory {

db_test.go

@@ -2635,3 +2635,37 @@ func TestCompactL0OnClose(t *testing.T) {
 		}
 	})
 }
+
+func TestCloseDBWhileReading(t *testing.T) {
+	dir := t.TempDir()
+	db, err := Open(DefaultOptions(dir))
+	require.NoError(t, err)
+
+	key := []byte("key")
+	err = db.Update(func(txn *Txn) error {
+		return txn.Set(key, []byte("value"))
+	})
+	require.NoError(t, err)
+
+	var wg sync.WaitGroup
+	for i := 0; i < 10; i++ {
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			for {
+				err := db.View(func(txn *Txn) error {
+					_, err := txn.Get(key)
+					return err
+				})
+				if err != nil {
+					require.Contains(t, err.Error(), "DB Closed")
+					break
+				}
+			}
+		}()
+	}
+
+	time.Sleep(time.Second)
+	require.NoError(t, db.Close())
+	wg.Wait()
+}