hongbo-miao/hongbomiao.com

View on GitHub
api-go/internal/grpc_server/utils/get_config.go

Summary

Maintainability
A
0 mins
Test Coverage
package utils

import (
    "github.com/joho/godotenv"
    "os"
)

type Config struct {
    AppEnv              string
    Port                string
    OpenCensusAgentHost string
    OpenCensusAgentPort string
}

func GetConfig() *Config {
    path := "config/grpc_server/"

    appEnv := os.Getenv("APP_ENV")
    if appEnv == "" {
        appEnv = "development"
    }

    _ = godotenv.Load(path + ".env." + appEnv + ".local")
    _ = godotenv.Load(path + ".env." + appEnv)

    return &Config{
        AppEnv:              appEnv,
        Port:                os.Getenv("PORT"),
        OpenCensusAgentHost: os.Getenv("OPEN_CENSUS_AGENT_HOST"),
        OpenCensusAgentPort: os.Getenv("OPEN_CENSUS_AGENT_PORT"),
    }
}