fix: set a prefix to the kv keys function via WithPrefix

nitrictech · Jun 27, 2024 · e26953a · e26953a
1 parent c165e3d
commit e26953a
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/api/keyvalue/store.go b/api/keyvalue/store.go
@@ -24,7 +24,16 @@ import (
 	v1 "github.com/nitrictech/nitric/core/pkg/proto/kvstore/v1"
 )
 
-type ScanKeysOption = func(*v1.KvStoreScanKeysRequest)
+type ScanKeysRequest = v1.KvStoreScanKeysRequest
+
+type ScanKeysOption = func(*ScanKeysRequest)
+
+// Apply a prefix to the scan keys request
+func WithPrefix(prefix string) ScanKeysOption {
+	return func(req *ScanKeysRequest) {
+		req.Prefix = prefix
+	}
+}
 
 // TODO: maybe move keystream to separate file
 type KeyStream struct {