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

Merge pull request #828 from micro/net-lookup

Browse Source 
Use dns resolver on peer nodes
This commit is contained in:
Asim Aslam 2019-10-08 10:36:04 +01:00 committed by GitHub
commit b886dd4b8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
network/default.go
View File

@ -11,6 +11,7 @@ import (
"github.com/micro/go-micro/client"
rtr "github.com/micro/go-micro/client/selector/router"
pbNet "github.com/micro/go-micro/network/proto"
"github.com/micro/go-micro/network/resolver/dns"
"github.com/micro/go-micro/proxy"
"github.com/micro/go-micro/router"
pbRtr "github.com/micro/go-micro/router/proto"
@ -73,7 +74,6 @@ func newNetwork(opts ...Option) Network {
// init tunnel address to the network bind address
options.Tunnel.Init(
tunnel.Address(options.Address),
tunnel.Nodes(options.Peers...),
)
// init router Id to the network id
@ -174,10 +174,22 @@ func (n *network) resolveNodes() ([]string, error) {
nodeMap[record.Address] = true
}
// use the dns resolver to expand peers
dns := &dns.Resolver{}
// append seed nodes if we have them
for _, node := range n.options.Peers {
if _, ok := nodeMap[node]; !ok {
nodes = append(nodes, node)
// resolve anything that looks like a host name
records, err := dns.Resolve(node)
if err != nil {
continue
}
// add to the node map
for _, record := range records {
if _, ok := nodeMap[record.Address]; !ok {
nodes = append(nodes, record.Address)
}
}
}