neoneo-go/examples/storage/storage.go

63 lines
1.5 KiB
Go
Raw Normal View History

package storagecontract
import (
"github.com/nspcc-dev/neo-go/pkg/interop/iterator"
"github.com/nspcc-dev/neo-go/pkg/interop/storage"
)
2020-08-10 10:42:02 +00:00
// ctx holds storage context for contract methods
var ctx storage.Context
// defaultKey represents the default key.
var defaultKey = []byte("default")
2020-08-10 10:42:02 +00:00
// init inits storage context before any other contract method is called
func init() {
ctx = storage.GetContext()
}
// Put puts the value at the key.
2020-08-10 10:42:02 +00:00
func Put(key, value []byte) []byte {
storage.Put(ctx, key, value)
return key
}
// PutDefault puts the value to the default key.
func PutDefault(value []byte) []byte {
storage.Put(ctx, defaultKey, value)
return defaultKey
}
// Get returns the value at the passed key.
func Get(key []byte) any {
2020-08-10 10:42:02 +00:00
return storage.Get(ctx, key)
}
// GetDefault returns the value at the default key.
func GetDefault() any {
return storage.Get(ctx, defaultKey)
}
// Delete deletes the value at the passed key.
2020-08-10 10:42:02 +00:00
func Delete(key []byte) bool {
storage.Delete(ctx, key)
return true
}
// Find returns an array of key-value pairs with the key that matches the passed value.
2020-08-10 10:42:02 +00:00
func Find(value []byte) []string {
iter := storage.Find(ctx, value, storage.None)
2020-08-10 10:42:02 +00:00
result := []string{}
for iterator.Next(iter) {
val := iterator.Value(iter).([]string)
result = append(result, val[0]+":"+val[1])
}
2020-08-10 10:42:02 +00:00
return result
}
// FindReturnIter returns an iterator over key-value pairs with the key that has the specified prefix.
func FindReturnIter(prefix []byte) iterator.Iterator {
iter := storage.Find(ctx, prefix, storage.None)
return iter
}