12c6d7e83f
* feat: unified storage migrations integration tests * chore: add comment and adjust db path name * chore: refactor test cases into interface * fix: unified SQLite migration with SQLStore migrator * revert changes to newResourceDBProvider
67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package dbimpl
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/grafana/grafana/pkg/storage/unified/sql/db"
|
|
)
|
|
|
|
// NewDB converts a *sql.DB to a db.DB.
|
|
func NewDB(d *sql.DB, driverName string) db.DB {
|
|
ret := sqlDB{
|
|
DB: d,
|
|
driverName: driverName,
|
|
}
|
|
ret.WithTxFunc = db.NewWithTxFunc(ret.BeginTx)
|
|
|
|
return ret
|
|
}
|
|
|
|
type sqlDB struct {
|
|
*sql.DB
|
|
db.WithTxFunc
|
|
driverName string
|
|
}
|
|
|
|
func (d sqlDB) DriverName() string {
|
|
return d.driverName
|
|
}
|
|
|
|
func (d sqlDB) QueryContext(ctx context.Context, query string, args ...any) (db.Rows, error) {
|
|
return d.DB.QueryContext(ctx, query, args...)
|
|
}
|
|
|
|
func (d sqlDB) QueryRowContext(ctx context.Context, query string, args ...any) db.Row {
|
|
return d.DB.QueryRowContext(ctx, query, args...)
|
|
}
|
|
|
|
func (d sqlDB) BeginTx(ctx context.Context, opts *sql.TxOptions) (db.Tx, error) {
|
|
tx, err := d.DB.BeginTx(ctx, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return sqlTx{tx}, err
|
|
}
|
|
|
|
type sqlTx struct {
|
|
*sql.Tx
|
|
}
|
|
|
|
// NewTx wraps an existing *sql.Tx with sqlTx
|
|
func NewTx(tx *sql.Tx) db.Tx {
|
|
return sqlTx{tx}
|
|
}
|
|
|
|
func (tx sqlTx) QueryContext(ctx context.Context, query string, args ...any) (db.Rows, error) {
|
|
// // codeql-suppress go/sql-query-built-from-user-controlled-sources "The query comes from a safe template source
|
|
// and the parameters are passed as arguments."
|
|
return tx.Tx.QueryContext(ctx, query, args...)
|
|
}
|
|
|
|
func (tx sqlTx) QueryRowContext(ctx context.Context, query string, args ...any) db.Row {
|
|
// // codeql-suppress go/sql-query-built-from-user-controlled-sources "The query comes from a safe template source
|
|
// and the parameters are passed as arguments."
|
|
return tx.Tx.QueryRowContext(ctx, query, args...)
|
|
}
|