TrueCloudLab/frostfs-node
20
1
Fork 27
Code Issues 92 Pull requests 4 Projects Releases 1 Activity Actions

WIP: tree: Sync replication #1468

Closed
dstepanov-yadro wants to merge 1 commit from dstepanov-yadro/frostfs-node:fix/tree_sync_rep into master
pull from: dstepanov-yadro/frostfs-node:fix/tree_sync_rep
merge into: TrueCloudLab:master
Conversation 0 Commits 1 Files changed 2 +10 -8
2 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pkg/services/tree/replicator.go
View file

@ -183,14 +183,15 @@ func (s *Service) replicate(op movePair) error {
return nil
}
func (s *Service) pushToQueue(cid cidSDK.ID, treeID string, op *pilorama.Move) {
func (s *Service) pushToQueue(ctx context.Context, cid cidSDK.ID, treeID string, op *pilorama.Move) {
select {
case s.replicateCh <- movePair{
cid: cid,
treeID: treeID,
op: op,
}:
default:
case <-s.closeCh:
case <-ctx.Done():
}
}

13
pkg/services/tree/service.go
View file

@ -135,7 +135,7 @@ func (s *Service) Add(ctx context.Context, req *AddRequest) (*AddResponse, error
return nil, err
}
s.pushToQueue(cid, b.GetTreeId(), log)
s.pushToQueue(ctx, cid, b.GetTreeId(), log)
return &AddResponse{
Body: &AddResponse_Body{
NodeId: log.Child,
@ -182,7 +182,7 @@ func (s *Service) AddByPath(ctx context.Context, req *AddByPathRequest) (*AddByP
}
for i := range logs {
s.pushToQueue(cid, b.GetTreeId(), &logs[i])
s.pushToQueue(ctx, cid, b.GetTreeId(), &logs[i])
}
nodes := make([]uint64, len(logs))
@ -236,7 +236,7 @@ func (s *Service) Remove(ctx context.Context, req *RemoveRequest) (*RemoveRespon
return nil, err
}
s.pushToQueue(cid, b.GetTreeId(), log)
s.pushToQueue(ctx, cid, b.GetTreeId(), log)
return new(RemoveResponse), nil
}
@ -281,7 +281,7 @@ func (s *Service) Move(ctx context.Context, req *MoveRequest) (*MoveResponse, er
return nil, err
}
s.pushToQueue(cid, b.GetTreeId(), log)
s.pushToQueue(ctx, cid, b.GetTreeId(), log)
return new(MoveResponse), nil
}
@ -586,7 +586,7 @@ func sortByFilename(nodes []pilorama.NodeInfo, d GetSubTreeRequest_Body_Order_Di
}
// Apply locally applies operation from the remote node to the tree.
func (s *Service) Apply(_ context.Context, req *ApplyRequest) (*ApplyResponse, error) {
func (s *Service) Apply(ctx context.Context, req *ApplyRequest) (*ApplyResponse, error) {
err := verifyMessage(req)
if err != nil {
return nil, err
@ -624,7 +624,8 @@ func (s *Service) Apply(_ context.Context, req *ApplyRequest) (*ApplyResponse, e
Meta: meta,
},
}:
default:
case <-s.closeCh:
case <-ctx.Done():
}
return &ApplyResponse{Body: &ApplyResponse_Body{}, Signature: &Signature{}}, nil
}