forked from TrueCloudLab/neoneo-go
a1e3655560
neo-storm has developed more wrappers for syscall APIs, so they can and should be used as a drop-in replacement for pkg/vm/api. Moving it out of vm, as it's not exactly related to the VM itself.
46 lines
811 B
Go
46 lines
811 B
Go
package storage_contract
|
|
|
|
import (
|
|
"github.com/CityOfZion/neo-go/pkg/interop/storage"
|
|
)
|
|
|
|
func Main(operation string, args []interface{}) interface{} {
|
|
ctx := storage.GetContext()
|
|
|
|
// Puts value at key
|
|
if operation == "put" {
|
|
if checkArgs(args, 2) {
|
|
key := args[0].([]byte)
|
|
value := args[1].([]byte)
|
|
storage.Put(ctx, key, value)
|
|
return key
|
|
}
|
|
}
|
|
|
|
// Returns the value at passed key
|
|
if operation == "get" {
|
|
if checkArgs(args, 1) {
|
|
key := args[0].([]byte)
|
|
return storage.Get(ctx, key)
|
|
}
|
|
}
|
|
|
|
// Deletes the value at passed key
|
|
if operation == "delete" {
|
|
key := args[0].([]byte)
|
|
storage.Delete(ctx, key)
|
|
return true
|
|
}
|
|
|
|
// TODO: storage.Find()
|
|
|
|
return false
|
|
}
|
|
|
|
func checkArgs(args []interface{}, length int) bool {
|
|
if len(args) == length {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|