0de949b575
They're useless in a package named 'stackitem', make this package a bit more user-friendly.
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package native
|
|
|
|
import (
|
|
"math/big"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
)
|
|
|
|
type candidate struct {
|
|
Registered bool
|
|
Votes big.Int
|
|
}
|
|
|
|
// Bytes marshals c to byte array.
|
|
func (c *candidate) Bytes() []byte {
|
|
w := io.NewBufBinWriter()
|
|
stackitem.EncodeBinary(c.toStackItem(), w.BinWriter)
|
|
return w.Bytes()
|
|
}
|
|
|
|
// FromBytes unmarshals candidate from byte array.
|
|
func (c *candidate) FromBytes(data []byte) *candidate {
|
|
r := io.NewBinReaderFromBuf(data)
|
|
item := stackitem.DecodeBinary(r)
|
|
if r.Err != nil {
|
|
panic(r.Err)
|
|
}
|
|
return c.fromStackItem(item)
|
|
}
|
|
|
|
func (c *candidate) toStackItem() stackitem.Item {
|
|
return stackitem.NewStruct([]stackitem.Item{
|
|
stackitem.NewBool(c.Registered),
|
|
stackitem.NewBigInteger(&c.Votes),
|
|
})
|
|
}
|
|
|
|
func (c *candidate) fromStackItem(item stackitem.Item) *candidate {
|
|
arr := item.(*stackitem.Struct).Value().([]stackitem.Item)
|
|
vs, err := arr[1].TryInteger()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
c.Registered, err = arr[0].TryBool()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
c.Votes = *vs
|
|
return c
|
|
}
|