You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
1.8 KiB
101 lines
1.8 KiB
package gin
|
|
|
|
import (
|
|
"yunyan/lego/sys/log"
|
|
"yunyan/lego/utils/mapstructure"
|
|
)
|
|
|
|
type Option func(*Options)
|
|
type Options struct {
|
|
ListenPort int //监听端口
|
|
CertFile string //tls文件
|
|
KeyFile string //tls文件
|
|
LetEncrypt bool //Let's Encrypt 证书模式
|
|
Domain []string //域名
|
|
MultipartMemory int64 //文件上传最大尺寸
|
|
Debug bool //日志是否开启
|
|
Log log.ILogger
|
|
}
|
|
|
|
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 SetLetEncrypt(v bool) Option {
|
|
return func(o *Options) {
|
|
o.LetEncrypt = v
|
|
}
|
|
}
|
|
|
|
func SetDomain(v []string) Option {
|
|
return func(o *Options) {
|
|
o.Domain = v
|
|
}
|
|
}
|
|
|
|
func SetMultipartMemory(v int64) Option {
|
|
return func(o *Options) {
|
|
o.MultipartMemory = v
|
|
}
|
|
}
|
|
|
|
func SetDebug(v bool) Option {
|
|
return func(o *Options) {
|
|
o.Debug = v
|
|
}
|
|
}
|
|
|
|
func SetLog(v log.ILogger) Option {
|
|
return func(o *Options) {
|
|
o.Log = v
|
|
}
|
|
}
|
|
|
|
func newOptions(config map[string]interface{}, opts ...Option) (options *Options, err error) {
|
|
options = &Options{
|
|
ListenPort: 8080,
|
|
CertFile: "",
|
|
KeyFile: "",
|
|
Debug: true,
|
|
}
|
|
if config != nil {
|
|
mapstructure.Decode(config, &options)
|
|
}
|
|
for _, o := range opts {
|
|
o(options)
|
|
}
|
|
if options.Log == nil {
|
|
options.Log = log.NewTurnlog(options.Debug, log.Clone("sys.gin", 3))
|
|
}
|
|
return
|
|
}
|
|
|
|
func newOptionsByOption(opts ...Option) (options *Options, err error) {
|
|
options = &Options{
|
|
ListenPort: 8080,
|
|
CertFile: "",
|
|
KeyFile: "",
|
|
Debug: false,
|
|
}
|
|
for _, o := range opts {
|
|
o(options)
|
|
}
|
|
if options.Log == nil {
|
|
options.Log = log.NewTurnlog(options.Debug, log.Clone("sys.gin", 3))
|
|
}
|
|
return
|
|
}
|
|
|