117 lines
2.2 KiB
Go
117 lines
2.2 KiB
Go
|
package pgxpool
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/jackc/pgx/v5"
|
||
|
|
"github.com/jackc/pgx/v5/pgconn"
|
||
|
|
)
|
||
|
|
|
||
|
|
type errRows struct {
|
||
|
|
err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (errRows) Close() {}
|
||
|
|
func (e errRows) Err() error { return e.err }
|
||
|
|
func (errRows) CommandTag() pgconn.CommandTag { return pgconn.CommandTag{} }
|
||
|
|
func (errRows) FieldDescriptions() []pgconn.FieldDescription { return nil }
|
||
|
|
func (errRows) Next() bool { return false }
|
||
|
|
func (e errRows) Scan(dest ...any) error { return e.err }
|
||
|
|
func (e errRows) Values() ([]any, error) { return nil, e.err }
|
||
|
|
func (e errRows) RawValues() [][]byte { return nil }
|
||
|
|
func (e errRows) Conn() *pgx.Conn { return nil }
|
||
|
|
|
||
|
|
type errRow struct {
|
||
|
|
err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e errRow) Scan(dest ...any) error { return e.err }
|
||
|
|
|
||
|
|
type poolRows struct {
|
||
|
|
r pgx.Rows
|
||
|
|
c *Conn
|
||
|
|
err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rows *poolRows) Close() {
|
||
|
|
rows.r.Close()
|
||
|
|
if rows.c != nil {
|
||
|
|
rows.c.Release()
|
||
|
|
rows.c = nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rows *poolRows) Err() error {
|
||
|
|
if rows.err != nil {
|
||
|
|
return rows.err
|
||
|
|
}
|
||
|
|
return rows.r.Err()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rows *poolRows) CommandTag() pgconn.CommandTag {
|
||
|
|
return rows.r.CommandTag()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rows *poolRows) FieldDescriptions() []pgconn.FieldDescription {
|
||
|
|
return rows.r.FieldDescriptions()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rows *poolRows) Next() bool {
|
||
|
|
if rows.err != nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
n := rows.r.Next()
|
||
|
|
if !n {
|
||
|
|
rows.Close()
|
||
|
|
}
|
||
|
|
return n
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rows *poolRows) Scan(dest ...any) error {
|
||
|
|
err := rows.r.Scan(dest...)
|
||
|
|
if err != nil {
|
||
|
|
rows.Close()
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rows *poolRows) Values() ([]any, error) {
|
||
|
|
values, err := rows.r.Values()
|
||
|
|
if err != nil {
|
||
|
|
rows.Close()
|
||
|
|
}
|
||
|
|
return values, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rows *poolRows) RawValues() [][]byte {
|
||
|
|
return rows.r.RawValues()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rows *poolRows) Conn() *pgx.Conn {
|
||
|
|
return rows.r.Conn()
|
||
|
|
}
|
||
|
|
|
||
|
|
type poolRow struct {
|
||
|
|
r pgx.Row
|
||
|
|
c *Conn
|
||
|
|
err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (row *poolRow) Scan(dest ...any) error {
|
||
|
|
if row.err != nil {
|
||
|
|
return row.err
|
||
|
|
}
|
||
|
|
|
||
|
|
panicked := true
|
||
|
|
defer func() {
|
||
|
|
if panicked && row.c != nil {
|
||
|
|
row.c.Release()
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
err := row.r.Scan(dest...)
|
||
|
|
panicked = false
|
||
|
|
if row.c != nil {
|
||
|
|
row.c.Release()
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|