go-gost/gost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add router handler option

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

12
pkg/listener/rtcp/listener.go
View File

@ -22,6 +22,7 @@ type rtcpListener struct {
chain *chain.Chain
ln net.Listener
md metadata
router *chain.Router
logger logger.Logger
closed chan struct{}
}
@ -34,13 +35,16 @@ func NewListener(opts ...listener.Option) listener.Listener {
return &rtcpListener{
addr: options.Addr,
closed: make(chan struct{}),
router: &chain.Router{
Logger: options.Logger,
},
logger: options.Logger,
}
}
// implements chain.Chainable interface
func (l *rtcpListener) WithChain(chain *chain.Chain) {
l.chain = chain
l.router.Chain = chain
}
func (l *rtcpListener) Init(md md.Metadata) (err error) {
@ -66,11 +70,7 @@ func (l *rtcpListener) Accept() (conn net.Conn, err error) {
}
if l.ln == nil {
r := (&chain.Router{}).
WithChain(l.chain).
WithRetry(l.md.retryCount).
WithLogger(l.logger)
l.ln, err = r.Bind(context.Background(), "tcp", l.laddr.String(),
l.ln, err = l.router.Bind(context.Background(), "tcp", l.laddr.String(),
connector.MuxBindOption(true),
)
if err != nil {