fix cache

getevo · Nov 29, 2023 · ef44274 · ef44274
1 parent c870162
commit ef44274
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 0 deletions.
diff --git a/lib/cache/drivers/memory/memory.go b/lib/cache/drivers/memory/memory.go
@@ -15,6 +15,10 @@ var unmarshaller func(bytes []byte, out interface{}) error = binary.Unmarshal
 
 type driver struct{}
 
+func (d driver) SetPrefix(p string) {
+	return
+}
+
 var items sync.Map
 
 func (driver) Register() error {

diff --git a/lib/cache/drivers/redis/redis.go b/lib/cache/drivers/redis/redis.go
@@ -242,3 +242,7 @@ func toInt64(n interface{}) int64 {
 	}
 	return 0
 }
+
+func (d driver) SetPrefix(p string) {
+	prefix = p
+}
diff --git a/lib/cache/functions.go b/lib/cache/functions.go
@@ -133,3 +133,7 @@ func Register() error {
 func DefaultDriver() Interface {
 	return defaultDriver
 }
+
+func SetPrefix(prefix string) {
+	defaultDriver.SetPrefix(prefix)
+}
diff --git a/lib/cache/interface.go b/lib/cache/interface.go
@@ -71,4 +71,7 @@ type Interface interface {
 
 	// SetUnMarshaller set []byte to interface{} unmarshalling function
 	SetUnMarshaller(func(bytes []byte, out interface{}) error)
+
+	// SetPrefix set a key prefix
+	SetPrefix(p string)
 }