do not drain if db is closed

drpcorg · Dec 27, 2024 · 4ef89f6 · 4ef89f6
1 parent 5cf44ed
commit 4ef89f6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/chotki.go b/chotki.go
@@ -159,7 +159,7 @@ type Chotki struct {
 	src   uint64
 	clock rdx.Clock
 
-	lock         sync.Mutex
+	lock         sync.RWMutex
 	db           *pebble.DB
 	net          *protocol.Net
 	dir          string
@@ -661,6 +661,11 @@ func (cho *Chotki) drain(ctx context.Context, recs protocol.Records) (err error)
 }
 
 func (cho *Chotki) Drain(ctx context.Context, recs protocol.Records) (err error) {
+	cho.lock.RLock()
+	defer cho.lock.RUnlock()
+	if cho.db == nil {
+		return ErrClosed
+	}
 	EventsBatchSize.Observe(float64(len(recs)))
 	return cho.drain(ctx, recs)
 }