59d06644b9
Also added temporary types withRebind and withTagTx, which will be later removed. Currently they help to avoid changing the whole codebase at the same time. Change-Id: I7f07ba8f4709a23a463bfa67464628665a05808f
48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package migrate
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"database/sql/driver"
|
|
|
|
"storj.io/storj/private/dbutil/txutil"
|
|
"storj.io/storj/private/tagsql"
|
|
)
|
|
|
|
// DB is the minimal implementation that is needed by migrations.
|
|
//
|
|
// DB can optionally have `Rebind(string) string` for translating `? queries for the specific database.
|
|
type DB interface {
|
|
BeginTx(ctx context.Context, txOptions *sql.TxOptions) (tagsql.Tx, error)
|
|
Driver() driver.Driver
|
|
}
|
|
|
|
// DBX contains additional methods for migrations.
|
|
type DBX interface {
|
|
DB
|
|
Schema() string
|
|
Rebind(string) string
|
|
}
|
|
|
|
// rebind uses Rebind method when the database has the func.
|
|
func rebind(db DB, s string) string {
|
|
if dbx, ok := db.(interface{ Rebind(string) string }); ok {
|
|
return dbx.Rebind(s)
|
|
}
|
|
return s
|
|
}
|
|
|
|
// WithTx runs the given callback in the context of a transaction.
|
|
func WithTx(ctx context.Context, db DB, fn func(ctx context.Context, tx tagsql.Tx) error) error {
|
|
tx, err := db.BeginTx(ctx, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return txutil.ExecuteInTx(ctx, db.Driver(), tx, func() error {
|
|
return fn(ctx, tx)
|
|
})
|
|
}
|