oss/forgejo-runner
oss/forgejo-runner
1
0
Fork 0
mirror of https://code.forgejo.org/forgejo/runner.git synced 2025-09-30 19:22:09 +00:00
Code Wiki Activity

fix: allow each poller to fetch at FetchInterval

Browse source
This commit is contained in:
Mathieu Fenniak 2025-09-10 13:07:51 -06:00
parent 4d98bda282
commit 59463c7c20
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
internal/app/poll/poller.go
View file

@ -9,6 +9,7 @@ import (
"fmt" "fmt"
"sync" "sync"
"sync/atomic" "sync/atomic"
"time"
runnerv1 "code.forgejo.org/forgejo/actions-proto/runner/v1" runnerv1 "code.forgejo.org/forgejo/actions-proto/runner/v1"
"connectrpc.com/connect" "connectrpc.com/connect"
@ -69,7 +70,11 @@ func (p *poller) init(ctx context.Context, cfg *config.Config, client client.Cli
} }
func (p *poller) Poll() { func (p *poller) Poll() {
limiter := rate.NewLimiter(rate.Every(p.cfg.Runner.FetchInterval), 1) // Limiter is a token bucket, which is refilled at a rate of 1 token every rate.Every(...time...). Our goal is to
// allow each poller to perform a fetch every FetchInterval, so we need to put a token into the bucket every (fetch
// interval / # of pollers).
every := p.cfg.Runner.FetchInterval / time.Duration(max(1, p.cfg.Runner.Capacity))
limiter := rate.NewLimiter(rate.Every(every), 1)
wg := &sync.WaitGroup{} wg := &sync.WaitGroup{}
for i := 0; i < p.cfg.Runner.Capacity; i++ { for i := 0; i < p.cfg.Runner.Capacity; i++ {
wg.Add(1) wg.Add(1)