[transfer] avoid setting limiters when max is 0

Signed-off-by: Derek McGowan <derek@mcg.dev>
dmcgowan · May 3, 2023 · d56466c · d56466c
1 parent a7ceac8
commit d56466c
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/pkg/transfer/local/transfer.go b/pkg/transfer/local/transfer.go
@@ -46,14 +46,19 @@ type localTransferService struct {
 }
 
 func NewTransferService(lm leases.Manager, cs content.Store, is images.Store, tc *TransferConfig) transfer.Transferrer {
-	return &localTransferService{
-		leases:   lm,
-		content:  cs,
-		images:   is,
-		limiterU: semaphore.NewWeighted(int64(tc.MaxConcurrentUploadedLayers)),
-		limiterD: semaphore.NewWeighted(int64(tc.MaxConcurrentDownloads)),
-		config:   *tc,
+	ts := &localTransferService{
+		leases:  lm,
+		content: cs,
+		images:  is,
+		config:  *tc,
 	}
+	if tc.MaxConcurrentUploadedLayers > 0 {
+		ts.limiterU = semaphore.NewWeighted(int64(tc.MaxConcurrentUploadedLayers))
+	}
+	if tc.MaxConcurrentDownloads > 0 {
+		ts.limiterD = semaphore.NewWeighted(int64(tc.MaxConcurrentDownloads))
+	}
+	return ts
 }
 
 func (ts *localTransferService) Transfer(ctx context.Context, src interface{}, dest interface{}, opts ...transfer.Opt) error {