wip with store stuff

streamingfast · Dec 20, 2023 · e32f58f · e32f58f
1 parent b1390d5
commit e32f58f
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 5 deletions.
diff --git a/wasm/bench/substreams_tiny_go/main.go b/wasm/bench/substreams_tiny_go/main.go
@@ -54,6 +54,8 @@ type blockStat struct {
 }
 
 func mapBlock(block *pb.Block, readStore *ReadStore) error {
+	//readStore.GetAt(0, []byte("key_123"))
+
 	rocketAddress := strings.ToLower("ae78736Cd615f374D3085123A210448E74Fc6393")
 
 	approvalTopic := strings.ToLower("8c5be1e5ebec7d5bd14f71427d1e84f3dd0314c0f7b2291e5b200ac8c7c3b925")
@@ -121,7 +123,7 @@ func Println(s string) {
 	println(unsafe.Pointer(&d[0]), uint32(len(d)))
 }
 
-//go:wasmimport store get_at
+//go:wasmimport state get_at
 //go:noescape
 func get_at(storeIndex uint32, ordinal uint64, keyPtr unsafe.Pointer, keyLen uint32, outputPtr unsafe.Pointer) int32
 
@@ -151,7 +153,7 @@ func NewWriteStore(storeIdx uint32) *WriteStore {
 	return &WriteStore{}
 }
 
-//go:wasmimport store set
+//go:wasmimport state set
 //go:noescape
 func set(ordinal uint64, keyPtr unsafe.Pointer, keyLen uint32, valuePtr unsafe.Pointer, valueLen uint32)
 

diff --git a/wasm/bench/substreams_tiny_go/main.wasm b/wasm/bench/substreams_tiny_go/main.wasm
diff --git a/wasm/wasi/module.go b/wasm/wasi/module.go
@@ -44,7 +44,11 @@ func newModule(ctx context.Context, wasmCode []byte, wasmCodeType string, regist
 	if err != nil {
 		return nil, err
 	}
-	hostModules = append(hostModules, loggerModule)
+	stateModule, err := sfwaz.AddHostFunctions(ctx, runtime, "state", sfwaz.StateFuncs)
+	if err != nil {
+		return nil, err
+	}
+	hostModules = append(hostModules, loggerModule, stateModule)
 
 	//startFunc := "main"
 	//switch wasmCodeType {

diff --git a/wasm/wazero/module.go b/wasm/wazero/module.go
@@ -52,7 +52,7 @@ func newModule(ctx context.Context, wasmCode []byte, wasmCodeType string, regist
 	if err != nil {
 		return nil, err
 	}
-	stateModule, err := AddHostFunctions(ctx, runtime, "state", stateFuncs)
+	stateModule, err := AddHostFunctions(ctx, runtime, "state", StateFuncs)
 	if err != nil {
 		return nil, err
 	}

diff --git a/wasm/wazero/state_hostmod.go b/wasm/wazero/state_hostmod.go
@@ -23,7 +23,7 @@ type funcs struct {
 	f      api.GoModuleFunction
 }
 
-var stateFuncs = []funcs{
+var StateFuncs = []funcs{
 	{
 		"set",
 		[]parm{i64, i32, i32, i32, i32},