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

add context for selector

Browse Source
This commit is contained in:
ginuerzh
2022-09-02 17:23:28 +08:00
parent 2835a5d44a
commit 7136710673
4 changed files with 25 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
chain/router.go
View File

@ -129,12 +129,16 @@ func (r *Router) dial(ctx context.Context, network, address string) (conn net.Co
for i := 0; i < count; i++ {
var route Route
if r.chain != nil {
route = r.chain.Route(network, address)
route = r.chain.Route(ctx, network, address)
}
if r.logger.IsLevelEnabled(logger.DebugLevel) {
buf := bytes.Buffer{}
for _, node := range route.Path() {
var path []*Node
if route != nil {
path = route.Path()
}
for _, node := range path {
fmt.Fprintf(&buf, "%s@%s > ", node.Name, node.Addr)
}
fmt.Fprintf(&buf, "%s", address)
@ -174,7 +178,7 @@ func (r *Router) Bind(ctx context.Context, network, address string, opts ...Bind
for i := 0; i < count; i++ {
var route Route
if r.chain != nil {
route = r.chain.Route(network, address)
route = r.chain.Route(ctx, network, address)
if route.Len() == 0 {
err = ErrEmptyRoute
return