diff options
Diffstat (limited to 'internal/executor/executor.go')
| -rw-r--r-- | internal/executor/executor.go | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/internal/executor/executor.go b/internal/executor/executor.go index 9dd37ae..1c9e667 100644 --- a/internal/executor/executor.go +++ b/internal/executor/executor.go @@ -179,6 +179,9 @@ func (p *Pool) executeResume(ctx context.Context, t *task.Task, exec *storage.Ex p.mu.Lock() p.active-- p.activePerAgent[agentType]-- + if p.activePerAgent[agentType] == 0 { + delete(p.activePerAgent, agentType) + } p.mu.Unlock() select { case p.doneCh <- struct{}{}: @@ -296,6 +299,9 @@ func (p *Pool) execute(ctx context.Context, t *task.Task) { now := time.Now() for agent := range p.runners { activeTasks[agent] = p.activePerAgent[agent] + if deadline, ok := p.rateLimited[agent]; ok && now.After(deadline) { + delete(p.rateLimited, agent) + } activeUntil := p.rateLimited[agent] isLimited := now.Before(activeUntil) rateLimited[agent] = isLimited @@ -326,6 +332,9 @@ func (p *Pool) execute(ctx context.Context, t *task.Task) { } p.mu.Lock() + if deadline, ok := p.rateLimited[agentType]; ok && time.Now().After(deadline) { + delete(p.rateLimited, agentType) + } p.activePerAgent[agentType]++ p.mu.Unlock() @@ -333,6 +342,9 @@ func (p *Pool) execute(ctx context.Context, t *task.Task) { p.mu.Lock() p.active-- p.activePerAgent[agentType]-- + if p.activePerAgent[agentType] == 0 { + delete(p.activePerAgent, agentType) + } p.mu.Unlock() select { case p.doneCh <- struct{}{}: |