forked from TrueCloudLab/frostfs-node
[#2165] services/tree: Parallelize synchronization
Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
This commit is contained in:
parent
de9957e076
commit
c299b98afe
2 changed files with 30 additions and 6 deletions
|
@ -11,6 +11,7 @@ import (
|
|||
"github.com/TrueCloudLab/frostfs-sdk-go/container/acl"
|
||||
cidSDK "github.com/TrueCloudLab/frostfs-sdk-go/container/id"
|
||||
netmapSDK "github.com/TrueCloudLab/frostfs-sdk-go/netmap"
|
||||
"github.com/panjf2000/ants/v2"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
|
@ -26,6 +27,7 @@ type Service struct {
|
|||
containerCache containerCache
|
||||
|
||||
syncChan chan struct{}
|
||||
syncPool *ants.Pool
|
||||
cnrMap map[cidSDK.ID]struct{}
|
||||
}
|
||||
|
||||
|
@ -54,6 +56,7 @@ func New(opts ...Option) *Service {
|
|||
s.containerCache.init(s.containerCacheSize)
|
||||
s.cnrMap = make(map[cidSDK.ID]struct{})
|
||||
s.syncChan = make(chan struct{})
|
||||
s.syncPool, _ = ants.NewPool(defaultSyncWorkerCount)
|
||||
|
||||
return &s
|
||||
}
|
||||
|
@ -75,6 +78,7 @@ func (s *Service) Start(ctx context.Context) {
|
|||
// Shutdown shutdowns the service.
|
||||
func (s *Service) Shutdown() {
|
||||
close(s.closeCh)
|
||||
s.syncPool.Release()
|
||||
}
|
||||
|
||||
func (s *Service) Add(ctx context.Context, req *AddRequest) (*AddResponse, error) {
|
||||
|
|
|
@ -7,12 +7,14 @@ import (
|
|||
"fmt"
|
||||
"io"
|
||||
"math/rand"
|
||||
"sync"
|
||||
|
||||
"github.com/TrueCloudLab/frostfs-node/pkg/local_object_storage/pilorama"
|
||||
"github.com/TrueCloudLab/frostfs-node/pkg/morph/client/netmap"
|
||||
"github.com/TrueCloudLab/frostfs-node/pkg/network"
|
||||
cid "github.com/TrueCloudLab/frostfs-sdk-go/container/id"
|
||||
netmapSDK "github.com/TrueCloudLab/frostfs-sdk-go/netmap"
|
||||
"github.com/panjf2000/ants/v2"
|
||||
"go.uber.org/zap"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/credentials/insecure"
|
||||
|
@ -22,6 +24,8 @@ import (
|
|||
// because the node is not included in the container.
|
||||
var ErrNotInContainer = errors.New("node is not in container")
|
||||
|
||||
const defaultSyncWorkerCount = 20
|
||||
|
||||
// SynchronizeAllTrees synchronizes all the trees of the container. It fetches
|
||||
// tree IDs from the other container nodes. Returns ErrNotInContainer if the node
|
||||
// is not included in the container.
|
||||
|
@ -271,17 +275,33 @@ func (s *Service) syncLoop(ctx context.Context) {
|
|||
}
|
||||
|
||||
// sync new containers
|
||||
var wg sync.WaitGroup
|
||||
for _, cnr := range cnrsToSync {
|
||||
wg.Add(1)
|
||||
cnr := cnr
|
||||
err := s.syncPool.Submit(func() {
|
||||
defer wg.Done()
|
||||
s.log.Debug("syncing container trees...", zap.Stringer("cid", cnr))
|
||||
|
||||
err = s.SynchronizeAllTrees(ctx, cnr)
|
||||
err := s.SynchronizeAllTrees(ctx, cnr)
|
||||
if err != nil {
|
||||
s.log.Error("could not sync trees", zap.Stringer("cid", cnr), zap.Error(err))
|
||||
continue
|
||||
return
|
||||
}
|
||||
|
||||
s.log.Debug("container trees have been synced", zap.Stringer("cid", cnr))
|
||||
})
|
||||
if err != nil {
|
||||
wg.Done()
|
||||
s.log.Error("could not query trees for synchronization",
|
||||
zap.Stringer("cid", cnr),
|
||||
zap.Error(err))
|
||||
if errors.Is(err, ants.ErrPoolClosed) {
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
wg.Wait()
|
||||
|
||||
// remove stored redundant trees
|
||||
for cnr := range s.cnrMap {
|
||||
|
|
Loading…
Reference in a new issue