31 lines
538 B
Go
31 lines
538 B
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
)
|
|
|
|
type Config struct {
|
|
Host string
|
|
Port string
|
|
GRPCPort string
|
|
AdminToken string
|
|
DBPath string
|
|
}
|
|
|
|
func Load() *Config {
|
|
return &Config{
|
|
Host: getenv("HOST", "0.0.0.0"),
|
|
Port: getenv("PORT", "9393"),
|
|
GRPCPort: getenv("GRPC_PORT", "9394"),
|
|
AdminToken: getenv("ADMIN_TOKEN", "shumengya520"),
|
|
DBPath: getenv("DB_PATH", "./data/servers.db"),
|
|
}
|
|
}
|
|
|
|
func getenv(key, fallback string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|