112 lines
2.6 KiB
Go
112 lines
2.6 KiB
Go
package config
|
|
|
|
import (
|
|
log "github.com/sirupsen/logrus"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
type Config struct {
|
|
LogLevel string
|
|
NumCPUs int
|
|
CheckPeriod int
|
|
TasksCfg TasksConfig
|
|
GrpcCfg GrpcConfig
|
|
OriginEnabled OriginEnabled
|
|
ExternalBrowser string
|
|
Metrics MetricsConfig
|
|
}
|
|
|
|
type GrpcConfig struct {
|
|
ServerHost string
|
|
ServerPort string
|
|
ApiClientHost string
|
|
ApiClientPort string
|
|
SurugayaScrapperHost string
|
|
SurugayaScrapperPort string
|
|
}
|
|
|
|
type TasksConfig struct {
|
|
RetryCount int
|
|
RetryMinutes int
|
|
TaskTimeout int
|
|
}
|
|
|
|
type OriginEnabled struct {
|
|
Surugaya bool
|
|
Mandarake bool
|
|
Amiami bool
|
|
}
|
|
|
|
type MetricsConfig struct {
|
|
Host string
|
|
Port string
|
|
GinMode string
|
|
}
|
|
|
|
func NewConfig() *Config {
|
|
return &Config{
|
|
LogLevel: getEnv("APP_LOG_LEVEL", "debug"),
|
|
NumCPUs: getEnvInt("APP_NUMCPUS", -1),
|
|
CheckPeriod: getEnvInt("APP_CHECK_PERIOD", 6),
|
|
ExternalBrowser: getEnv("EXTERNAL_BROWSER", ""),
|
|
|
|
GrpcCfg: GrpcConfig{
|
|
ServerHost: getEnv("GRPC_SERVER_HOST", "0.0.0.0"),
|
|
ServerPort: getEnv("GRPC_SERVER_PORT", "9060"),
|
|
ApiClientHost: getEnv("GRPC_API_CLIENT_HOST", "0.0.0.0"),
|
|
ApiClientPort: getEnv("GRPC_API_CLIENT_PORT", "9050"),
|
|
SurugayaScrapperHost: getEnv("GRPC_SURUGAYA_SCRAPPER_HOST", "0.0.0.0"),
|
|
SurugayaScrapperPort: getEnv("GRPC_SURUGAYA_SCRAPPER_PORT", "9070"),
|
|
},
|
|
|
|
TasksCfg: TasksConfig{
|
|
RetryCount: getEnvInt("TASK_RETRY_COUNT", 3),
|
|
RetryMinutes: getEnvInt("TASK_RETRY_MINUTES", 5),
|
|
TaskTimeout: getEnvInt("TASK_TIMEOUT_MINUTES", 5),
|
|
},
|
|
|
|
OriginEnabled: OriginEnabled{
|
|
Surugaya: getEnvBool("ORIGIN_SURUGAYA_ENABLED", false),
|
|
Mandarake: getEnvBool("ORIGIN_MANDARAKE_ENABLED", false),
|
|
Amiami: getEnvBool("ORIGIN_AMIAMI_ENABLED", false),
|
|
},
|
|
|
|
Metrics: MetricsConfig{
|
|
Host: getEnv("METRICS_HOST", "0.0.0.0"),
|
|
Port: getEnv("METRICS_PORT", "49060"),
|
|
GinMode: getEnv("METRICS_GIN_MODE", "dev"),
|
|
},
|
|
}
|
|
}
|
|
|
|
func getEnv(key, fallback string) string {
|
|
if value, ok := os.LookupEnv(key); ok {
|
|
return value
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func getEnvInt(key string, fallback int) int {
|
|
if value, ok := os.LookupEnv(key); ok {
|
|
num, err := strconv.Atoi(value)
|
|
if err != nil {
|
|
log.WithField("default", -1).Warn("Config | Can't parse value as int")
|
|
return fallback
|
|
}
|
|
return num
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func getEnvBool(key string, fallback bool) bool {
|
|
if value, ok := os.LookupEnv(key); ok {
|
|
val, err := strconv.ParseBool(value)
|
|
if err != nil {
|
|
log.WithField("default", false).Warn("Config | Can't parse value as bool")
|
|
return fallback
|
|
}
|
|
return val
|
|
}
|
|
return fallback
|
|
}
|