mirror of
https://github.com/grafana/grafana.git
synced 2024-12-02 13:39:19 -06:00
89 lines
2.7 KiB
Go
89 lines
2.7 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/mattn/go-sqlite3"
|
|
"xorm.io/xorm"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
)
|
|
|
|
var tsclogger = log.New("sqlstore.transactions")
|
|
|
|
// WithTransactionalDbSession calls the callback with a session within a transaction.
|
|
func (ss *SQLStore) WithTransactionalDbSession(ctx context.Context, callback DBTransactionFunc) error {
|
|
return inTransactionWithRetryCtx(ctx, ss.engine, ss.bus, callback, 0)
|
|
}
|
|
|
|
func (ss *SQLStore) InTransaction(ctx context.Context, fn func(ctx context.Context) error) error {
|
|
return ss.inTransactionWithRetry(ctx, fn, 0)
|
|
}
|
|
|
|
func (ss *SQLStore) inTransactionWithRetry(ctx context.Context, fn func(ctx context.Context) error, retry int) error {
|
|
return inTransactionWithRetryCtx(ctx, ss.engine, ss.bus, func(sess *DBSession) error {
|
|
withValue := context.WithValue(ctx, ContextSessionKey{}, sess)
|
|
return fn(withValue)
|
|
}, retry)
|
|
}
|
|
|
|
func inTransactionWithRetryCtx(ctx context.Context, engine *xorm.Engine, bus bus.Bus, callback DBTransactionFunc, retry int) error {
|
|
sess, isNew, err := startSessionOrUseExisting(ctx, engine, true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !sess.transactionOpen && !isNew {
|
|
// this should not happen because the only place that creates reusable session begins a new transaction.
|
|
return fmt.Errorf("cannot reuse existing session that did not start transaction")
|
|
}
|
|
|
|
if isNew { // if this call initiated the session, it should be responsible for closing it.
|
|
defer sess.Close()
|
|
}
|
|
|
|
err = callback(sess)
|
|
|
|
if !isNew {
|
|
tsclogger.Debug("skip committing the transaction because it belongs to a session created in the outer scope")
|
|
// Do not commit the transaction if the session was reused.
|
|
return err
|
|
}
|
|
|
|
// special handling of database locked errors for sqlite, then we can retry 5 times
|
|
var sqlError sqlite3.Error
|
|
if errors.As(err, &sqlError) && retry < 5 && (sqlError.Code == sqlite3.ErrLocked || sqlError.Code == sqlite3.ErrBusy) {
|
|
if rollErr := sess.Rollback(); rollErr != nil {
|
|
return fmt.Errorf("rolling back transaction due to error failed: %s: %w", rollErr, err)
|
|
}
|
|
|
|
time.Sleep(time.Millisecond * time.Duration(10))
|
|
sqlog.Info("Database locked, sleeping then retrying", "error", err, "retry", retry)
|
|
return inTransactionWithRetryCtx(ctx, engine, bus, callback, retry+1)
|
|
}
|
|
|
|
if err != nil {
|
|
if rollErr := sess.Rollback(); rollErr != nil {
|
|
return fmt.Errorf("rolling back transaction due to error failed: %s: %w", rollErr, err)
|
|
}
|
|
return err
|
|
}
|
|
if err := sess.Commit(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(sess.events) > 0 {
|
|
for _, e := range sess.events {
|
|
if err = bus.Publish(ctx, e); err != nil {
|
|
tsclogger.Error("Failed to publish event after commit.", "error", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|