From 42742b6330ffa6dd9507a971720715bfa4f45269 Mon Sep 17 00:00:00 2001
From: "Grot (@grafanabot)" <43478413+grafanabot@users.noreply.github.com>
Date: Tue, 30 Jul 2024 04:26:18 +0200
Subject: [PATCH] chore: [k213] chore: add congestion control to AWS (#13705)

Co-authored-by: Ed Welch <edward.welch@grafana.com>
---
 pkg/storage/factory.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pkg/storage/factory.go b/pkg/storage/factory.go
index 5932b08afe8f..01066d69ad5f 100644
--- a/pkg/storage/factory.go
+++ b/pkg/storage/factory.go
@@ -488,6 +488,9 @@ func NewChunkClient(name string, cfg Config, schemaCfg config.SchemaConfig, cc c
 			if err != nil {
 				return nil, err
 			}
+			if cfg.CongestionControl.Enabled {
+				c = cc.Wrap(c)
+			}
 			return client.NewClientWithMaxParallel(c, nil, cfg.MaxParallelGetChunk, schemaCfg), nil
 
 		case types.StorageTypeGCS: