mirror/act_runner
1
0
Fork 0
mirror of https://gitea.com/gitea/act_runner.git synced 2025-03-21 14:39:55 +01:00
Code Issues Packages Projects Releases Wiki Activity

Catch the panic and print the error

Browse Source
This commit is contained in:
CaiCandong 2023-07-23 22:59:23 +08:00
parent 1a7ec5f339
commit 44f0f2d3fb
1 changed files with 23 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
internal/app/poll/poller.go
View File

@ -6,6 +6,7 @@ package poll
import ( import (
"context" "context"
"errors" "errors"
"fmt"
"sync" "sync"
runnerv1 "code.gitea.io/actions-proto-go/runner/v1" runnerv1 "code.gitea.io/actions-proto-go/runner/v1"
@ -45,19 +46,30 @@ func (p *Poller) Poll(ctx context.Context) {
func (p *Poller) poll(ctx context.Context, wg *sync.WaitGroup, limiter *rate.Limiter) { func (p *Poller) poll(ctx context.Context, wg *sync.WaitGroup, limiter *rate.Limiter) {
defer wg.Done() defer wg.Done()
for { for {
if err := limiter.Wait(ctx); err != nil { p.pollTaskWithRateLimit(ctx, limiter)
if ctx.Err() != nil { }
log.WithError(err).Debug("limiter wait failed") }
}
return func (p *Poller) pollTaskWithRateLimit(ctx context.Context, limiter *rate.Limiter) {
defer func() {
if r := recover(); r != nil {
// we don't know whether log will panic, so just use fmt
fmt.Printf("panic: %v\n", r)
} }
task, ok := p.fetchTask(ctx) }()
if !ok {
continue if err := limiter.Wait(ctx); err != nil {
} if ctx.Err() != nil {
if err := p.runner.Run(ctx, task); err != nil { log.WithError(err).Debug("limiter wait failed")
log.WithError(err).Error("failed to run task")
} }
return
}
task, ok := p.fetchTask(ctx)
if !ok {
return
}
if err := p.runner.Run(ctx, task); err != nil {
log.WithError(err).Error("failed to run task")
} }
} }