domi/forgejo-domi
1
0
Fork 0
forked from forgejo/forgejo
Code Pull requests Activity

Next round of db.DefaultContext refactor (#27089)

Browse source 
Part of #27065
This commit is contained in:
JakobDev 2023-09-16 16:39:12 +02:00 • committed by GitHub
parent a1b2a11812
commit f91dbbba98
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
90 changed files with 434 additions and 464 deletions
Download patch file Download diff file Expand all files Collapse all files

21
services/task/migrate.go
View file

@ -4,6 +4,7 @@
package task
import (
"context"
"errors"
"fmt"
"strings"
@ -40,7 +41,7 @@ func handleCreateError(owner *user_model.User, err error) error {
}
}
func runMigrateTask(t *admin_model.Task) (err error) {
func runMigrateTask(ctx context.Context, t *admin_model.Task) (err error) {
defer func() {
if e := recover(); e != nil {
err = fmt.Errorf("PANIC whilst trying to do migrate task: %v", e)
@ -48,9 +49,9 @@ func runMigrateTask(t *admin_model.Task) (err error) {
}
if err == nil {
err = admin_model.FinishMigrateTask(t)
err = admin_model.FinishMigrateTask(ctx, t)
if err == nil {
notify_service.MigrateRepository(db.DefaultContext, t.Doer, t.Owner, t.Repo)
notify_service.MigrateRepository(ctx, t.Doer, t.Owner, t.Repo)
return
}
@ -63,14 +64,14 @@ func runMigrateTask(t *admin_model.Task) (err error) {
t.Status = structs.TaskStatusFailed
t.Message = err.Error()
if err := t.UpdateCols("status", "message", "end_time"); err != nil {
if err := t.UpdateCols(ctx, "status", "message", "end_time"); err != nil {
log.Error("Task UpdateCols failed: %v", err)
}
// then, do not delete the repository, otherwise the users won't be able to see the last error
}()
if err = t.LoadRepo(); err != nil {
if err = t.LoadRepo(ctx); err != nil {
return err
}
@ -79,10 +80,10 @@ func runMigrateTask(t *admin_model.Task) (err error) {
return nil
}
if err = t.LoadDoer(); err != nil {
if err = t.LoadDoer(ctx); err != nil {
return err
}
if err = t.LoadOwner(); err != nil {
if err = t.LoadOwner(ctx); err != nil {
return err
}
@ -100,7 +101,7 @@ func runMigrateTask(t *admin_model.Task) (err error) {
t.StartTime = timeutil.TimeStampNow()
t.Status = structs.TaskStatusRunning
if err = t.UpdateCols("start_time", "status"); err != nil {
if err = t.UpdateCols(ctx, "start_time", "status"); err != nil {
return err
}
@ -112,7 +113,7 @@ func runMigrateTask(t *admin_model.Task) (err error) {
case <-ctx.Done():
return
}
task, _ := admin_model.GetMigratingTask(t.RepoID)
task, _ := admin_model.GetMigratingTask(ctx, t.RepoID)
if task != nil && task.Status != structs.TaskStatusRunning {
log.Debug("MigrateTask[%d] by DoerID[%d] to RepoID[%d] for OwnerID[%d] is canceled due to status is not 'running'", t.ID, t.DoerID, t.RepoID, t.OwnerID)
cancel()
@ -128,7 +129,7 @@ func runMigrateTask(t *admin_model.Task) (err error) {
}
bs, _ := json.Marshal(message)
t.Message = string(bs)
_ = t.UpdateCols("message")
_ = t.UpdateCols(ctx, "message")
})
if err == nil {