TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

rpcclient: ensure nns.RecordType can be passed as a parameter to invoker

Browse source 
Implement smartcontract.Convertible for nns.RecordType.

Signed-off-by: Anna Shaleva <shaleva.ann@nspcc.ru>
This commit is contained in:
Anna Shaleva 2024-01-23 14:50:26 +03:00
parent a22a7177e3
commit 161b83fd7f
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
pkg/rpcclient/nns/record.go
View file

@ -3,7 +3,9 @@ package nns
import ( import (
"errors" "errors"
"fmt" "fmt"
"math/big"
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem" "github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
) )
@ -17,6 +19,10 @@ type RecordState struct {
// RecordType is domain name service record types. // RecordType is domain name service record types.
type RecordType byte type RecordType byte
// Ensure RecordType implements smartcontract.Convertible for proper handling as
// a parameter to invoker.Invoker methods.
var _ = smartcontract.Convertible(RecordType(0))
// Record types are defined in [RFC 1035](https://tools.ietf.org/html/rfc1035) // Record types are defined in [RFC 1035](https://tools.ietf.org/html/rfc1035)
const ( const (
// A represents address record type. // A represents address record type.
@ -33,6 +39,14 @@ const (
AAAA RecordType = 28 AAAA RecordType = 28
) )
// ToSCParameter implements smartcontract.Convertible interface.
func (r RecordType) ToSCParameter() (smartcontract.Parameter, error) {
return smartcontract.Parameter{
Type: smartcontract.IntegerType,
Value: big.NewInt(int64(r)),
}, nil
}
// FromStackItem fills RecordState with data from the given stack item if it can // FromStackItem fills RecordState with data from the given stack item if it can
// be correctly converted to RecordState. // be correctly converted to RecordState.
func (r *RecordState) FromStackItem(itm stackitem.Item) error { func (r *RecordState) FromStackItem(itm stackitem.Item) error {