minor, see prev commit

kataras · Oct 28, 2023 · 979ffb5 · 979ffb5
1 parent e75bffa
commit 979ffb5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/db_table_listener.go b/db_table_listener.go
@@ -3,7 +3,9 @@ package pg
 import (
 	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
+	"io"
 	"sync/atomic"
 )
 
@@ -111,8 +113,12 @@ func (db *DB) ListenTable(ctx context.Context, table string, callback func(Table
 		for {
 			var evt TableNotificationJSON
 
-			notification, err := conn.Accept(context.Background())
+			notification, err := conn.Accept(ctx)
 			if err != nil {
+				if errors.Is(err, io.ErrUnexpectedEOF) {
+					return // may produced by close.
+				}
+
 				if callback(evt, err) != nil {
 					return
 				}