70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
// +build go1.8
|
|
|
|
package squirrel
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/lann/builder"
|
|
)
|
|
|
|
func (d *selectData) ExecContext(ctx context.Context) (sql.Result, error) {
|
|
if d.RunWith == nil {
|
|
return nil, RunnerNotSet
|
|
}
|
|
ctxRunner, ok := d.RunWith.(ExecerContext)
|
|
if !ok {
|
|
return nil, NoContextSupport
|
|
}
|
|
return ExecContextWith(ctx, ctxRunner, d)
|
|
}
|
|
|
|
func (d *selectData) QueryContext(ctx context.Context) (*sql.Rows, error) {
|
|
if d.RunWith == nil {
|
|
return nil, RunnerNotSet
|
|
}
|
|
ctxRunner, ok := d.RunWith.(QueryerContext)
|
|
if !ok {
|
|
return nil, NoContextSupport
|
|
}
|
|
return QueryContextWith(ctx, ctxRunner, d)
|
|
}
|
|
|
|
func (d *selectData) QueryRowContext(ctx context.Context) RowScanner {
|
|
if d.RunWith == nil {
|
|
return &Row{err: RunnerNotSet}
|
|
}
|
|
queryRower, ok := d.RunWith.(QueryRowerContext)
|
|
if !ok {
|
|
if _, ok := d.RunWith.(QueryerContext); !ok {
|
|
return &Row{err: RunnerNotQueryRunner}
|
|
}
|
|
return &Row{err: NoContextSupport}
|
|
}
|
|
return QueryRowContextWith(ctx, queryRower, d)
|
|
}
|
|
|
|
// ExecContext builds and ExecContexts the query with the Runner set by RunWith.
|
|
func (b SelectBuilder) ExecContext(ctx context.Context) (sql.Result, error) {
|
|
data := builder.GetStruct(b).(selectData)
|
|
return data.ExecContext(ctx)
|
|
}
|
|
|
|
// QueryContext builds and QueryContexts the query with the Runner set by RunWith.
|
|
func (b SelectBuilder) QueryContext(ctx context.Context) (*sql.Rows, error) {
|
|
data := builder.GetStruct(b).(selectData)
|
|
return data.QueryContext(ctx)
|
|
}
|
|
|
|
// QueryRowContext builds and QueryRowContexts the query with the Runner set by RunWith.
|
|
func (b SelectBuilder) QueryRowContext(ctx context.Context) RowScanner {
|
|
data := builder.GetStruct(b).(selectData)
|
|
return data.QueryRowContext(ctx)
|
|
}
|
|
|
|
// ScanContext is a shortcut for QueryRowContext().Scan.
|
|
func (b SelectBuilder) ScanContext(ctx context.Context, dest ...interface{}) error {
|
|
return b.QueryRowContext(ctx).Scan(dest...)
|
|
}
|