feat: hide rate limiter under flag

This commit is contained in:
Yanislav Igonin 2021-09-08 18:54:46 +03:00
parent fc650b2d2a
commit 828deb52a4
3 changed files with 15 additions and 7 deletions

View File

@ -1,4 +1,5 @@
ENV=debug
PORT=3000
SEED_DB=true
IS_RATE_LIMITER_ENABLED=true
POSTGRES_URL=postgres://localhost/micrach

View File

@ -10,6 +10,7 @@ type AppConfig struct {
Env string
Port int
SeedDb bool
IsRateLimiterEnabled bool
}
type DbConfig struct {
@ -34,10 +35,14 @@ func getAppConfig() AppConfig {
seedDbString := os.Getenv("SEED_DB")
seedDb := seedDbString == "true"
isRateLimiterEnabledString := os.Getenv("IS_RATE_LIMITER_ENABLED")
isRateLimiterEnabled := isRateLimiterEnabledString == "true"
return AppConfig{
Env: env,
Port: port,
SeedDb: seedDb,
IsRateLimiterEnabled: isRateLimiterEnabled,
}
}

View File

@ -43,7 +43,9 @@ func main() {
router := gin.Default()
router.LoadHTMLGlob("templates/*.html")
router.ForwardedByClientIP = true
if Config.App.IsRateLimiterEnabled {
router.Use(middleware)
}
router.Static("/uploads", "./uploads")
router.Static("/static", "./static")
router.GET("/", Controllers.GetThreads)