forked from TrueCloudLab/frostfs-node
[#452] morph/client: Add reputation contract client wrapper
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
parent
9f1a49a562
commit
b9a1aaec23
4 changed files with 204 additions and 0 deletions
84
pkg/morph/client/reputation/wrapper/get.go
Normal file
84
pkg/morph/client/reputation/wrapper/get.go
Normal file
|
@ -0,0 +1,84 @@
|
||||||
|
package wrapper
|
||||||
|
|
||||||
|
import (
|
||||||
|
reputationClient "github.com/nspcc-dev/neofs-node/pkg/morph/client/reputation"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
// GetArgs groups the arguments of "get reputation value" test invocation.
|
||||||
|
GetArgs struct {
|
||||||
|
epoch uint64
|
||||||
|
peerID reputation.PeerID
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetByIDArgs groups the arguments of "get reputation value by
|
||||||
|
// reputation id" test invocation.
|
||||||
|
GetByIDArgs struct {
|
||||||
|
id ReputationID
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetResults groups the result of "get reputation value" and
|
||||||
|
// "get reputation value by reputation id" test invocations.
|
||||||
|
GetResult struct {
|
||||||
|
reputations [][]byte // todo: replace with the slice of structures
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// 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 reputation.PeerID) {
|
||||||
|
g.peerID = v
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetID sets id of expected reputation value in reputation contract.
|
||||||
|
func (g *GetByIDArgs) SetID(v ReputationID) {
|
||||||
|
g.id = v
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reputations returns slice of reputation values.
|
||||||
|
func (g GetResult) Reputations() [][]byte {
|
||||||
|
return g.reputations
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get invokes the call of "get reputation value" method of reputation contract.
|
||||||
|
func (w *ClientWrapper) Get(v GetArgs) (*GetResult, error) {
|
||||||
|
args := reputationClient.GetArgs{}
|
||||||
|
args.SetEpoch(v.epoch)
|
||||||
|
args.SetPeerID(v.peerID.Bytes())
|
||||||
|
|
||||||
|
data, err := (*reputationClient.Client)(w).Get(args)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return parseGetResult(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetByID invokes the call of "get reputation value by reputation id" method
|
||||||
|
// of reputation contract.
|
||||||
|
func (w *ClientWrapper) GetByID(v GetByIDArgs) (*GetResult, error) {
|
||||||
|
args := reputationClient.GetByIDArgs{}
|
||||||
|
args.SetID(v.id)
|
||||||
|
|
||||||
|
data, err := (*reputationClient.Client)(w).GetByID(args)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return parseGetResult(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseGetResult(data *reputationClient.GetResult) (*GetResult, error) {
|
||||||
|
rawReputations := data.Reputations()
|
||||||
|
|
||||||
|
// todo: unmarshal all reputation values into structure
|
||||||
|
|
||||||
|
return &GetResult{
|
||||||
|
reputations: rawReputations,
|
||||||
|
}, nil
|
||||||
|
}
|
55
pkg/morph/client/reputation/wrapper/list.go
Normal file
55
pkg/morph/client/reputation/wrapper/list.go
Normal file
|
@ -0,0 +1,55 @@
|
||||||
|
package wrapper
|
||||||
|
|
||||||
|
import (
|
||||||
|
reputationClient "github.com/nspcc-dev/neofs-node/pkg/morph/client/reputation"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
// ReputationID is an ID of the reputation record in reputation contract.
|
||||||
|
ReputationID []byte
|
||||||
|
|
||||||
|
// ListByEpochArgs groups the arguments of
|
||||||
|
// "list reputation ids by epoch" test invoke call.
|
||||||
|
ListByEpochArgs struct {
|
||||||
|
epoch uint64
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListByEpochResult groups the result of "list reputation ids by epoch"
|
||||||
|
// test invoke.
|
||||||
|
ListByEpochResult struct {
|
||||||
|
ids []ReputationID
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// SetEpoch sets epoch of expected reputation ids.
|
||||||
|
func (l *ListByEpochArgs) SetEpoch(v uint64) {
|
||||||
|
l.epoch = v
|
||||||
|
}
|
||||||
|
|
||||||
|
// IDs returns slice of reputation id values.
|
||||||
|
func (l ListByEpochResult) IDs() []ReputationID {
|
||||||
|
return l.ids
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListByEpoch invokes the call of "list reputation ids by epoch" method of
|
||||||
|
// reputation contract.
|
||||||
|
func (w *ClientWrapper) ListByEpoch(v ListByEpochArgs) (*ListByEpochResult, error) {
|
||||||
|
args := reputationClient.ListByEpochArgs{}
|
||||||
|
args.SetEpoch(v.epoch)
|
||||||
|
|
||||||
|
data, err := (*reputationClient.Client)(w).ListByEpoch(args)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
ids := data.IDs()
|
||||||
|
|
||||||
|
result := make([]ReputationID, 0, len(ids))
|
||||||
|
for i := range ids {
|
||||||
|
result = append(result, ids[i])
|
||||||
|
}
|
||||||
|
|
||||||
|
return &ListByEpochResult{
|
||||||
|
ids: result,
|
||||||
|
}, nil
|
||||||
|
}
|
51
pkg/morph/client/reputation/wrapper/put.go
Normal file
51
pkg/morph/client/reputation/wrapper/put.go
Normal file
|
@ -0,0 +1,51 @@
|
||||||
|
package wrapper
|
||||||
|
|
||||||
|
import (
|
||||||
|
reputationClient "github.com/nspcc-dev/neofs-node/pkg/morph/client/reputation"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
// PutArgs groups the arguments of "put reputation value" invocation call.
|
||||||
|
PutArgs struct {
|
||||||
|
epoch uint64
|
||||||
|
peerID reputation.PeerID
|
||||||
|
value []byte // todo: replace with struct
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// SetEpoch sets epoch of reputation value.
|
||||||
|
func (p *PutArgs) SetEpoch(v uint64) {
|
||||||
|
p.epoch = v
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetPeerID sets peer id of reputation value.
|
||||||
|
func (p *PutArgs) SetPeerID(v reputation.PeerID) {
|
||||||
|
p.peerID = v
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetValue sets reputation value.
|
||||||
|
func (p *PutArgs) SetValue(v []byte) {
|
||||||
|
p.value = v
|
||||||
|
}
|
||||||
|
|
||||||
|
// Put invokes direct call of "put reputation value" method of reputation contract.
|
||||||
|
func (w *ClientWrapper) Put(v PutArgs) error {
|
||||||
|
args := reputationClient.PutArgs{}
|
||||||
|
args.SetEpoch(v.epoch)
|
||||||
|
args.SetPeerID(v.peerID.Bytes())
|
||||||
|
args.SetValue(v.value) // todo: marshal reputation value to `value` bytes there
|
||||||
|
|
||||||
|
return (*reputationClient.Client)(w).Put(args)
|
||||||
|
}
|
||||||
|
|
||||||
|
// PutViaNotary invokes notary call of "put reputation value" method of
|
||||||
|
// reputation contract.
|
||||||
|
func (w *ClientWrapper) PutViaNotary(v PutArgs) error {
|
||||||
|
args := reputationClient.PutArgs{}
|
||||||
|
args.SetEpoch(v.epoch)
|
||||||
|
args.SetPeerID(v.peerID.Bytes())
|
||||||
|
args.SetValue(v.value) // todo: marshal reputation value to `value` bytes there
|
||||||
|
|
||||||
|
return (*reputationClient.Client)(w).PutViaNotary(args)
|
||||||
|
}
|
14
pkg/morph/client/reputation/wrapper/wrapper.go
Normal file
14
pkg/morph/client/reputation/wrapper/wrapper.go
Normal file
|
@ -0,0 +1,14 @@
|
||||||
|
package wrapper
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/morph/client/reputation"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ClientWrapper is a wrapper over reputation contract
|
||||||
|
// client which implements storage of reputation values.
|
||||||
|
type ClientWrapper reputation.Client
|
||||||
|
|
||||||
|
// WrapClient wraps reputation contract client and returns ClientWrapper instance.
|
||||||
|
func WrapClient(c *reputation.Client) *ClientWrapper {
|
||||||
|
return (*ClientWrapper)(c)
|
||||||
|
}
|
Loading…
Reference in a new issue