[#40] morph/client: Implement read config method

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
Leonard Lyubich 2020-10-22 17:19:07 +03:00 committed by Alex Vanin
parent d05b2ff500
commit 90c38fc5e5
2 changed files with 79 additions and 1 deletions

View file

@ -37,7 +37,8 @@ type cfg struct {
snapshotMethod, // get network map snapshot method name snapshotMethod, // get network map snapshot method name
updateStateMethod, // update state method name for invocation updateStateMethod, // update state method name for invocation
innerRingListMethod, // IR list method name for invocation innerRingListMethod, // IR list method name for invocation
epochMethod string // get epoch number method name epochMethod, // get epoch number method name
configMethod string // get config value method name
} }
const ( const (
@ -48,6 +49,7 @@ const (
defaultUpdateStateMethod = "updateState" // default update state method name defaultUpdateStateMethod = "updateState" // default update state method name
defaultInnerRIngListMethod = "innerRingList" // default IR list method name defaultInnerRIngListMethod = "innerRingList" // default IR list method name
defaultEpochMethod = "epoch" // default get epoch number method name defaultEpochMethod = "epoch" // default get epoch number method name
defaultConfigMethod = "config" // default get config value method name
) )
func defaultConfig() *cfg { func defaultConfig() *cfg {
@ -59,6 +61,7 @@ func defaultConfig() *cfg {
updateStateMethod: defaultUpdateStateMethod, updateStateMethod: defaultUpdateStateMethod,
innerRingListMethod: defaultInnerRIngListMethod, innerRingListMethod: defaultInnerRIngListMethod,
epochMethod: defaultEpochMethod, epochMethod: defaultEpochMethod,
configMethod: defaultConfigMethod,
} }
} }
@ -177,3 +180,17 @@ func WithEpochMethod(n string) Option {
} }
} }
} }
// WithConfigMethod returns a client constructor option that
// specifies the method name of config value receiving operation.
//
// Ignores empty value.
//
// If option not provided, "config" is used.
func WithConfigMethod(n string) Option {
return func(c *cfg) {
if n != "" {
c.configMethod = n
}
}
}

View file

@ -0,0 +1,61 @@
package netmap
import (
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
"github.com/nspcc-dev/neofs-node/pkg/morph/client"
"github.com/pkg/errors"
)
// ConfigArgs groups the arguments
// of get config value test invoke call.
type ConfigArgs struct {
key []byte
}
// EpochValues groups the stack parameters
// returned by get epoch number test invoke.
type ConfigValues struct {
val interface{}
}
// SetKey sets binary key to configuration parameter.
func (c *ConfigArgs) SetKey(v []byte) {
c.key = v
}
// Value returns configuration value.
func (c ConfigValues) Value() interface{} {
return c.val
}
// Config performs the test invoke of get config value
// method of NeoFS Netmap contract.
func (c *Client) Config(args ConfigArgs, assert func(stackitem.Item) (interface{}, error)) (*ConfigValues, error) {
items, err := c.client.TestInvoke(
c.configMethod,
args.key,
)
if err != nil {
return nil, errors.Wrapf(err,
"could not perform test invocation (%s)",
c.configMethod)
}
if ln := len(items); ln != 1 {
return nil, errors.Errorf("unexpected stack item count (%s): %d",
c.configMethod, ln)
}
val, err := assert(items[0])
if err != nil {
return nil, errors.Wrap(err, "value type assertion failed")
}
return &ConfigValues{
val: val,
}, nil
}
func IntegerAssert(item stackitem.Item) (interface{}, error) {
return client.IntFromStackItem(item)
}