go template

src/with/pg.go

@@ -42,7 +42,10 @@ func PSQL(ctx context.Context, conn string, foo func(db *sql.DB) error) error {
 				if ok {
 					return
 				}
-			case <-time.After(time.Second * 5):
+			}
+			select {
+			case <-ctx.Done():
+			case <-time.After(time.Second):
 			}
 		}
 	}()