77 lines
1.3 KiB
Go
77 lines
1.3 KiB
Go
package gin
|
|
|
|
import (
|
|
"go_dreamfactory/lego/sys/log"
|
|
"go_dreamfactory/lego/utils/mapstructure"
|
|
)
|
|
|
|
type Option func(*Options)
|
|
type Options struct {
|
|
ListenPort int //监听端口
|
|
CertFile string //tls文件
|
|
KeyFile string //tls文件
|
|
Debug bool //日志是否开启
|
|
Log log.ILog
|
|
}
|
|
|
|
func SetListenPort(v int) Option {
|
|
return func(o *Options) {
|
|
o.ListenPort = v
|
|
}
|
|
}
|
|
|
|
func SetCertFile(v string) Option {
|
|
return func(o *Options) {
|
|
o.CertFile = v
|
|
}
|
|
}
|
|
|
|
func SetKeyFile(v string) Option {
|
|
return func(o *Options) {
|
|
o.KeyFile = v
|
|
}
|
|
}
|
|
|
|
func SetDebug(v bool) Option {
|
|
return func(o *Options) {
|
|
o.Debug = v
|
|
}
|
|
}
|
|
|
|
func SetLog(v log.ILog) Option {
|
|
return func(o *Options) {
|
|
o.Log = v
|
|
}
|
|
}
|
|
|
|
func newOptions(config map[string]interface{}, opts ...Option) Options {
|
|
options := Options{
|
|
ListenPort: 8080,
|
|
CertFile: "",
|
|
KeyFile: "",
|
|
Debug: true,
|
|
Log: log.Clone(log.SetLoglayer(2)),
|
|
}
|
|
if config != nil {
|
|
mapstructure.Decode(config, &options)
|
|
}
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
return options
|
|
}
|
|
|
|
func newOptionsByOption(opts ...Option) Options {
|
|
options := Options{
|
|
ListenPort: 8080,
|
|
CertFile: "",
|
|
KeyFile: "",
|
|
Debug: true,
|
|
Log: log.Clone(log.SetLoglayer(2)),
|
|
}
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
return options
|
|
}
|