399d77a0fd
* cleanup dependencies and improve list method * Improve Resource Server API, remove unnecessary dependencies * Reduce the API footprint of ResourceDBInterface and its implementation * Improve LifecycleHooks to use context * Improve testing * reduce API size and improve code * sqltemplate: add DialectForDriver func and improve naming * improve lifecycle API * many small fixes after adding more tests
65 lines
1.2 KiB
Go
65 lines
1.2 KiB
Go
package dbimpl
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/grafana/grafana/pkg/storage/unified/sql/db"
|
|
)
|
|
|
|
func NewDB(d *sql.DB, driverName string) db.DB {
|
|
// remove the suffix from the instrumented driver created by the older
|
|
// Grafana code
|
|
driverName = strings.TrimSuffix(driverName, "WithHooks")
|
|
|
|
return sqldb{
|
|
DB: d,
|
|
driverName: driverName,
|
|
}
|
|
}
|
|
|
|
type sqldb struct {
|
|
*sql.DB
|
|
driverName string
|
|
}
|
|
|
|
func (d sqldb) DriverName() string {
|
|
return d.driverName
|
|
}
|
|
|
|
func (d sqldb) BeginTx(ctx context.Context, opts *sql.TxOptions) (db.Tx, error) {
|
|
t, err := d.DB.BeginTx(ctx, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return tx{
|
|
Tx: t,
|
|
}, nil
|
|
}
|
|
|
|
func (d sqldb) WithTx(ctx context.Context, opts *sql.TxOptions, f db.TxFunc) error {
|
|
t, err := d.BeginTx(ctx, opts)
|
|
if err != nil {
|
|
return fmt.Errorf("begin tx: %w", err)
|
|
}
|
|
|
|
if err := f(ctx, t); err != nil {
|
|
if rollbackErr := t.Rollback(); rollbackErr != nil {
|
|
return fmt.Errorf("tx err: %w; rollback err: %w", err, rollbackErr)
|
|
}
|
|
return fmt.Errorf("tx err: %w", err)
|
|
}
|
|
|
|
if err = t.Commit(); err != nil {
|
|
return fmt.Errorf("commit err: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type tx struct {
|
|
*sql.Tx
|
|
}
|