c34ad62eea
* Mulitple Gitea Doctor improvements (#10943) Backport #10943 * Add `gitea doctor --list` flag to list the checks that will be run, including those by default * Add `gitea doctor --run` to run specific checks * Add `gitea doctor --all` to run all checks * Add db version checker * Add non-default recalculate merge bases check/fixer to doctor * Add hook checker (Fix #9878) and ensure hooks are executable (Fix #6319) * Fix authorized_keys checker - slight change of functionality here because parsing the command is fragile and we should just check if the authorized_keys file is essentially the same as what gitea would produce. (This is still not perfect as order matters - we should probably just md5sum the two files.) * Add SCRIPT_TYPE check (Fix #10977) * Add `gitea doctor --fix` to attempt to fix what is possible to easily fix * Add `gitea doctor --log-file` to set the log-file, be it a file, stdout or to switch off completely. (Fixes previously undetected bug with certain xorm logging configurations - see @6543 comment.) Signed-off-by: Andrew Thornton <art27@cantab.net> * Switch to io.Writer instead of io.StringWriter Signed-off-by: Andrew Thornton <art27@cantab.net>
68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
// Copyright 2019 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 models
|
|
|
|
import (
|
|
"code.gitea.io/gitea/modules/setting"
|
|
"xorm.io/builder"
|
|
)
|
|
|
|
// DBContext represents a db context
|
|
type DBContext struct {
|
|
e Engine
|
|
}
|
|
|
|
// DefaultDBContext represents a DBContext with default Engine
|
|
func DefaultDBContext() DBContext {
|
|
return DBContext{x}
|
|
}
|
|
|
|
// Committer represents an interface to Commit or Close the dbcontext
|
|
type Committer interface {
|
|
Commit() error
|
|
Close()
|
|
}
|
|
|
|
// TxDBContext represents a transaction DBContext
|
|
func TxDBContext() (DBContext, Committer, error) {
|
|
sess := x.NewSession()
|
|
if err := sess.Begin(); err != nil {
|
|
sess.Close()
|
|
return DBContext{}, nil, err
|
|
}
|
|
|
|
return DBContext{sess}, sess, nil
|
|
}
|
|
|
|
// WithContext represents executing database operations
|
|
func WithContext(f func(ctx DBContext) error) error {
|
|
return f(DBContext{x})
|
|
}
|
|
|
|
// WithTx represents executing database operations on a trasaction
|
|
func WithTx(f func(ctx DBContext) error) error {
|
|
sess := x.NewSession()
|
|
if err := sess.Begin(); err != nil {
|
|
sess.Close()
|
|
return err
|
|
}
|
|
|
|
if err := f(DBContext{sess}); err != nil {
|
|
sess.Close()
|
|
return err
|
|
}
|
|
|
|
err := sess.Commit()
|
|
sess.Close()
|
|
return err
|
|
}
|
|
|
|
// Iterate iterates the databases and doing something
|
|
func Iterate(ctx DBContext, tableBean interface{}, cond builder.Cond, fun func(idx int, bean interface{}) error) error {
|
|
return ctx.e.Where(cond).
|
|
BufferSize(setting.Database.IterateBufferSize).
|
|
Iterate(tableBean, fun)
|
|
}
|