85 lines
2.4 KiB
Go
85 lines
2.4 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
type Config struct {
|
|
GRPCPort int
|
|
GRPCEnableReflection bool
|
|
HTTPPort int
|
|
LogLevel string
|
|
DBHost string
|
|
DBPort int
|
|
DBUser string
|
|
DBPassword string
|
|
DBName string
|
|
RedisAddr string
|
|
UserServiceAddr string
|
|
SubmissionServiceAddr string
|
|
TaskServiceAddr string
|
|
CacheEnabled bool
|
|
}
|
|
|
|
func Load() (*Config, error) {
|
|
_ = godotenv.Load()
|
|
|
|
return &Config{
|
|
GRPCPort: mustGetInt("RESULTS_GRPC_PORT", 50054),
|
|
GRPCEnableReflection: mustGetBool("RESULTS_GRPC_ENABLE_REFLECTION", false),
|
|
HTTPPort: mustGetInt("RESULTS_HTTP_PORT", 8083),
|
|
LogLevel: getEnv("LOG_LEVEL", "info"),
|
|
DBHost: getEnv("POSTGRES_HOST", "localhost"),
|
|
DBPort: mustGetInt("POSTGRES_PORT", 5432),
|
|
DBUser: getEnv("POSTGRES_USERNAME", "postgres"),
|
|
DBPassword: getEnv("POSTGRES_PASSWORD", "postgres"),
|
|
DBName: getEnv("POSTGRES_DATABASE", "postgres"),
|
|
RedisAddr: getEnv("REDIS_ADDR", "localhost:6379"),
|
|
UserServiceAddr: getEnv("USER_GRPC_ADDR", "localhost:50051"),
|
|
SubmissionServiceAddr: getEnv("SUBMISSION_GRPC_ADDR", "localhost:50055"),
|
|
TaskServiceAddr: getEnv("TASK_GRPC_ADDR", "localhost:50056"),
|
|
CacheEnabled: mustGetBool("CACHE_ENABLED", false),
|
|
}, nil
|
|
}
|
|
|
|
func getEnv(key, def string) string {
|
|
if val := os.Getenv(key); val != "" {
|
|
return val
|
|
}
|
|
return def
|
|
}
|
|
|
|
func mustGetInt(key string, def int) int {
|
|
val := getEnv(key, strconv.Itoa(def))
|
|
n, err := strconv.Atoi(val)
|
|
if err != nil {
|
|
log.Fatalf("invalid int for %s: %v", key, err)
|
|
}
|
|
return n
|
|
}
|
|
|
|
func mustGetBool(key string, def bool) bool {
|
|
val := getEnv(key, strconv.FormatBool(def))
|
|
b, err := strconv.ParseBool(val)
|
|
if err != nil {
|
|
log.Fatalf("invalid bool for %s: %v", key, err)
|
|
}
|
|
return b
|
|
}
|
|
|
|
func (c Config) BuildPostgresConnStr() string {
|
|
return fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable",
|
|
c.DBHost, c.DBPort, c.DBUser, c.DBPassword, c.DBName)
|
|
}
|
|
|
|
func (c Config) BuildPostgresDSN() string {
|
|
return fmt.Sprintf("postgresql://%s:%s@%s/%s?sslmode=disable",
|
|
c.DBUser, c.DBPassword, net.JoinHostPort(c.DBHost, strconv.Itoa(c.DBPort)), c.DBName)
|
|
}
|