luckyimnotanigger/forgejo
1
0
Fork 0
mirror of https://codeberg.org/forgejo/forgejo.git synced 2025-05-16 06:52:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

Move db related basic functions to models/db (#17075)

Browse source 
* Move db related basic functions to models/db

* Fix lint

* Fix lint

* Fix test

* Fix lint

* Fix lint

* revert unnecessary change

* Fix test

* Fix wrong replace string

* Use *Context

* Correct committer spelling and fix wrong replaced words

Co-authored-by: zeripath <art27@cantab.net>
This commit is contained in:
Lunny Xiao 2021-09-19 19:49:59 +08:00 • committed by GitHub
parent 462306e263
commit a4bfef265d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
335 changed files with 4191 additions and 3654 deletions
Download patch file Download diff file Expand all files Collapse all files

27
models/task.go
View file

@ -7,6 +7,7 @@ package models
import (
"fmt"
"code.gitea.io/gitea/models/db"
"code.gitea.io/gitea/modules/json"
migration "code.gitea.io/gitea/modules/migrations/base"
"code.gitea.io/gitea/modules/secret"
@ -36,6 +37,10 @@ type Task struct {
Created timeutil.TimeStamp `xorm:"created"`
}
func init() {
db.RegisterModel(new(Task))
}
// TranslatableMessage represents JSON struct that can be translated with a Locale
type TranslatableMessage struct {
Format string
@ -44,10 +49,10 @@ type TranslatableMessage struct {
// LoadRepo loads repository of the task
func (task *Task) LoadRepo() error {
return task.loadRepo(x)
return task.loadRepo(db.DefaultContext().Engine())
}
func (task *Task) loadRepo(e Engine) error {
func (task *Task) loadRepo(e db.Engine) error {
if task.Repo != nil {
return nil
}
@ -71,7 +76,7 @@ func (task *Task) LoadDoer() error {
}
var doer User
has, err := x.ID(task.DoerID).Get(&doer)
has, err := db.DefaultContext().Engine().ID(task.DoerID).Get(&doer)
if err != nil {
return err
} else if !has {
@ -91,7 +96,7 @@ func (task *Task) LoadOwner() error {
}
var owner User
has, err := x.ID(task.OwnerID).Get(&owner)
has, err := db.DefaultContext().Engine().ID(task.OwnerID).Get(&owner)
if err != nil {
return err
} else if !has {
@ -106,7 +111,7 @@ func (task *Task) LoadOwner() error {
// UpdateCols updates some columns
func (task *Task) UpdateCols(cols ...string) error {
_, err := x.ID(task.ID).Cols(cols...).Update(task)
_, err := db.DefaultContext().Engine().ID(task.ID).Cols(cols...).Update(task)
return err
}
@ -165,7 +170,7 @@ func GetMigratingTask(repoID int64) (*Task, error) {
RepoID: repoID,
Type: structs.TaskTypeMigrateRepo,
}
has, err := x.Get(&task)
has, err := db.DefaultContext().Engine().Get(&task)
if err != nil {
return nil, err
} else if !has {
@ -181,7 +186,7 @@ func GetMigratingTaskByID(id, doerID int64) (*Task, *migration.MigrateOptions, e
DoerID: doerID,
Type: structs.TaskTypeMigrateRepo,
}
has, err := x.Get(&task)
has, err := db.DefaultContext().Engine().Get(&task)
if err != nil {
return nil, nil, err
} else if !has {
@ -212,16 +217,16 @@ func (opts FindTaskOptions) ToConds() builder.Cond {
// FindTasks find all tasks
func FindTasks(opts FindTaskOptions) ([]*Task, error) {
tasks := make([]*Task, 0, 10)
err := x.Where(opts.ToConds()).Find(&tasks)
err := db.DefaultContext().Engine().Where(opts.ToConds()).Find(&tasks)
return tasks, err
}
// CreateTask creates a task on database
func CreateTask(task *Task) error {
return createTask(x, task)
return createTask(db.DefaultContext().Engine(), task)
}
func createTask(e Engine, task *Task) error {
func createTask(e db.Engine, task *Task) error {
_, err := e.Insert(task)
return err
}
@ -248,7 +253,7 @@ func FinishMigrateTask(task *Task) error {
}
task.PayloadContent = string(confBytes)
sess := x.NewSession()
sess := db.DefaultContext().NewSession()
defer sess.Close()
if err := sess.Begin(); err != nil {
return err