|
|
package nnet
|
|
|
|
|
|
import (
|
|
|
"git.noahlan.cn/northlan/nnet/component"
|
|
|
"git.noahlan.cn/northlan/nnet/env"
|
|
|
"git.noahlan.cn/northlan/nnet/log"
|
|
|
"git.noahlan.cn/northlan/nnet/pipeline"
|
|
|
"net/http"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
type Option func(options *Options)
|
|
|
type WSOption func(opts *WSOptions)
|
|
|
|
|
|
func WithLogger(logger log.Logger) Option {
|
|
|
return func(_ *Options) {
|
|
|
log.SetLogger(logger)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func WithPipeline(pipeline pipeline.Pipeline) Option {
|
|
|
return func(options *Options) {
|
|
|
options.Pipeline = pipeline
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func WithComponents(components *component.Components) Option {
|
|
|
return func(options *Options) {
|
|
|
options.Components = components
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func WithHeartbeatInterval(d time.Duration) Option {
|
|
|
return func(options *Options) {
|
|
|
options.HeartbeatInterval = d
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// WithTimerPrecision 设置Timer精度
|
|
|
// 注：精度需大于1ms, 并且不能在运行时更改
|
|
|
// 默认精度是 time.Second
|
|
|
func WithTimerPrecision(precision time.Duration) Option {
|
|
|
if precision < time.Millisecond {
|
|
|
panic("time precision can not less than a Millisecond")
|
|
|
}
|
|
|
return func(_ *Options) {
|
|
|
env.TimerPrecision = precision
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// WithWebsocket 开启Websocket, 参数是websocket的相关参数 nnet.WSOption
|
|
|
func WithWebsocket(wsOpts ...WSOption) Option {
|
|
|
return func(options *Options) {
|
|
|
for _, opt := range wsOpts {
|
|
|
opt(&options.WS)
|
|
|
}
|
|
|
options.WS.IsWebsocket = true
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// WithWSPath 设置websocket的path
|
|
|
func WithWSPath(path string) WSOption {
|
|
|
return func(opts *WSOptions) {
|
|
|
opts.WebsocketPath = path
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// WithWSTLSConfig 设置websocket的证书和密钥
|
|
|
func WithWSTLSConfig(certificate, key string) WSOption {
|
|
|
return func(opts *WSOptions) {
|
|
|
opts.TLSCertificate = certificate
|
|
|
opts.TLSKey = key
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func WithWSCheckOriginFunc(fn func(*http.Request) bool) WSOption {
|
|
|
return func(opts *WSOptions) {
|
|
|
if fn != nil {
|
|
|
opts.CheckOrigin = fn
|
|
|
}
|
|
|
}
|
|
|
}
|