-
Notifications
You must be signed in to change notification settings - Fork 3
/
load.go
82 lines (69 loc) · 2.6 KB
/
load.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
package confKit
import (
"github.com/richelieu-yang/chimera/v2/src/core/fileKit"
"github.com/richelieu-yang/chimera/v2/src/log/logrusKit"
"github.com/sirupsen/logrus"
"github.com/zeromicro/go-zero/core/conf"
)
// MustLoad 读取配置文件并反序列化为 指定结构体指针ptr
/*
PS:
(1) default对应的是 该行被注释掉(而非值为空) 的情况;
(2) default和 '|' 一起使用;
(3) range和 ':' 一起使用.
!!!:
(0) 所有字段首字母大写 && 有json tag;
(1) 结构体可以参考 go-zero的 rest.RestConf,可以通过 tag 控制配置的值(默认值default、范围range、可选optional...;可以组合)
(2) 值为数组时要注意(要么有值(size > 0),要么全注释掉),以免返回error.
(3) tag range(定义当前字段数值范围)支持:
[:5] (:5] [:5) (:5)
[1:] [1:) (1:] (1:)
[1:5] [1:5) (1:5] (1:5)
e.g.
type config struct {
Number1 int `json:"number1,range=[1:]"`
}
可能返回的error:
(a) wrong number range setting: 不满足tag range
@param path 配置文件的路径(推荐使用.yaml)
@param ptr [不能为nil] 结构体实例的指针
@param opts e.g. conf.UseEnv()
e.g. 结构体属性的tag(https://www.w3cschool.cn/gozero/gozero-eo623nm5.html)
json:"name,optional"
json:"gender,options=male|female"
json:"gender,default=male"
json:"age,range=[0:120]"
e.g.1 组合多个tag
Port int `json:"port,optional,range=[-1:65535]"`
Port int `json:"port,default=-1,range=[-1:65535]"`
*/
func MustLoad(path string, ptr any, options ...conf.Option) {
if err := Load(path, ptr, options...); err != nil {
logrusKit.DisableQuote(nil)
logrus.Fatalf("%+v", err)
}
}
func Load(path string, ptr any, options ...conf.Option) error {
if err := fileKit.AssertExistAndIsFile(path); err != nil {
return err
}
return conf.Load(path, ptr, options...)
}
// MustLoadFromJsonBytes 加载 .json 格式的配置文件内容.
func MustLoadFromJsonBytes(jsonContent []byte, v any) {
if err := LoadFromJsonBytes(jsonContent, v); err != nil {
logrusKit.DisableQuote(nil)
logrus.Fatalf("%+v", err)
}
}
// LoadFromJsonBytes 加载 .json 格式的配置文件内容.
var LoadFromJsonBytes func(jsonContent []byte, v any) error = conf.LoadFromJsonBytes
// MustLoadFromYamlBytes 加载 .yaml 格式的配置文件内容.
func MustLoadFromYamlBytes(yamlContent []byte, v any) {
if err := LoadFromYamlBytes(yamlContent, v); err != nil {
logrusKit.DisableQuote(nil)
logrus.Fatalf("%+v", err)
}
}
// LoadFromYamlBytes 加载 .yaml 格式的配置文件内容.
var LoadFromYamlBytes func(yamlContent []byte, v any) error = conf.LoadFromYamlBytes