TrueCloudLab
/
frostfs-sdk-go
30
0
Fork 19
Code Issues 29 Pull Requests 3 Actions Releases Activity

[#48] client: Refactor EndpointInfo() 

Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
pull/60/head
Evgenii Stratonikov 2023-04-14 16:32:22 +03:00 committed by fyrchik
parent 40d966bec2
commit a16fc40c39
1 changed files with 47 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
client/netmap.go
View File

@ -19,6 +19,16 @@ type PrmEndpointInfo struct {
prmCommonMeta prmCommonMeta
} }
func (x *PrmEndpointInfo) buildRequest(c *Client) (*v2netmap.LocalNodeInfoRequest, error) {
meta := new(v2session.RequestMetaHeader)
writeXHeadersToMeta(x.xHeaders, meta)
req := new(v2netmap.LocalNodeInfoRequest)
req.SetBody(new(v2netmap.LocalNodeInfoRequestBody))
c.prepareRequest(req, meta)
return req, nil
}
// ResEndpointInfo group resulting values of EndpointInfo operation. // ResEndpointInfo group resulting values of EndpointInfo operation.
type ResEndpointInfo struct { type ResEndpointInfo struct {
statusRes statusRes
@ -56,25 +66,25 @@ func (x ResEndpointInfo) NodeInfo() netmap.NodeInfo {
// Return statuses: // Return statuses:
// - global (see Client docs). // - global (see Client docs).
func (c *Client) EndpointInfo(ctx context.Context, prm PrmEndpointInfo) (*ResEndpointInfo, error) { func (c *Client) EndpointInfo(ctx context.Context, prm PrmEndpointInfo) (*ResEndpointInfo, error) {
// form request req, err := prm.buildRequest(c)
var req v2netmap.LocalNodeInfoRequest if err != nil {
return nil, err
// init call context }
var ( if err := signature.SignServiceMessage(&c.prm.key, req); err != nil {
cc contextCall return nil, fmt.Errorf("sign request: %w", err)
res ResEndpointInfo }
)
resp, err := rpcapi.LocalNodeInfo(&c.c, req, client.WithContext(ctx))
c.initCallContext(&cc) if err != nil {
cc.meta = prm.prmCommonMeta return nil, err
cc.req = &req }
cc.statusRes = &res
cc.call = func() (responseV2, error) { var res ResEndpointInfo
return rpcapi.LocalNodeInfo(&c.c, &req, client.WithContext(ctx)) res.st, err = c.processResponse(resp)
if err != nil || !apistatus.IsSuccessful(res.st) {
return &res, err
} }
cc.result = func(r responseV2) {
resp := r.(*v2netmap.LocalNodeInfoResponse)
body := resp.GetBody() body := resp.GetBody()
@ -82,36 +92,21 @@ func (c *Client) EndpointInfo(ctx context.Context, prm PrmEndpointInfo) (*ResEnd
verV2 := body.GetVersion() verV2 := body.GetVersion()
if verV2 == nil { if verV2 == nil {
cc.err = newErrMissingResponseField(fieldVersion) return nil, newErrMissingResponseField(fieldVersion)
return
} }
if err := res.version.ReadFromV2(*verV2); err != nil {
cc.err = res.version.ReadFromV2(*verV2) return nil, newErrInvalidResponseField(fieldVersion, err)
if cc.err != nil {
cc.err = newErrInvalidResponseField(fieldVersion, cc.err)
return
} }
const fieldNodeInfo = "node info" const fieldNodeInfo = "node info"
nodeInfoV2 := body.GetNodeInfo() nodeInfoV2 := body.GetNodeInfo()
if nodeInfoV2 == nil { if nodeInfoV2 == nil {
cc.err = newErrMissingResponseField(fieldNodeInfo) return nil, newErrMissingResponseField(fieldNodeInfo)
return
} }
if err := res.ni.ReadFromV2(*nodeInfoV2); err != nil {
cc.err = res.ni.ReadFromV2(*nodeInfoV2) return nil, newErrInvalidResponseField(fieldNodeInfo, err)
if cc.err != nil {
cc.err = newErrInvalidResponseField(fieldNodeInfo, cc.err)
return
} }
}
// process call
if !cc.processCall() {
return nil, cc.err
}
return &res, nil return &res, nil
} }