TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-11-27 03:58:06 +00:00
Code Issues Projects Releases Packages Wiki Activity

native: unify some transfer code

Browse source
This commit is contained in:
Roman Khimov 2020-06-23 22:20:17 +03:00
parent fccad11716
commit 48fac6f87d
1 changed files with 24 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
pkg/core/native/native_nep5.go
View file

@ -148,6 +148,28 @@ func (c *nep5TokenNative) emitTransfer(ic *interop.Context, from, to *util.Uint1
ic.Notifications = append(ic.Notifications, ne) ic.Notifications = append(ic.Notifications, ne)
} }
func (c *nep5TokenNative) updateAccBalance(ic *interop.Context, acc util.Uint160, amount *big.Int) error {
key := makeAccountKey(acc)
si := ic.DAO.GetStorageItem(c.ContractID, key)
if si == nil {
if amount.Sign() <= 0 {
return errors.New("insufficient funds")
}
si = new(state.StorageItem)
}
err := c.incBalance(ic, acc, si, amount)
if err != nil {
return err
}
if si.Value == nil {
err = ic.DAO.DeleteStorageItem(c.ContractID, key)
} else {
err = ic.DAO.PutStorageItem(c.ContractID, key, si)
}
return err
}
func (c *nep5TokenNative) transfer(ic *interop.Context, from, to util.Uint160, amount *big.Int) error { func (c *nep5TokenNative) transfer(ic *interop.Context, from, to util.Uint160, amount *big.Int) error {
if amount.Sign() == -1 { if amount.Sign() == -1 {
return errors.New("negative amount") return errors.New("negative amount")
@ -164,12 +186,6 @@ func (c *nep5TokenNative) transfer(ic *interop.Context, from, to util.Uint160, a
return errors.New("invalid signature") return errors.New("invalid signature")
} }
keyFrom := makeAccountKey(from)
siFrom := ic.DAO.GetStorageItem(c.ContractID, keyFrom)
if siFrom == nil {
return errors.New("insufficient funds")
}
isEmpty := from.Equals(to) || amount.Sign() == 0 isEmpty := from.Equals(to) || amount.Sign() == 0
inc := amount inc := amount
if isEmpty { if isEmpty {
@ -177,33 +193,12 @@ func (c *nep5TokenNative) transfer(ic *interop.Context, from, to util.Uint160, a
} else { } else {
inc = new(big.Int).Neg(inc) inc = new(big.Int).Neg(inc)
} }
if err := c.incBalance(ic, from, siFrom, inc); err != nil { if err := c.updateAccBalance(ic, from, inc); err != nil {
return err
}
if siFrom.Value == nil {
err = ic.DAO.DeleteStorageItem(c.ContractID, keyFrom)
} else {
err = ic.DAO.PutStorageItem(c.ContractID, keyFrom, siFrom)
}
if err != nil {
return err return err
} }
if !isEmpty { if !isEmpty {
keyTo := makeAccountKey(to) if err := c.updateAccBalance(ic, to, amount); err != nil {
siTo := ic.DAO.GetStorageItem(c.ContractID, keyTo)
if siTo == nil {
siTo = new(state.StorageItem)
}
if err := c.incBalance(ic, to, siTo, amount); err != nil {
return err
}
if siTo.Value == nil {
err = ic.DAO.DeleteStorageItem(c.ContractID, keyTo)
} else {
err = ic.DAO.PutStorageItem(c.ContractID, keyTo, siTo)
}
if err != nil {
return err return err
} }
} }