modify config
This commit is contained in:
@@ -5,14 +5,22 @@ import (
|
||||
"github.com/spf13/viper"
|
||||
)
|
||||
|
||||
func Load(file string) (*viper.Viper, error) {
|
||||
func Load(file, app string) (*viper.Viper, error) {
|
||||
v := viper.NewWithOptions(viper.KeyDelimiter("_"))
|
||||
v.AutomaticEnv()
|
||||
v.SetConfigFile(file)
|
||||
|
||||
if file == "" {
|
||||
v.SetConfigType("toml")
|
||||
v.SetConfigName(app)
|
||||
v.AddConfigPath("/etc")
|
||||
v.AddConfigPath("/usr/local/etc")
|
||||
} else {
|
||||
v.SetConfigFile(file)
|
||||
}
|
||||
|
||||
if err := v.ReadInConfig(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err := container.Container.Provide(func() (*viper.Viper, error) {
|
||||
return v, nil
|
||||
})
|
||||
|
||||
@@ -4,10 +4,21 @@ import (
|
||||
"fmt"
|
||||
|
||||
"github.com/go-redis/redis/v8"
|
||||
"github.com/rogeecn/atom/container"
|
||||
"github.com/rogeecn/atom/utils/opt"
|
||||
)
|
||||
|
||||
const DefaultPrefix = "Redis"
|
||||
|
||||
func DefaultProvider() container.ProviderContainer {
|
||||
return container.ProviderContainer{
|
||||
Provider: Provide,
|
||||
Options: []opt.Option{
|
||||
opt.Prefix(DefaultPrefix),
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
Host string
|
||||
Port uint
|
||||
|
||||
Reference in New Issue
Block a user