safer checks for EOF

Signed-off-by: Shlomi Noach <2607934+shlomi-noach@users.noreply.github.com>
vitessio · Jan 1, 2025 · b204ca6 · b204ca6
1 parent 61c8f5f
commit b204ca6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/go/vt/binlog/binlog_streamer.go b/go/vt/binlog/binlog_streamer.go
@@ -35,6 +35,7 @@ import (
 	"vitess.io/vitess/go/vt/dbconfigs"
 	"vitess.io/vitess/go/vt/log"
 	"vitess.io/vitess/go/vt/sqlparser"
+	"vitess.io/vitess/go/vt/vterrors"
 	"vitess.io/vitess/go/vt/vttablet/tabletserver/schema"
 
 	binlogdatapb "vitess.io/vitess/go/vt/proto/binlogdata"
@@ -279,7 +280,7 @@ func (bls *Streamer) parseEvents(ctx context.Context, events <-chan mysql.Binlog
 				Position:  replication.EncodePosition(pos),
 			}
 			if err = bls.sendTransaction(eventToken, statements); err != nil {
-				if errors.Is(err, io.EOF) {
+				if errors.Is(vterrors.UnwrapAll(err), io.EOF) {
 					return ErrClientEOF
 				}
 				return fmt.Errorf("send reply error: %v", err)