108 lines
3.0 KiB
Go
108 lines
3.0 KiB
Go
package pgx
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/jackc/pgx/v5/pgconn"
|
|
)
|
|
|
|
// QueryTracer traces Query, QueryRow, and Exec.
|
|
type QueryTracer interface {
|
|
// TraceQueryStart is called at the beginning of Query, QueryRow, and Exec calls. The returned context is used for the
|
|
// rest of the call and will be passed to TraceQueryEnd.
|
|
TraceQueryStart(ctx context.Context, conn *Conn, data TraceQueryStartData) context.Context
|
|
|
|
TraceQueryEnd(ctx context.Context, conn *Conn, data TraceQueryEndData)
|
|
}
|
|
|
|
type TraceQueryStartData struct {
|
|
SQL string
|
|
Args []any
|
|
}
|
|
|
|
type TraceQueryEndData struct {
|
|
CommandTag pgconn.CommandTag
|
|
Err error
|
|
}
|
|
|
|
// BatchTracer traces SendBatch.
|
|
type BatchTracer interface {
|
|
// TraceBatchStart is called at the beginning of SendBatch calls. The returned context is used for the
|
|
// rest of the call and will be passed to TraceBatchQuery and TraceBatchEnd.
|
|
TraceBatchStart(ctx context.Context, conn *Conn, data TraceBatchStartData) context.Context
|
|
|
|
TraceBatchQuery(ctx context.Context, conn *Conn, data TraceBatchQueryData)
|
|
TraceBatchEnd(ctx context.Context, conn *Conn, data TraceBatchEndData)
|
|
}
|
|
|
|
type TraceBatchStartData struct {
|
|
Batch *Batch
|
|
}
|
|
|
|
type TraceBatchQueryData struct {
|
|
SQL string
|
|
Args []any
|
|
CommandTag pgconn.CommandTag
|
|
Err error
|
|
}
|
|
|
|
type TraceBatchEndData struct {
|
|
Err error
|
|
}
|
|
|
|
// CopyFromTracer traces CopyFrom.
|
|
type CopyFromTracer interface {
|
|
// TraceCopyFromStart is called at the beginning of CopyFrom calls. The returned context is used for the
|
|
// rest of the call and will be passed to TraceCopyFromEnd.
|
|
TraceCopyFromStart(ctx context.Context, conn *Conn, data TraceCopyFromStartData) context.Context
|
|
|
|
TraceCopyFromEnd(ctx context.Context, conn *Conn, data TraceCopyFromEndData)
|
|
}
|
|
|
|
type TraceCopyFromStartData struct {
|
|
TableName Identifier
|
|
ColumnNames []string
|
|
}
|
|
|
|
type TraceCopyFromEndData struct {
|
|
CommandTag pgconn.CommandTag
|
|
Err error
|
|
}
|
|
|
|
// PrepareTracer traces Prepare.
|
|
type PrepareTracer interface {
|
|
// TracePrepareStart is called at the beginning of Prepare calls. The returned context is used for the
|
|
// rest of the call and will be passed to TracePrepareEnd.
|
|
TracePrepareStart(ctx context.Context, conn *Conn, data TracePrepareStartData) context.Context
|
|
|
|
TracePrepareEnd(ctx context.Context, conn *Conn, data TracePrepareEndData)
|
|
}
|
|
|
|
type TracePrepareStartData struct {
|
|
Name string
|
|
SQL string
|
|
}
|
|
|
|
type TracePrepareEndData struct {
|
|
AlreadyPrepared bool
|
|
Err error
|
|
}
|
|
|
|
// ConnectTracer traces Connect and ConnectConfig.
|
|
type ConnectTracer interface {
|
|
// TraceConnectStart is called at the beginning of Connect and ConnectConfig calls. The returned context is used for
|
|
// the rest of the call and will be passed to TraceConnectEnd.
|
|
TraceConnectStart(ctx context.Context, data TraceConnectStartData) context.Context
|
|
|
|
TraceConnectEnd(ctx context.Context, data TraceConnectEndData)
|
|
}
|
|
|
|
type TraceConnectStartData struct {
|
|
ConnConfig *ConnConfig
|
|
}
|
|
|
|
type TraceConnectEndData struct {
|
|
Conn *Conn
|
|
Err error
|
|
}
|