Fix: Allow multiple quorums in allowlist (Layr-Labs#314)

bxue-l2 · Mar 5, 2024 · 9827a87 · 9827a87
1 parent 780e1be
commit 9827a87
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/disperser/apiserver/rate_config.go b/disperser/apiserver/rate_config.go
@@ -154,11 +154,19 @@ func ReadCLIConfig(c *cli.Context) (RateConfig, error) {
 			log.Printf("invalid allowlist entry: failed to convert throughput from string: %s", allowlistEntry)
 			continue
 		}
-		allowlist[ip] = map[core.QuorumID]PerUserRateInfo{
-			core.QuorumID(quorumID): {
+		rateInfoByQuorum, ok := allowlist[ip]
+		if !ok {
+			allowlist[ip] = map[core.QuorumID]PerUserRateInfo{
+				core.QuorumID(quorumID): {
+					Throughput: common.RateParam(byteRate),
+					BlobRate:   common.RateParam(blobRate * blobRateMultiplier),
+				},
+			}
+		} else {
+			rateInfoByQuorum[core.QuorumID(quorumID)] = PerUserRateInfo{
 				Throughput: common.RateParam(byteRate),
 				BlobRate:   common.RateParam(blobRate * blobRateMultiplier),
-			},
+			}
 		}
 	}