unistack-org/micro
4
0
Fork 3
Code Issues 20 Pull Requests 1 Actions Packages Projects Releases 282 Wiki Activity

Merge pull request #732 from micro/massive-cruft

Browse Source 
Fix massive cruft in tunnel dial to set the link on discovered
This commit is contained in:
Asim Aslam 2019-09-04 20:27:07 +01:00 committed by GitHub
commit e33bd17894
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tunnel/default.go
View File

@ -2,6 +2,7 @@ package tunnel
import (
"errors"
"math/rand"
"strings"
"sync"
"time"
@ -529,7 +530,7 @@ func (t *tun) listen(link *link) {
// bail if no session or listener has been found
if !exists {
log.Debugf("Tunnel skipping no session exists")
log.Debugf("Tunnel skipping no session %s %s exists", channel, sessionId)
// drop it, we don't care about
// messages we don't know about
continue
@ -837,6 +838,8 @@ func (t *tun) Dial(channel string, opts ...DialOption) (Session, error) {
return wait
}
var links []string
// non multicast so we need to find the link
t.RLock()
for _, link := range t.links {
@ -847,11 +850,20 @@ func (t *tun) Dial(channel string, opts ...DialOption) (Session, error) {
// we have at least one channel mapping
if ok {
c.discovered = true
break
links = append(links, link.id)
}
}
t.RUnlock()
// discovered so set the link
// TODO: pick the link efficiently based
// on link status and saturation.
if c.discovered {
// set the link
i := rand.Intn(len(links))
c.link = links[i]
}
// shit fuck
if !c.discovered {
msg := c.newMessage("discover")