/
options.go
56 lines (49 loc) · 1.29 KB
/
options.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
/*
* @Date: 2022-04-24 09:56:04
* @LastEditTime: 2022-04-29 18:32:17
* @FilePath: /github.com/hyetpang/go-frame/pkgs/options/options.go
*/
package options
import (
"go.uber.org/fx"
)
type (
Options struct {
ConfigFile string // 配置文件路径
FxOptions []fx.Option // 需要注入的对象
IsStart bool // true=>运行一次就退出
UseHttp bool // 是否使用http
UseGraceRestart bool // 是否使用优雅重启
}
Option func(*Options) // 选项函数类型
)
// 注册需要提供的对象
func WithProviders(providers ...any) Option {
return func(o *Options) {
for _, provider := range providers {
o.FxOptions = append(o.FxOptions, fx.Provide(provider))
}
}
}
// 注册需要被调用的函数
func WithInvokes(Invokes ...any) Option {
return func(o *Options) {
for _, Invoke := range Invokes {
o.FxOptions = append(o.FxOptions, fx.Invoke(Invoke))
}
}
}
// 一次注册单个fx.Option
func WithFxOption(fxOptions ...fx.Option) Option {
return func(o *Options) {
o.FxOptions = append(o.FxOptions, fxOptions...)
}
}
// 一次注册多个fx.Option
func WithFxOptions(fxOptions ...[]fx.Option) Option {
return func(o *Options) {
for _, op := range fxOptions {
o.FxOptions = append(o.FxOptions, op...)
}
}
}