Added ResetCache function to reset the all in-memory cache.

turbot · Dec 22, 2024 · 5ae8a13 · 5ae8a13
1 parent a3efb34
commit 5ae8a13
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,12 @@
 
 Shared Pipes Component
 
+## v1.7.1 [tbd]
+
+_What's new_
+
+* Added `ResetCache` function to reset the all in-memory cache.
+
 ## v1.7.0 [2024-12-20]
 
 _What's new_

diff --git a/cache/index.go b/cache/index.go
@@ -23,7 +23,7 @@ var (
 	connectionCacheInitOnce sync.Once
 )
 
-func inMemoryInitialize(config *ristretto.Config) *InMemoryCache {
+func initializeInMemoryCache(config *ristretto.Config) *InMemoryCache {
 	if config == nil {
 		config = &ristretto.Config{
 			NumCounters: 100000,   // number of keys to track frequency
@@ -43,7 +43,7 @@ func inMemoryInitialize(config *ristretto.Config) *InMemoryCache {
 
 func GetCache() *InMemoryCache {
 	inMemoryCacheInitOnce.Do(func() {
-		inMemoryInitialize(nil)
+		initializeInMemoryCache(nil)
 	})
 	return inMemoryCache
 }
@@ -104,6 +104,17 @@ func ResetConnectionCache() {
 	initializeConnectionCache()
 }
 
+func ResetInMemoryCache() {
+	inMemoryCache = nil
+	initializeInMemoryCache(nil)
+}
+
+func ResetAllCache() {
+	ResetCredentialCache()
+	ResetConnectionCache()
+	ResetInMemoryCache()
+}
+
 func (cache *InMemoryCache) SetWithTTL(key string, value interface{}, ttl time.Duration) bool {
 	res := cache.cache.SetWithTTL(key, value, 1, ttl)