add router handler option
This commit is contained in:
@ -2,11 +2,13 @@ package handler
|
||||
|
||||
import (
|
||||
"github.com/go-gost/gost/pkg/bypass"
|
||||
"github.com/go-gost/gost/pkg/chain"
|
||||
"github.com/go-gost/gost/pkg/logger"
|
||||
"github.com/go-gost/gost/pkg/resolver"
|
||||
)
|
||||
|
||||
type Options struct {
|
||||
Router *chain.Router
|
||||
Bypass bypass.Bypass
|
||||
Resolver resolver.Resolver
|
||||
Logger logger.Logger
|
||||
@ -14,9 +16,9 @@ type Options struct {
|
||||
|
||||
type Option func(opts *Options)
|
||||
|
||||
func LoggerOption(logger logger.Logger) Option {
|
||||
func RouterOption(router *chain.Router) Option {
|
||||
return func(opts *Options) {
|
||||
opts.Logger = logger
|
||||
opts.Router = router
|
||||
}
|
||||
}
|
||||
|
||||
@ -26,8 +28,8 @@ func BypassOption(bypass bypass.Bypass) Option {
|
||||
}
|
||||
}
|
||||
|
||||
func ResolverOption(resolver resolver.Resolver) Option {
|
||||
func LoggerOption(logger logger.Logger) Option {
|
||||
return func(opts *Options) {
|
||||
opts.Resolver = resolver
|
||||
opts.Logger = logger
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user