Skip to content

Commit

Permalink
smartcontract: add support for maps in NewParameterFromValue
Browse files Browse the repository at this point in the history
It's just not possible to use maps in invokers/actors without this. And maps
have too many combinations to try pushing them into a type switch, that's
where reflection kicks in and solves it easily.

Signed-off-by: Roman Khimov <roman@nspcc.ru>
  • Loading branch information
roman-khimov committed Nov 15, 2024
1 parent 5040e35 commit 53a1d10
Show file tree
Hide file tree
Showing 3 changed files with 70 additions and 8 deletions.
6 changes: 3 additions & 3 deletions pkg/rpcclient/invoker/invoker_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -83,17 +83,17 @@ func TestInvoker(t *testing.T) {
require.NoError(t, err)
require.Equal(t, resExp, res)

_, err = inv.Verify(util.Uint160{}, nil, make(map[int]int))
_, err = inv.Verify(util.Uint160{}, nil, make(chan struct{}))
require.Error(t, err)

_, err = inv.Call(util.Uint160{}, "method", make(map[int]int))
_, err = inv.Call(util.Uint160{}, "method", make(chan struct{}))
require.Error(t, err)

res, err = inv.CallAndExpandIterator(util.Uint160{}, "method", 10, 42)
require.NoError(t, err)
require.Equal(t, resExp, res)

_, err = inv.CallAndExpandIterator(util.Uint160{}, "method", 10, make(map[int]int))
_, err = inv.CallAndExpandIterator(util.Uint160{}, "method", 10, make(chan struct{}))
require.Error(t, err)
}
t.Run("standard", func(t *testing.T) {
Expand Down
16 changes: 16 additions & 0 deletions pkg/smartcontract/parameter.go
Original file line number Diff line number Diff line change
Expand Up @@ -362,6 +362,22 @@ func NewParameterFromValue(value any) (Parameter, error) {
}
result.Type = ArrayType
result.Value = res
case reflect.Map:
res := make([]ParameterPair, 0, rv.Len())
iter := rv.MapRange()
for iter.Next() {
k, err := NewParameterFromValue(iter.Key().Interface())
if err != nil {
return result, err
}
v, err := NewParameterFromValue(iter.Value().Interface())
if err != nil {
return result, err
}
res = append(res, ParameterPair{Key: k, Value: v})
}
result.Type = MapType
result.Value = res
default:
return result, fmt.Errorf("%w: %T type", errors.ErrUnsupported, value)
}
Expand Down
56 changes: 51 additions & 5 deletions pkg/smartcontract/parameter_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -887,12 +887,58 @@ func TestParameterFromValue(t *testing.T) {
err: "invalid i value",
},
{
value: make(map[string]int),
err: "unsupported operation: map[string]int type",
value: make(map[string]int),
expType: MapType,
expVal: []ParameterPair{},
},
{
value: []any{1, 2, make(map[string]int)},
err: "unsupported operation: map[string]int type",
value: make(map[string]int),
expType: MapType,
expVal: []ParameterPair{},
},
{
value: map[string]string{"key": "value"},
expType: MapType,
expVal: []ParameterPair{{
Key: Parameter{
Type: StringType,
Value: "key",
},
Value: Parameter{
Type: StringType,
Value: "value",
},
}},
},
{
value: map[int]int64{42: 100500},
expType: MapType,
expVal: []ParameterPair{{
Key: Parameter{
Type: IntegerType,
Value: big.NewInt(42),
},
Value: Parameter{
Type: IntegerType,
Value: big.NewInt(100500),
},
}},
},
{
value: make(chan int),
err: "unsupported operation: chan int type",
},
{
value: []any{1, 2, make(chan int)},
err: "unsupported operation: chan int type",
},
{
value: map[string]chan int{"aaa": make(chan int)},
err: "unsupported operation: chan int type",
},
{
value: map[error]string{errors.New("some"): "value"},
err: "unsupported operation: *errors.errorString type",
},
}

Expand Down Expand Up @@ -924,6 +970,6 @@ func TestParametersFromValues(t *testing.T) {
Type: ByteArrayType,
Value: []byte{3, 2, 1},
}}, res)
_, err = NewParametersFromValues(42, make(map[int]int), []byte{3, 2, 1})
_, err = NewParametersFromValues(42, make(chan int), []byte{3, 2, 1})
require.Error(t, err)
}

0 comments on commit 53a1d10

Please sign in to comment.