68 lines
1.7 KiB
Go
68 lines
1.7 KiB
Go
|
package sqldb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type connOrTx interface {
|
||
|
|
executer
|
||
|
|
queryPerformer
|
||
|
|
}
|
||
|
|
|
||
|
|
type executer interface {
|
||
|
|
// Exec executes a query and does not expect any result.
|
||
|
|
Exec(ctx context.Context, query string, args ...interface{}) ExecResult
|
||
|
|
ExecNamed(ctx context.Context, query string, arg interface{}) ExecResult
|
||
|
|
}
|
||
|
|
|
||
|
|
type queryPerformer interface {
|
||
|
|
// Query executes a query and returns a result object that can later be used
|
||
|
|
// to retrieve values.
|
||
|
|
Query(ctx context.Context, query string, args ...interface{}) QueryResult
|
||
|
|
QueryNamed(ctx context.Context, query string, arg interface{}) QueryResult
|
||
|
|
}
|
||
|
|
|
||
|
|
type stdConnOrTx interface {
|
||
|
|
ExecContext(ctx context.Context, query string, args ...interface{}) (sql.Result, error)
|
||
|
|
QueryContext(ctx context.Context, query string, args ...interface{}) (*sql.Rows, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
type caller struct {
|
||
|
|
db stdConnOrTx
|
||
|
|
cb *callbacks
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *caller) Exec(ctx context.Context, query string, args ...interface{}) ExecResult {
|
||
|
|
c.cb.callBefore(ctx, query)
|
||
|
|
startedAt := time.Now()
|
||
|
|
res, err := c.db.ExecContext(ctx, query, args...)
|
||
|
|
c.cb.callAfter(ctx, query, time.Since(startedAt), err)
|
||
|
|
|
||
|
|
return &execResult{
|
||
|
|
db: c,
|
||
|
|
err: err,
|
||
|
|
res: res,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *caller) ExecNamed(ctx context.Context, query string, arg interface{}) ExecResult {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *caller) Query(ctx context.Context, query string, args ...interface{}) QueryResult {
|
||
|
|
c.cb.callBefore(ctx, query)
|
||
|
|
startedAt := time.Now()
|
||
|
|
rows, err := c.db.QueryContext(ctx, query, args...)
|
||
|
|
c.cb.callAfter(ctx, query, time.Since(startedAt), err)
|
||
|
|
return &queryResult{
|
||
|
|
err: err,
|
||
|
|
rows: rows,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *caller) QueryNamed(ctx context.Context, query string, arg interface{}) QueryResult {
|
||
|
|
return nil
|
||
|
|
}
|