tdsds.com/gitea
1
1
Fork 0
mirror of https://github.com/go-gitea/gitea synced 2025-10-27 09:28:26 +00:00
Code Releases Activity

feat(bots): move runner status update function to context

Browse Source
This commit is contained in:
fuxiaohei
2022-10-11 20:42:35 +08:00
committed by Jason Song
parent dff0d4b907
commit 9648482ed6
2 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
routers/api/bots/runner/runner.go
View File

@@ -18,7 +18,6 @@ import (
"code.gitea.io/gitea/models/user"
"code.gitea.io/gitea/modules/json"
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/timeutil"
runnerv1 "gitea.com/gitea/proto-go/runner/v1"
"gitea.com/gitea/proto-go/runner/v1/runnerv1connect"
@@ -129,8 +128,6 @@ func (s *Service) Register(
return res, nil
}
const runnerOnlineTimeDeltaSecs = 30
// FetchTask assigns a task to the runner
func (s *Service) FetchTask(
ctx context.Context,
@@ -138,21 +135,6 @@ func (s *Service) FetchTask(
) (*connect.Response[runnerv1.FetchTaskResponse], error) {
runner := GetRunner(ctx)
// update runner online status
if runner.Status == runnerv1.RunnerStatus_RUNNER_STATUS_OFFLINE {
runner.LastOnline = timeutil.TimeStampNow()
runner.Status = runnerv1.RunnerStatus_RUNNER_STATUS_ACTIVE
if err := bots_model.UpdateRunner(ctx, runner, "last_online", "status"); err != nil {
log.Error("can't update runner status: %v", err)
}
}
if timeutil.TimeStampNow()-runner.LastOnline >= runnerOnlineTimeDeltaSecs {
runner.LastOnline = timeutil.TimeStampNow()
if err := bots_model.UpdateRunner(ctx, runner, "last_online"); err != nil {
log.Error("can't update runner last_online: %v", err)
}
}
var task *runnerv1.Task
if t, ok, err := s.pickTask(ctx, runner); err != nil {
return nil, status.Errorf(codes.Internal, "pick task: %v", err)