You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/go-micro/plugins/v4/config/encoder/toml"
|
|
"github.com/go-micro/plugins/v4/config/encoder/yaml"
|
|
"github.com/pkg/errors"
|
|
"go-micro.dev/v4/config"
|
|
"go-micro.dev/v4/config/reader"
|
|
"go-micro.dev/v4/config/reader/json"
|
|
"go-micro.dev/v4/config/source/env"
|
|
"go-micro.dev/v4/config/source/file"
|
|
"go-micro.dev/v4/logger"
|
|
"jochum.dev/jo-micro/auth2/internal/util"
|
|
)
|
|
|
|
// Load will load configurations and update it when changed
|
|
func Load(cfg interface{}) error {
|
|
var configor config.Config
|
|
var err error
|
|
switch strings.ToLower(os.Getenv("CONFIG_TYPE")) {
|
|
case "toml":
|
|
filename := "config.toml"
|
|
if name := os.Getenv("CONFIG_FILE"); len(name) > 0 {
|
|
filename = name
|
|
}
|
|
configor, err = config.NewConfig(
|
|
config.WithSource(file.NewSource(file.WithPath(filename))),
|
|
config.WithReader(json.NewReader(reader.WithEncoder(toml.NewEncoder()))),
|
|
)
|
|
case "yaml":
|
|
filename := "config.yaml"
|
|
if name := os.Getenv("CONFIG_FILE"); len(name) > 0 {
|
|
filename = name
|
|
}
|
|
configor, err = config.NewConfig(
|
|
config.WithSource(file.NewSource(file.WithPath(filename))),
|
|
config.WithReader(json.NewReader(reader.WithEncoder(yaml.NewEncoder()))),
|
|
)
|
|
default:
|
|
configor, err = config.NewConfig(
|
|
config.WithSource(env.NewSource()),
|
|
)
|
|
}
|
|
if err != nil {
|
|
return errors.Wrap(err, "configor.New")
|
|
}
|
|
if err := configor.Load(); err != nil {
|
|
return errors.Wrap(err, "configor.Load")
|
|
}
|
|
if err := configor.Scan(cfg); err != nil {
|
|
return errors.Wrap(err, "configor.Scan")
|
|
}
|
|
|
|
w, err := configor.Watch()
|
|
if err != nil {
|
|
return errors.Wrap(err, "configor.Watch")
|
|
}
|
|
util.GoSafe(func() {
|
|
for {
|
|
v, err := w.Next()
|
|
if err != nil {
|
|
logger.Error(err)
|
|
return
|
|
}
|
|
if err := v.Scan(cfg); err != nil {
|
|
logger.Error(err)
|
|
return
|
|
}
|
|
}
|
|
})
|
|
return nil
|
|
}
|