Allow batch callback function to override error

https://github.com/jackc/pgx/pull/1538#issuecomment-1486083411
pull/1565/head
Jack Christensen 2023-03-31 20:18:05 -05:00
parent ca022267db
commit be79f1c8f5
1 changed files with 2 additions and 2 deletions

View File

@ -225,7 +225,7 @@ func (br *batchResults) Close() error {
for br.err == nil && !br.closed && br.b != nil && br.qqIdx < len(br.b.queuedQueries) {
if br.b.queuedQueries[br.qqIdx].fn != nil {
err := br.b.queuedQueries[br.qqIdx].fn(br)
if err != nil && br.err == nil {
if err != nil {
br.err = err
}
} else {
@ -398,7 +398,7 @@ func (br *pipelineBatchResults) Close() error {
for br.err == nil && !br.closed && br.b != nil && br.qqIdx < len(br.b.queuedQueries) {
if br.b.queuedQueries[br.qqIdx].fn != nil {
err := br.b.queuedQueries[br.qqIdx].fn(br)
if err != nil && br.err == nil {
if err != nil {
br.err = err
}
} else {