forked from TrueCloudLab/frostfs-sdk-go
[#48] client: Refactor EndpointInfo()
Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
This commit is contained in:
parent
40d966bec2
commit
a16fc40c39
1 changed files with 47 additions and 52 deletions
|
@ -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,62 +66,47 @@ 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 (
|
|
||||||
cc contextCall
|
|
||||||
res ResEndpointInfo
|
|
||||||
)
|
|
||||||
|
|
||||||
c.initCallContext(&cc)
|
|
||||||
cc.meta = prm.prmCommonMeta
|
|
||||||
cc.req = &req
|
|
||||||
cc.statusRes = &res
|
|
||||||
cc.call = func() (responseV2, error) {
|
|
||||||
return rpcapi.LocalNodeInfo(&c.c, &req, client.WithContext(ctx))
|
|
||||||
}
|
|
||||||
cc.result = func(r responseV2) {
|
|
||||||
resp := r.(*v2netmap.LocalNodeInfoResponse)
|
|
||||||
|
|
||||||
body := resp.GetBody()
|
|
||||||
|
|
||||||
const fieldVersion = "version"
|
|
||||||
|
|
||||||
verV2 := body.GetVersion()
|
|
||||||
if verV2 == nil {
|
|
||||||
cc.err = newErrMissingResponseField(fieldVersion)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
cc.err = res.version.ReadFromV2(*verV2)
|
|
||||||
if cc.err != nil {
|
|
||||||
cc.err = newErrInvalidResponseField(fieldVersion, cc.err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
const fieldNodeInfo = "node info"
|
|
||||||
|
|
||||||
nodeInfoV2 := body.GetNodeInfo()
|
|
||||||
if nodeInfoV2 == nil {
|
|
||||||
cc.err = newErrMissingResponseField(fieldNodeInfo)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
cc.err = res.ni.ReadFromV2(*nodeInfoV2)
|
|
||||||
if cc.err != nil {
|
|
||||||
cc.err = newErrInvalidResponseField(fieldNodeInfo, cc.err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// process call
|
if err := signature.SignServiceMessage(&c.prm.key, req); err != nil {
|
||||||
if !cc.processCall() {
|
return nil, fmt.Errorf("sign request: %w", err)
|
||||||
return nil, cc.err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
resp, err := rpcapi.LocalNodeInfo(&c.c, req, client.WithContext(ctx))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var res ResEndpointInfo
|
||||||
|
res.st, err = c.processResponse(resp)
|
||||||
|
if err != nil || !apistatus.IsSuccessful(res.st) {
|
||||||
|
return &res, err
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.GetBody()
|
||||||
|
|
||||||
|
const fieldVersion = "version"
|
||||||
|
|
||||||
|
verV2 := body.GetVersion()
|
||||||
|
if verV2 == nil {
|
||||||
|
return nil, newErrMissingResponseField(fieldVersion)
|
||||||
|
}
|
||||||
|
if err := res.version.ReadFromV2(*verV2); err != nil {
|
||||||
|
return nil, newErrInvalidResponseField(fieldVersion, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
const fieldNodeInfo = "node info"
|
||||||
|
|
||||||
|
nodeInfoV2 := body.GetNodeInfo()
|
||||||
|
if nodeInfoV2 == nil {
|
||||||
|
return nil, newErrMissingResponseField(fieldNodeInfo)
|
||||||
|
}
|
||||||
|
if err := res.ni.ReadFromV2(*nodeInfoV2); err != nil {
|
||||||
|
return nil, newErrInvalidResponseField(fieldNodeInfo, err)
|
||||||
|
}
|
||||||
return &res, nil
|
return &res, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue