37 lines
805 B
Go
37 lines
805 B
Go
|
// Copyright (C) 2020 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package piecedeletion
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"golang.org/x/sync/semaphore"
|
||
|
|
|
||
|
|
"storj.io/common/pb"
|
||
|
|
)
|
||
|
|
|
||
|
|
// LimitedHandler wraps handler with a concurrency limit.
|
||
|
|
type LimitedHandler struct {
|
||
|
|
active *semaphore.Weighted
|
||
|
|
Handler
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewLimitedHandler wraps handler with a concurrency limit.
|
||
|
|
func NewLimitedHandler(handler Handler, limit int) *LimitedHandler {
|
||
|
|
return &LimitedHandler{
|
||
|
|
active: semaphore.NewWeighted(int64(limit)),
|
||
|
|
Handler: handler,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Handle handles the job queue.
|
||
|
|
func (handler *LimitedHandler) Handle(ctx context.Context, node *pb.Node, queue Queue) {
|
||
|
|
if err := handler.active.Acquire(ctx, 1); err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer handler.active.Release(1)
|
||
|
|
|
||
|
|
handler.Handler.Handle(ctx, node, queue)
|
||
|
|
}
|