Skip to content

Commit

Permalink
[#607] network: Make ClientCache to accept AddressGroup
Browse files Browse the repository at this point in the history
Change type of the `ClientCache.Get` method's parameter to `AddressGroup`.
Use `GroupFromAddress` to call the method from the wrappers in order to no
change their interface.

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
  • Loading branch information
Leonard Lyubich authored and cthulhu-rider committed Jun 28, 2021
1 parent 5db7c5c commit e11f50e
Show file tree
Hide file tree
Showing 5 changed files with 16 additions and 11 deletions.
4 changes: 2 additions & 2 deletions cmd/neofs-node/container.go
Original file line number Diff line number Diff line change
Expand Up @@ -205,7 +205,7 @@ type remoteLoadAnnounceProvider struct {
loadAddrSrc network.LocalAddressSource

clientCache interface {
Get(network.Address) (apiClient.Client, error)
Get(network.AddressGroup) (apiClient.Client, error)
}

deadEndProvider loadcontroller.WriterProvider
Expand All @@ -230,7 +230,7 @@ func (r *remoteLoadAnnounceProvider) InitRemote(srv loadroute.ServerInfo) (loadc
return loadcontroller.SimpleWriterProvider(new(nopLoadWriter)), nil
}

c, err := r.clientCache.Get(netAddr)
c, err := r.clientCache.Get(network.GroupFromAddress(netAddr))
if err != nil {
return nil, fmt.Errorf("could not initialize API client: %w", err)
}
Expand Down
4 changes: 2 additions & 2 deletions cmd/neofs-node/object.go
Original file line number Diff line number Diff line change
Expand Up @@ -422,7 +422,7 @@ type reputationClientConstructor struct {
trustStorage *truststorage.Storage

basicConstructor interface {
Get(network.Address) (client.Client, error)
Get(network.AddressGroup) (client.Client, error)
}
}

Expand Down Expand Up @@ -507,7 +507,7 @@ func (c *reputationClient) SearchObject(ctx context.Context, prm *client.SearchO
}

func (c *reputationClientConstructor) Get(addr network.Address) (client.Client, error) {
cl, err := c.basicConstructor.Get(addr)
cl, err := c.basicConstructor.Get(network.GroupFromAddress(addr))
if err != nil {
return nil, err
}
Expand Down
4 changes: 2 additions & 2 deletions cmd/neofs-node/reputation/common/remote.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ import (
)

type clientCache interface {
Get(network.Address) (apiClient.Client, error)
Get(network.AddressGroup) (apiClient.Client, error)
}

// clientKeyRemoteProvider must provide remote writer and take into account
Expand Down Expand Up @@ -84,7 +84,7 @@ func (rtp *remoteTrustProvider) InitRemote(srv reputationcommon.ServerInfo) (rep
return trustcontroller.SimpleWriterProvider(new(NopReputationWriter)), nil
}

c, err := rtp.clientCache.Get(netAddr)
c, err := rtp.clientCache.Get(network.GroupFromAddress(netAddr))
if err != nil {
return nil, fmt.Errorf("could not initialize API client: %w", err)
}
Expand Down
4 changes: 2 additions & 2 deletions pkg/innerring/rpc.go
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ type (
ClientCache struct {
log *zap.Logger
cache interface {
Get(address network.Address) (client.Client, error)
Get(address network.AddressGroup) (client.Client, error)
CloseAll()
}
key *ecdsa.PrivateKey
Expand Down Expand Up @@ -52,7 +52,7 @@ func newClientCache(p *clientCacheParams) *ClientCache {
func (c *ClientCache) Get(address network.Address) (client.Client, error) {
// Because cache is used by `ClientCache` exclusively,
// client will always have valid key.
return c.cache.Get(address)
return c.cache.Get(network.GroupFromAddress(address))
}

// GetSG polls the container from audit task to get the object by id.
Expand Down
11 changes: 8 additions & 3 deletions pkg/network/cache/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -28,11 +28,16 @@ func NewSDKClientCache(opts ...client.Option) *ClientCache {
}

// Get function returns existing client or creates a new one.
func (c *ClientCache) Get(netAddr network.Address) (client.Client, error) {
func (c *ClientCache) Get(netAddr network.AddressGroup) (client.Client, error) {
// multiaddr is used as a key in client cache since
// same host may have different connections(with tls or not),
// therefore, host+port pair is not unique
mAddr := netAddr.String()

// FIXME: we should calculate map key regardless of the address order,
// but network.StringifyGroup is order-dependent.
// This works until the same mixed group is transmitted
// (for a network map, it seems to be true).
mAddr := network.StringifyGroup(netAddr)

c.mu.RLock()
if cli, ok := c.clients[mAddr]; ok {
Expand All @@ -53,7 +58,7 @@ func (c *ClientCache) Get(netAddr network.Address) (client.Client, error) {
return cli, nil
}

cli := newMultiClient(network.GroupFromAddress(netAddr), c.opts)
cli := newMultiClient(netAddr, c.opts)

c.clients[mAddr] = cli

Expand Down

0 comments on commit e11f50e

Please sign in to comment.