34 lines
664 B
Go
34 lines
664 B
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
type Config struct {
|
|
ListenAddr string
|
|
DatabaseURL string
|
|
AuthDBPath string
|
|
TushareAPIURL string
|
|
}
|
|
|
|
func Load() (*Config, error) {
|
|
cfg := &Config{
|
|
ListenAddr: getenv("LISTEN_ADDR", ":8080"),
|
|
DatabaseURL: os.Getenv("DATABASE_URL"),
|
|
AuthDBPath: getenv("AUTH_DB_PATH", "/app/auth/auth.db"),
|
|
TushareAPIURL: getenv("TUSHARE_API_URL", "http://tushare:8000"),
|
|
}
|
|
if cfg.DatabaseURL == "" {
|
|
return nil, fmt.Errorf("DATABASE_URL 环境变量未设置")
|
|
}
|
|
return cfg, nil
|
|
}
|
|
|
|
func getenv(key, fallback string) string {
|
|
if v, ok := os.LookupEnv(key); ok && v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|