forked from TrueCloudLab/neoneo-go
4e6b1c4a38
Add method that returns iterator. It's needed for the workshop. Signed-off-by: Anna Shaleva <shaleva.ann@nspcc.ru>
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package storagecontract
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/iterator"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/storage"
|
|
)
|
|
|
|
// ctx holds storage context for contract methods
|
|
var ctx storage.Context
|
|
|
|
// defaultKey represents the default key.
|
|
var defaultKey = []byte("default")
|
|
|
|
// init inits storage context before any other contract method is called
|
|
func init() {
|
|
ctx = storage.GetContext()
|
|
}
|
|
|
|
// Put puts the value at the key.
|
|
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 {
|
|
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.
|
|
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.
|
|
func Find(value []byte) []string {
|
|
iter := storage.Find(ctx, value, storage.None)
|
|
result := []string{}
|
|
for iterator.Next(iter) {
|
|
val := iterator.Value(iter).([]string)
|
|
result = append(result, val[0]+":"+val[1])
|
|
}
|
|
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
|
|
}
|