status-im
/
go-libp2p
mirror of https://github.com/status-im/go-libp2p.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

make NewStream run FindPeer if no addresses found

This commit is contained in:
Jeromy 2015-02-14 19:10:15 +00:00
parent 01be69d391
commit 90fa929051
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
host/routed/routed.go
View File

@ -100,6 +100,14 @@ func (rh *RoutedHost) SetStreamHandler(pid protocol.ID, handler inet.StreamHandl
rh.host.SetStreamHandler(pid, handler) rh.host.SetStreamHandler(pid, handler)
} }
func (rh *RoutedHost) NewStream(pid protocol.ID, p peer.ID) (inet.Stream, error) { func (rh *RoutedHost) NewStream(pid protocol.ID, p peer.ID) (inet.Stream, error) {
if len(rh.Peerstore().Addrs(p)) < 1 {
ctx, _ := context.WithTimeout(context.TODO(), time.Second*30)
pi, err := rh.route.FindPeer(ctx, p)
if err != nil {
return nil, err
}
rh.Peerstore().AddAddrs(p, pi.Addrs, peer.TempAddrTTL)
}
return rh.host.NewStream(pid, p) return rh.host.NewStream(pid, p)
} }
func (rh *RoutedHost) Close() error { func (rh *RoutedHost) Close() error {