Update dialer to have peer addrs map

This commit is contained in:
mcrakhman 2023-05-05 16:18:46 +02:00
parent ba5ca3d6be
commit 590e8080b8
No known key found for this signature in database
GPG Key ID: DED12CFEF5B8396B

View File

@ -32,12 +32,11 @@ var (
var log = logger.NewNamed(CName) var log = logger.NewNamed(CName)
func New() Dialer { func New() Dialer {
return &dialer{} return &dialer{peerAddrs: map[string][]string{}}
} }
type Dialer interface { type Dialer interface {
Dial(ctx context.Context, peerId string) (peer peer.Peer, err error) Dial(ctx context.Context, peerId string) (peer peer.Peer, err error)
UpdateAddrs(addrs map[string][]string)
SetPeerAddrs(peerId string, addrs []string) SetPeerAddrs(peerId string, addrs []string)
app.Component app.Component
} }
@ -62,18 +61,9 @@ func (d *dialer) Name() (name string) {
return CName return CName
} }
func (d *dialer) UpdateAddrs(addrs map[string][]string) {
d.mu.Lock()
d.peerAddrs = addrs
d.mu.Unlock()
}
func (d *dialer) SetPeerAddrs(peerId string, addrs []string) { func (d *dialer) SetPeerAddrs(peerId string, addrs []string) {
d.mu.Lock() d.mu.Lock()
defer d.mu.Unlock() defer d.mu.Unlock()
if d.peerAddrs == nil {
return
}
d.peerAddrs[peerId] = addrs d.peerAddrs[peerId] = addrs
} }