forked from TrueCloudLab/frostfs-node
[#452] morph/client: Add reputation contract client
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
parent
ee29ee2c47
commit
9f1a49a562
4 changed files with 337 additions and 0 deletions
99
pkg/morph/client/reputation/get.go
Normal file
99
pkg/morph/client/reputation/get.go
Normal file
|
@ -0,0 +1,99 @@
|
|||
package reputation
|
||||
|
||||
import (
|
||||
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/morph/client"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
// GetArgs groups the arguments of "get reputation value" test invocation.
|
||||
type GetArgs struct {
|
||||
epoch uint64
|
||||
peerID []byte // object of reputation evaluation
|
||||
}
|
||||
|
||||
// GetByIDArgs groups the arguments of "get reputation value by reputation id"
|
||||
// test invocation.
|
||||
type GetByIDArgs struct {
|
||||
id []byte // id of reputation value in reputation contract
|
||||
}
|
||||
|
||||
// GetResult groups the stack parameters returned by
|
||||
// "get" and "get by id" test invocations.
|
||||
type GetResult struct {
|
||||
reputations [][]byte
|
||||
}
|
||||
|
||||
// SetEpoch sets epoch of expected reputation value.
|
||||
func (g *GetArgs) SetEpoch(v uint64) {
|
||||
g.epoch = v
|
||||
}
|
||||
|
||||
// SetPeerID sets peer id of expected reputation value.
|
||||
func (g *GetArgs) SetPeerID(v []byte) {
|
||||
g.peerID = v
|
||||
}
|
||||
|
||||
// SetID sets id of expected reputation value in reputation contract.
|
||||
func (g *GetByIDArgs) SetID(v []byte) {
|
||||
g.id = v
|
||||
}
|
||||
|
||||
// Reputations returns slice of marshalled reputation values.
|
||||
func (g GetResult) Reputations() [][]byte {
|
||||
return g.reputations
|
||||
}
|
||||
|
||||
// Get invokes the call of "get reputation value" method of reputation contract.
|
||||
func (c *Client) Get(args GetArgs) (*GetResult, error) {
|
||||
prms, err := c.client.TestInvoke(
|
||||
c.getMethod,
|
||||
int64(args.epoch),
|
||||
args.peerID,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, errors.Wrapf(err, "could not perform test invocation (%s)", c.getMethod)
|
||||
}
|
||||
|
||||
return parseReputations(prms, c.getMethod)
|
||||
}
|
||||
|
||||
// GetByID invokes the call of "get reputation value by reputation id" method
|
||||
// of reputation contract.
|
||||
func (c *Client) GetByID(args GetByIDArgs) (*GetResult, error) {
|
||||
prms, err := c.client.TestInvoke(
|
||||
c.getByIDMethod,
|
||||
args.id,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, errors.Wrapf(err, "could not perform test invocation (%s)", c.getByIDMethod)
|
||||
}
|
||||
|
||||
return parseReputations(prms, c.getByIDMethod)
|
||||
}
|
||||
|
||||
func parseReputations(items []stackitem.Item, method string) (*GetResult, error) {
|
||||
if ln := len(items); ln != 1 {
|
||||
return nil, errors.Errorf("unexpected stack item count (%s): %d", method, ln)
|
||||
}
|
||||
|
||||
items, err := client.ArrayFromStackItem(items[0])
|
||||
if err != nil {
|
||||
return nil, errors.Wrapf(err, "could not get stack item array from stack item (%s)", method)
|
||||
}
|
||||
|
||||
res := &GetResult{
|
||||
reputations: make([][]byte, 0, len(items)),
|
||||
}
|
||||
|
||||
for i := range items {
|
||||
rawReputation, err := client.BytesFromStackItem(items[i])
|
||||
if err != nil {
|
||||
return nil, errors.Wrapf(err, "could not get byte array from stack item (%s)", method)
|
||||
}
|
||||
|
||||
res.reputations = append(res.reputations, rawReputation)
|
||||
}
|
||||
|
||||
return res, nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue