fix(helper): QueryBy()没有实现ctx中断

joyparty · Jun 30, 2023 · a4aeddc · a4aeddc
1 parent 3253d64
commit a4aeddc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/helper.go b/helper.go
@@ -141,6 +141,12 @@ func QueryBy(ctx context.Context, db DB, stmt *goqu.SelectDataset, fn func(ctx c
 	defer rows.Close()
 
 	for rows.Next() {
+		select {
+		case <-ctx.Done():
+			return ctx.Err()
+		default:
+		}
+
 		if err := fn(ctx, rows); err != nil {
 			return fmt.Errorf("handle row, %w", err)
 		}