34283a74e8
Fix #19513 This PR introduce a new db method `InTransaction(context.Context)`, and also builtin check on `db.TxContext` and `db.WithTx`. There is also a new method `db.AutoTx` has been introduced but could be used by other PRs. `WithTx` will always open a new transaction, if a transaction exist in context, return an error. `AutoTx` will try to open a new transaction if no transaction exist in context. That means it will always enter a transaction if there is no error. Co-authored-by: delvh <dev.lh@web.de> Co-authored-by: 6543 <6543@obermui.de>
78 lines
1.7 KiB
Go
78 lines
1.7 KiB
Go
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package db
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"code.gitea.io/gitea/modules/util"
|
|
)
|
|
|
|
var ErrAlreadyInTransaction = errors.New("database connection has already been in a transaction")
|
|
|
|
// ErrCancelled represents an error due to context cancellation
|
|
type ErrCancelled struct {
|
|
Message string
|
|
}
|
|
|
|
// IsErrCancelled checks if an error is a ErrCancelled.
|
|
func IsErrCancelled(err error) bool {
|
|
_, ok := err.(ErrCancelled)
|
|
return ok
|
|
}
|
|
|
|
func (err ErrCancelled) Error() string {
|
|
return "Cancelled: " + err.Message
|
|
}
|
|
|
|
// ErrCancelledf returns an ErrCancelled for the provided format and args
|
|
func ErrCancelledf(format string, args ...interface{}) error {
|
|
return ErrCancelled{
|
|
fmt.Sprintf(format, args...),
|
|
}
|
|
}
|
|
|
|
// ErrSSHDisabled represents an "SSH disabled" error.
|
|
type ErrSSHDisabled struct{}
|
|
|
|
// IsErrSSHDisabled checks if an error is a ErrSSHDisabled.
|
|
func IsErrSSHDisabled(err error) bool {
|
|
_, ok := err.(ErrSSHDisabled)
|
|
return ok
|
|
}
|
|
|
|
func (err ErrSSHDisabled) Error() string {
|
|
return "SSH is disabled"
|
|
}
|
|
|
|
// ErrNotExist represents a non-exist error.
|
|
type ErrNotExist struct {
|
|
Resource string
|
|
ID int64
|
|
}
|
|
|
|
// IsErrNotExist checks if an error is an ErrNotExist
|
|
func IsErrNotExist(err error) bool {
|
|
_, ok := err.(ErrNotExist)
|
|
return ok
|
|
}
|
|
|
|
func (err ErrNotExist) Error() string {
|
|
name := "record"
|
|
if err.Resource != "" {
|
|
name = err.Resource
|
|
}
|
|
|
|
if err.ID != 0 {
|
|
return fmt.Sprintf("%s does not exist [id: %d]", name, err.ID)
|
|
}
|
|
return fmt.Sprintf("%s does not exist", name)
|
|
}
|
|
|
|
// Unwrap unwraps this as a ErrNotExist err
|
|
func (err ErrNotExist) Unwrap() error {
|
|
return util.ErrNotExist
|
|
}
|