Jack Christensen
c3e41872a8
This is in preparation for a Begin / Tx interface that will similate nested transactions with savepoints. In addition, this passes the TxOptions struct by value and thereby removes an allocation.
92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package pgxpool
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/jackc/pgconn"
|
|
"github.com/jackc/pgx/v4"
|
|
"github.com/jackc/puddle"
|
|
)
|
|
|
|
// Conn is an acquired *pgx.Conn from a Pool.
|
|
type Conn struct {
|
|
res *puddle.Resource
|
|
p *Pool
|
|
}
|
|
|
|
// Release returns c to the pool it was acquired from. Once Release has been called, other methods must not be called.
|
|
// However, it is safe to call Release multiple times. Subsequent calls after the first will be ignored.
|
|
func (c *Conn) Release() {
|
|
if c.res == nil {
|
|
return
|
|
}
|
|
|
|
conn := c.Conn()
|
|
res := c.res
|
|
c.res = nil
|
|
|
|
now := time.Now()
|
|
if !conn.IsAlive() || conn.PgConn().TxStatus != 'I' || (now.Sub(res.CreationTime()) > c.p.maxConnLifetime) {
|
|
res.Destroy()
|
|
return
|
|
}
|
|
|
|
if c.p.afterRelease == nil {
|
|
res.Release()
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
if c.p.afterRelease(conn) {
|
|
res.Release()
|
|
} else {
|
|
res.Destroy()
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (c *Conn) Exec(ctx context.Context, sql string, arguments ...interface{}) (pgconn.CommandTag, error) {
|
|
return c.Conn().Exec(ctx, sql, arguments...)
|
|
}
|
|
|
|
func (c *Conn) Query(ctx context.Context, sql string, args ...interface{}) (pgx.Rows, error) {
|
|
return c.Conn().Query(ctx, sql, args...)
|
|
}
|
|
|
|
func (c *Conn) QueryRow(ctx context.Context, sql string, args ...interface{}) pgx.Row {
|
|
return c.Conn().QueryRow(ctx, sql, args...)
|
|
}
|
|
|
|
func (c *Conn) SendBatch(ctx context.Context, b *pgx.Batch) pgx.BatchResults {
|
|
return c.Conn().SendBatch(ctx, b)
|
|
}
|
|
|
|
func (c *Conn) CopyFrom(ctx context.Context, tableName pgx.Identifier, columnNames []string, rowSrc pgx.CopyFromSource) (int64, error) {
|
|
return c.Conn().CopyFrom(ctx, tableName, columnNames, rowSrc)
|
|
}
|
|
|
|
func (c *Conn) Begin(ctx context.Context) (*pgx.Tx, error) {
|
|
return c.Conn().Begin(ctx)
|
|
}
|
|
|
|
func (c *Conn) BeginEx(ctx context.Context, txOptions pgx.TxOptions) (*pgx.Tx, error) {
|
|
return c.Conn().BeginEx(ctx, txOptions)
|
|
}
|
|
|
|
func (c *Conn) Conn() *pgx.Conn {
|
|
return c.connResource().conn
|
|
}
|
|
|
|
func (c *Conn) connResource() *connResource {
|
|
return c.res.Value().(*connResource)
|
|
}
|
|
|
|
func (c *Conn) getPoolRow(r pgx.Row) *poolRow {
|
|
return c.connResource().getPoolRow(c, r)
|
|
}
|
|
|
|
func (c *Conn) getPoolRows(r pgx.Rows) *poolRows {
|
|
return c.connResource().getPoolRows(c, r)
|
|
}
|