mirror of
https://git.rwth-aachen.de/acs/public/villas/web-backend-go/
synced 2025-03-30 00:00:12 +01:00
6d4c84ec4e
- Check more errors in InitDB - fix for healthz endpoint (error checking was wrong) - remove dbinit parameter from config due to redundancy with mode parameter
87 lines
2.6 KiB
Go
87 lines
2.6 KiB
Go
package configuration
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/zpatrick/go-config"
|
|
)
|
|
|
|
// Global configuration
|
|
var GolbalConfig *config.Config = nil
|
|
|
|
func InitConfig() error {
|
|
if GolbalConfig != nil {
|
|
return nil
|
|
}
|
|
|
|
var (
|
|
dbHost = flag.String("dbhost", "/var/run/postgresql", "Host of the PostgreSQL database (default is /var/run/postgresql for localhost DB on Ubuntu systems)")
|
|
dbName = flag.String("dbname", "villasdb", "Name of the database to use (default is villasdb)")
|
|
dbUser = flag.String("dbuser", "", "Username of database connection (default is <empty>)")
|
|
dbPass = flag.String("dbpass", "", "Password of database connection (default is <empty>)")
|
|
dbSSLMode = flag.String("dbsslmode", "disable", "SSL mode of DB (default is disable)") // TODO: change default for production
|
|
amqpURL = flag.String("amqp", "", "If set, use this url to connect to an AMQP broker (default is disabled)")
|
|
configFile = flag.String("configFile", "", "Path to YAML configuration file")
|
|
mode = flag.String("mode", "release", "Select debug/release/test mode (default is release)")
|
|
baseHost = flag.String("base-host", "localhost:4000", "The host:port at which the backend is hosted (default: localhost:4000)")
|
|
basePath = flag.String("base-path", "/api/v2", "The path at which the API routes are located (default /api/v2)")
|
|
)
|
|
flag.Parse()
|
|
|
|
static := map[string]string{
|
|
"db.host": *dbHost,
|
|
"db.name": *dbName,
|
|
"db.user": *dbUser,
|
|
"db.pass": *dbPass,
|
|
"db.ssl": *dbSSLMode,
|
|
"amqp.url": *amqpURL,
|
|
"mode": *mode,
|
|
"base.host": *baseHost,
|
|
"base.path": *basePath,
|
|
}
|
|
|
|
mappings := map[string]string{
|
|
"DB_HOST": "db.host",
|
|
"DB_NAME": "db.name",
|
|
"DB_USER": "db.user",
|
|
"DB_PASS": "db.pass",
|
|
"DB_SSLMODE": "db.ssl",
|
|
"AMQP_URL": "amqp.url",
|
|
"BASE_HOST": "base.host",
|
|
"BASE_PATH": "base.path",
|
|
"MODE": "mode",
|
|
}
|
|
|
|
defaults := config.NewStatic(static)
|
|
env := config.NewEnvironment(mappings)
|
|
|
|
if _, err := os.Stat(*configFile); os.IsExist(err) {
|
|
yamlFile := config.NewYAMLFile(*configFile)
|
|
GolbalConfig = config.NewConfig([]config.Provider{defaults, yamlFile, env})
|
|
} else {
|
|
GolbalConfig = config.NewConfig([]config.Provider{defaults, env})
|
|
}
|
|
|
|
err := GolbalConfig.Load()
|
|
if err != nil {
|
|
log.Fatal("failed to parse config")
|
|
return err
|
|
}
|
|
|
|
m, err := GolbalConfig.String("mode")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if m != "test" {
|
|
settings, _ := GolbalConfig.Settings()
|
|
log.Print("All settings:")
|
|
for key, val := range settings {
|
|
log.Printf(" %s = %s \n", key, val)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|