-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.go
90 lines (73 loc) · 3.01 KB
/
config.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package config
import (
"fmt"
"io/ioutil"
"os"
"time"
"github.com/DataWorkbench/common/gormwrap"
"github.com/DataWorkbench/common/grpcwrap"
"github.com/DataWorkbench/common/gtrace"
"github.com/DataWorkbench/common/metrics"
"github.com/DataWorkbench/loader"
"github.com/go-playground/validator/v10"
"gopkg.in/yaml.v3"
)
// The config file path used by Load config
var FilePath string
const (
envPrefix = "JOB_MANAGER"
)
// Config is the configuration settings for spacemanager
type Config struct {
LogLevel int8 `json:"log_level" yaml:"log_level" env:"LOG_LEVEL" validate:"gte=1,lte=5"`
ZeppelinAddress string `json:"zeppelin_address" yaml:"zeppelin_address" env:"ZEPPELIN_ADDRESS" validate:"required"`
GRPCServer *grpcwrap.ServerConfig `json:"grpc_server" yaml:"grpc_server" env:"GRPC_SERVER" validate:"required"`
MetricsServer *metrics.Config `json:"metrics_server" yaml:"metrics_server" env:"METRICS_SERVER" validate:"required"`
MySQL *gormwrap.MySQLConfig `json:"mysql" yaml:"mysql" env:"MYSQL" validate:"required"`
//ETCD *getcd.Config `json:"etcd" yaml:"etcd" env:"ETCD" validate:"required"`
Tracer *gtrace.Config `json:"tracer" yaml:"tracer" env:"TRACER" validate:"required"`
SpaceManager *grpcwrap.ClientConfig `json:"space_manager" yaml:"space_manager" env:"SPACE_MANAGER" validate:"required"`
//ResourceManagerServer *grpcwrap.ClientConfig `json:"resourcemanager_server" yaml:"resourcemanager_server" env:"RESOURCEMANAGER_SERVER" validate:"required"`
//EngineManagerServer *grpcwrap.ClientConfig `json:"enginemanager_server" yaml:"enginemanager_server" env:"ENGINEMANAGER_SERVER" validate:"required"`
//UdfManagerServer *grpcwrap.ClientConfig `json:"udfmanager_server" yaml:"udfmanager_server" env:"UDFMANAGER_SERVER" validate:"required"`
}
func loadFromFile(cfg *Config) (err error) {
if FilePath == "" {
return
}
fmt.Printf("%s load config from file <%s>\n", time.Now().Format(time.RFC3339Nano), FilePath)
var b []byte
b, err = ioutil.ReadFile(FilePath)
if err != nil && os.IsNotExist(err) {
return
}
err = yaml.Unmarshal(b, cfg)
if err != nil {
fmt.Println("parse config file error:", err)
}
return
}
// LoadConfig load all configuration from specified file
// Must be set `FilePath` before called
func Load() (cfg *Config, err error) {
cfg = &Config{}
_ = loadFromFile(cfg)
l := loader.New(
loader.WithPrefix(envPrefix),
loader.WithTagName("env"),
loader.WithOverride(true),
)
if err = l.Load(cfg); err != nil {
return
}
// output the config content
fmt.Printf("%s pid=%d the latest configuration: \n", time.Now().Format(time.RFC3339Nano), os.Getpid())
fmt.Println("")
b, _ := yaml.Marshal(cfg)
fmt.Println(string(b))
validate := validator.New()
if err = validate.Struct(cfg); err != nil {
return
}
return
}