add new var to config

This commit is contained in:
Yanislav Igonin 2021-11-18 10:36:06 +02:00
parent d01924f057
commit f577dbcc7f

View File

@ -2,6 +2,7 @@ package config
import ( import (
"fmt" "fmt"
"log"
"os" "os"
"strconv" "strconv"
) )
@ -11,6 +12,7 @@ type AppConfig struct {
Port int Port int
SeedDb bool SeedDb bool
IsRateLimiterEnabled bool IsRateLimiterEnabled bool
ThreadsMaxCount int
} }
type DbConfig struct { type DbConfig struct {
@ -29,7 +31,7 @@ func getAppConfig() AppConfig {
} }
port, err := strconv.Atoi(portString) port, err := strconv.Atoi(portString)
if err != nil { if err != nil {
panic(fmt.Sprintf("Could not parse %s to int", portString)) log.Panicln(fmt.Sprintf("Could not parse %s to int", portString))
} }
seedDbString := os.Getenv("SEED_DB") seedDbString := os.Getenv("SEED_DB")
@ -38,11 +40,18 @@ func getAppConfig() AppConfig {
isRateLimiterEnabledString := os.Getenv("IS_RATE_LIMITER_ENABLED") isRateLimiterEnabledString := os.Getenv("IS_RATE_LIMITER_ENABLED")
isRateLimiterEnabled := isRateLimiterEnabledString == "true" isRateLimiterEnabled := isRateLimiterEnabledString == "true"
threadsMaxCountString := os.Getenv("THREADS_MAX_COUNT")
threadsMaxCount, err := strconv.Atoi(threadsMaxCountString)
if err != nil {
log.Panicln(fmt.Sprintf("Could not parse %s to int", threadsMaxCountString))
}
return AppConfig{ return AppConfig{
Env: env, Env: env,
Port: port, Port: port,
SeedDb: seedDb, SeedDb: seedDb,
IsRateLimiterEnabled: isRateLimiterEnabled, IsRateLimiterEnabled: isRateLimiterEnabled,
ThreadsMaxCount: threadsMaxCount,
} }
} }