a-savchuk/frostfs-sdk-go
1
0
Fork 0
forked from TrueCloudLab/frostfs-sdk-go
Code Pull requests Activity

[#212] pool: Control sub tree nodes order

Browse source 
Signed-off-by: Alex Vanin <a.vanin@yadro.com>
This commit is contained in:
Alexey Vanin 2024-04-01 13:17:28 +03:00
parent ec0cb2169f
commit 3790142b10
1 changed files with 19 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
pool/tree/pool.go
View file

@ -26,6 +26,16 @@ const (
defaultStreamTimeout = 10 * time.Second defaultStreamTimeout = 10 * time.Second
) )
// SubTreeSort defines an order of nodes returned from GetSubTree RPC.
type SubTreeSort int32
const (
// NoneOrder does not specify order of nodes returned in GetSubTree RPC.
NoneOrder SubTreeSort = iota
// AscendingOrder specifies ascending alphabetical order of nodes based on FilePath attribute.
AscendingOrder
)
var ( var (
// ErrNodeNotFound is returned from Tree service in case of not found error. // ErrNodeNotFound is returned from Tree service in case of not found error.
ErrNodeNotFound = errors.New("not found") ErrNodeNotFound = errors.New("not found")
@ -119,6 +129,7 @@ type GetSubTreeParams struct {
RootID uint64 RootID uint64
Depth uint32 Depth uint32
BearerToken []byte BearerToken []byte
Order SubTreeSort
} }
// AddNodeParams groups parameters of Pool.AddNode operation. // AddNodeParams groups parameters of Pool.AddNode operation.
@ -388,12 +399,17 @@ func (p *Pool) GetSubTree(ctx context.Context, prm GetSubTreeParams) (*SubTreeRe
RootId: prm.RootID, RootId: prm.RootID,
Depth: prm.Depth, Depth: prm.Depth,
BearerToken: prm.BearerToken, BearerToken: prm.BearerToken,
OrderBy: &grpcService.GetSubTreeRequest_Body_Order{ OrderBy: new(grpcService.GetSubTreeRequest_Body_Order),
Direction: grpcService.GetSubTreeRequest_Body_Order_Asc,
},
}, },
} }
switch prm.Order {
case AscendingOrder:
request.Body.OrderBy.Direction = grpcService.GetSubTreeRequest_Body_Order_Asc
default:
request.Body.OrderBy.Direction = grpcService.GetSubTreeRequest_Body_Order_None
}
if err := p.signRequest(request.Body, func(key, sign []byte) { if err := p.signRequest(request.Body, func(key, sign []byte) {
request.Signature = &grpcService.Signature{ request.Signature = &grpcService.Signature{
Key: key, Key: key,