0f4df23290
Some checks failed
kotlin / test (push) Has been skipped
python / test (push) Has been skipped
typescript / test (push) Has been skipped
go / build darwin/amd64 (push) Has been cancelled
go / build linux/amd64 (push) Has been cancelled
go / build windows/amd64 (push) Has been cancelled
go / build darwin/arm64 (push) Has been cancelled
go / build linux/arm64 (push) Has been cancelled
go / build windows/arm64 (push) Has been cancelled
go / test (push) Has been cancelled
go / vuln_check (push) Has been cancelled
38 lines
855 B
Cheetah
38 lines
855 B
Cheetah
{{define "dbCodeTemplatePgx"}}
|
|
|
|
type DBTX interface {
|
|
Exec(context.Context, string, ...any) (pgconn.CommandTag, error)
|
|
Query(context.Context, string, ...any) (pgx.Rows, error)
|
|
QueryRow(context.Context, string, ...any) pgx.Row
|
|
{{- if .UsesCopyFrom }}
|
|
CopyFrom(ctx context.Context, tableName pgx.Identifier, columnNames []string, rowSrc pgx.CopyFromSource) (int64, error)
|
|
{{- end }}
|
|
{{- if .UsesBatch }}
|
|
SendBatch(context.Context, *pgx.Batch) pgx.BatchResults
|
|
{{- end }}
|
|
}
|
|
|
|
{{ if .EmitMethodsWithDBArgument}}
|
|
func New() *Queries {
|
|
return &Queries{}
|
|
{{- else -}}
|
|
func New(db DBTX) *Queries {
|
|
return &Queries{db: db}
|
|
{{- end}}
|
|
}
|
|
|
|
type Queries struct {
|
|
{{if not .EmitMethodsWithDBArgument}}
|
|
db DBTX
|
|
{{end}}
|
|
}
|
|
|
|
{{if not .EmitMethodsWithDBArgument}}
|
|
func (q *Queries) WithTx(tx pgx.Tx) *Queries {
|
|
return &Queries{
|
|
db: tx,
|
|
}
|
|
}
|
|
{{end}}
|
|
{{end}}
|