add router handler option

This commit is contained in:
ginuerzh
2021-12-31 00:03:24 +08:00
parent a430384bba
commit 9769efe33c
35 changed files with 98 additions and 281 deletions

View File

@ -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
}
}