You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package nnet
|
|
|
|
import (
|
|
"git.noahlan.cn/northlan/nnet/component"
|
|
"git.noahlan.cn/northlan/nnet/log"
|
|
"git.noahlan.cn/northlan/nnet/pipeline"
|
|
"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
|
|
}
|
|
}
|
|
|
|
func WithWebsocket(wsOpts ...WSOption) Option {
|
|
return func(options *Options) {
|
|
for _, opt := range wsOpts {
|
|
opt(&options.WS)
|
|
}
|
|
options.WS.IsWebsocket = true
|
|
}
|
|
}
|
|
|
|
func WithWSPath(path string) WSOption {
|
|
return func(opts *WSOptions) {
|
|
opts.WebsocketPath = path
|
|
}
|
|
}
|
|
|
|
func WithWSTLS(certificate, key string) WSOption {
|
|
return func(opts *WSOptions) {
|
|
opts.TLSCertificate = certificate
|
|
opts.TLSKey = key
|
|
}
|
|
}
|