closing feed when drain already closed

drpcorg · Sep 12, 2024 · ed7b3e8 · ed7b3e8
1 parent a173527
commit ed7b3e8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sync.go b/sync.go
@@ -175,6 +175,10 @@ func (sync *Syncer) GetDrainState() SyncState {
 }
 
 func (sync *Syncer) Feed(ctx context.Context) (recs protocol.Records, err error) {
+	// other side closed the connection already
+	if sync.GetDrainState() == SendNone {
+		sync.SetFeedState(ctx, SendNone)
+	}
 	switch sync.GetFeedState() {
 	case SendHandshake:
 		recs, err = sync.FeedHandshake()