add context to keyvaluemodule get/set

This commit is contained in:
Joel Wetzell
2026-05-27 18:30:23 -05:00
parent 3adbae8068
commit 52ca801fbd
5 changed files with 10 additions and 10 deletions
+1 -1
View File
@@ -77,7 +77,7 @@ func (kvg *KVGet) Process(ctx context.Context, wrappedPayload common.WrappedPayl
kvg.module = kvModule
}
value, err := kvg.module.Get(kvg.Key)
value, err := kvg.module.Get(ctx, kvg.Key)
if err != nil {
wrappedPayload.End = true
return wrappedPayload, fmt.Errorf("kv.get error getting key: %w", err)
+1 -1
View File
@@ -78,7 +78,7 @@ func (kvs *KVSet) Process(ctx context.Context, wrappedPayload common.WrappedPayl
kvs.module = kvModule
}
err := kvs.module.Set(kvs.Key, wrappedPayload.Payload)
err := kvs.module.Set(ctx, kvs.Key, wrappedPayload.Payload)
if err != nil {
wrappedPayload.End = true
return wrappedPayload, fmt.Errorf("kv.set error setting key: %w", err)