forked from TrueCloudLab/frostfs-node
Anton Nikiforov
50dccff7c1
Applicable for both cases: when node uses local cache for netmap and when it disabled. Change-Id: I3050f537e20312a4b39e944aca763b77bd1e74c4 Signed-off-by: Anton Nikiforov <an.nikiforov@yadro.com>
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/netmap"
|
|
netmapClient "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client/netmap"
|
|
netmapSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/netmap"
|
|
)
|
|
|
|
type rawNetmapSource struct {
|
|
client *netmapClient.Client
|
|
}
|
|
|
|
func newRawNetmapStorage(client *netmapClient.Client) netmap.Source {
|
|
return &rawNetmapSource{
|
|
client: client,
|
|
}
|
|
}
|
|
|
|
func (s *rawNetmapSource) GetNetMap(ctx context.Context, diff uint64) (*netmapSDK.NetMap, error) {
|
|
nm, err := s.client.GetNetMap(ctx, diff)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
candidates, err := s.client.GetCandidates(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
updates := getNetMapNodesToUpdate(nm, candidates)
|
|
if len(updates) > 0 {
|
|
mergeNetmapWithCandidates(updates, nm)
|
|
}
|
|
return nm, nil
|
|
}
|
|
|
|
func (s *rawNetmapSource) GetNetMapByEpoch(ctx context.Context, epoch uint64) (*netmapSDK.NetMap, error) {
|
|
nm, err := s.client.GetNetMapByEpoch(ctx, epoch)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
candidates, err := s.client.GetCandidates(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
updates := getNetMapNodesToUpdate(nm, candidates)
|
|
if len(updates) > 0 {
|
|
mergeNetmapWithCandidates(updates, nm)
|
|
}
|
|
return nm, nil
|
|
}
|
|
|
|
func (s *rawNetmapSource) Epoch(ctx context.Context) (uint64, error) {
|
|
return s.client.Epoch(ctx)
|
|
}
|