CyberCoveLLC/threshold
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(ratelimit): Create rate limits from configs

Browse Source
This commit is contained in:
Timothy Messier committed 2023-12-04 16:20:44 +00:00
1 parent 61fcd857ec
commit 6da47443fd
6 files changed
+684 -11
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

internal/cmd/config/config.go
+1 -1
View File
@@ -693,7 +693,7 @@ func Parse(d string) (*Config, error) {
}
if result.Controller.ApiRateLimiterMaxEntries <= 0 {
result.Controller.ApiRateLimiterMaxEntries = ratelimit.DefaultLimiterMaxEntries
result.Controller.ApiRateLimiterMaxEntries = ratelimit.DefaultLimiterMaxEntries()
}
}
internal/cmd/config/config_load_test.go
+5 -5
View File
@@ -419,7 +419,7 @@ func TestLoad(t *testing.T) {
WorkerStatusGracePeriodDuration: 0,
LivenessTimeToStaleDuration: 0,
ApiRateLimits: make(ratelimit.Configs, 0),
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries,
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries(),
},
DevController: false,
DevUiPassthroughDir: "",
@@ -844,7 +844,7 @@ func TestLoad(t *testing.T) {
WorkerStatusGracePeriodDuration: 0,
LivenessTimeToStaleDuration: 0,
ApiRateLimits: make(ratelimit.Configs, 0),
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries,
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries(),
},
DevController: false,
DevUiPassthroughDir: "",
@@ -1283,7 +1283,7 @@ func TestLoad(t *testing.T) {
Unlimited: false,
},
},
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries,
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries(),
},
DevController: false,
DevUiPassthroughDir: "",
@@ -1703,7 +1703,7 @@ func TestLoad(t *testing.T) {
WorkerStatusGracePeriodDuration: 0,
LivenessTimeToStaleDuration: 0,
ApiRateLimits: make(ratelimit.Configs, 0),
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries,
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries(),
},
DevController: false,
DevUiPassthroughDir: "",
@@ -1794,7 +1794,7 @@ func TestLoad(t *testing.T) {
WorkerStatusGracePeriodDuration: 0,
LivenessTimeToStaleDuration: 0,
ApiRateLimits: make(ratelimit.Configs, 0),
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries,
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries(),
},
DevController: false,
DevUiPassthroughDir: "",
internal/cmd/config/config_test.go
+2 -2
View File
@@ -113,7 +113,7 @@ func TestDevController(t *testing.T) {
Name: "dev-controller",
Description: "A default controller created in dev mode",
ApiRateLimits: make(ratelimit.Configs, 0),
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries,
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries(),
},
DevController: true,
}
@@ -488,7 +488,7 @@ func TestDevCombined(t *testing.T) {
Name: "dev-controller",
Description: "A default controller created in dev mode",
ApiRateLimits: make(ratelimit.Configs, 0),
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries,
ApiRateLimiterMaxEntries: ratelimit.DefaultLimiterMaxEntries(),
},
DevController: true,
Worker: &Worker{