tutus-chain/pkg/core/native/util.go

71 lines
1.5 KiB
Go

package native
import (
"math/big"
"git.marketally.com/tutus-one/tutus-chain/pkg/core/dao"
"git.marketally.com/tutus-one/tutus-chain/pkg/core/storage"
"git.marketally.com/tutus-one/tutus-chain/pkg/util"
"git.marketally.com/tutus-one/tutus-chain/pkg/vm/stackitem"
)
var intOne = big.NewInt(1)
var intTwo = big.NewInt(2)
func getConvertibleFromDAO(id int32, d *dao.Simple, key []byte, conv stackitem.Convertible) error {
si := d.GetStorageItem(id, key)
if si == nil {
return storage.ErrKeyNotFound
}
return stackitem.DeserializeConvertible(si, conv)
}
func putConvertibleToDAO(id int32, d *dao.Simple, key []byte, conv stackitem.Convertible) error {
item, err := conv.ToStackItem()
if err != nil {
return err
}
data, err := d.GetItemCtx().Serialize(item, false)
if err != nil {
return err
}
d.PutStorageItem(id, key, data)
return nil
}
func setIntWithKey(id int32, dao *dao.Simple, key []byte, value int64) {
dao.PutBigInt(id, key, big.NewInt(value))
}
func getIntWithKey(id int32, dao *dao.Simple, key []byte) int64 {
res, err := dao.GetInt(id, key)
if err != nil {
panic(err)
}
return res
}
// makeUint160Key creates a key from the account script hash.
func makeUint160Key(prefix byte, h util.Uint160) []byte {
k := make([]byte, util.Uint160Size+1)
k[0] = prefix
copy(k[1:], h.BytesBE())
return k
}
func toString(item stackitem.Item) string {
s, err := stackitem.ToString(item)
if err != nil {
panic(err)
}
return s
}
func toBool(item stackitem.Item) bool {
b, err := item.TryBool()
if err != nil {
panic(err)
}
return b
}