64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
package viperx
|
||
|
||
import (
|
||
"bytes"
|
||
"fmt"
|
||
"os"
|
||
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
var paths = []string{
|
||
".",
|
||
"./etc",
|
||
"./config",
|
||
}
|
||
|
||
// ConfigFileEncryptorHook 配置文件加密器
|
||
type ConfigFileEncryptorHook interface {
|
||
IsCipherText([]byte) bool
|
||
Encrypt([]byte) []byte
|
||
Decrypt([]byte) []byte
|
||
}
|
||
|
||
var configFileEH ConfigFileEncryptorHook
|
||
|
||
// RegisterConfigEncryptor 注册配置文件加密器
|
||
func RegisterConfigEncryptor(h ConfigFileEncryptorHook) {
|
||
configFileEH = h
|
||
}
|
||
|
||
// GetViper 获取viper配置
|
||
// name: 配置文件名,不带后缀
|
||
// filetype: 配置文件类型，如json、yaml、ini等
|
||
func GetViper(name, filetype string) *viper.Viper {
|
||
buf, err := ReadFile(name, filetype)
|
||
if err != nil {
|
||
panic(fmt.Sprintf("Error while reading config file %s: %v", name+filetype, err))
|
||
}
|
||
|
||
if configFileEH != nil {
|
||
if configFileEH.IsCipherText(buf) {
|
||
buf = configFileEH.Decrypt(buf)
|
||
}
|
||
}
|
||
|
||
vp := viper.New()
|
||
vp.SetConfigName(name)
|
||
vp.SetConfigType(filetype)
|
||
if err = vp.ReadConfig(bytes.NewReader(buf)); err != nil {
|
||
panic(fmt.Sprintf("Error while reading config file %s: %v", name+filetype, err))
|
||
}
|
||
return vp
|
||
}
|
||
|
||
func ReadFile(name, filetype string) ([]byte, error) {
|
||
for _, v := range paths {
|
||
file := fmt.Sprintf("%s/%s.%s", v, name, filetype)
|
||
if _, err := os.Stat(file); err == nil {
|
||
return os.ReadFile(file)
|
||
}
|
||
}
|
||
return nil, nil
|
||
}
|