chore: purge the previous cached

fufuok · Mar 23, 2024 · f2b50ac · f2b50ac
1 parent c659864
commit f2b50ac
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 2 deletions.
diff --git a/web/fiber/middleware/blacklist.go b/web/fiber/middleware/blacklist.go
@@ -20,8 +20,10 @@ var (
 )
 
 // UseBlacklistCache 重新设置黑名单检查时缓存, 配置变化时可选再次调用, 由应用端 Start() Runtime() 调用
-func UseBlacklistCache(capacity, lifetime uint32) (err error) {
+func UseBlacklistCache(capacity, lifetime uint32) error {
 	useBlacklistLRU.Store(false)
+	PurgeBlacklistCache()
+
 	if capacity == 0 {
 		return nil
 	}

diff --git a/web/fiber/middleware/whitelist.go b/web/fiber/middleware/whitelist.go
@@ -27,6 +27,8 @@ var (
 // UseWhitelistCache 重新设置白名单检查时缓存, 配置变化时可选再次调用, 由应用端 Start() Runtime() 调用
 func UseWhitelistCache(capacity, lifetime uint32) error {
 	useWhitelistLRU.Store(false)
+	PurgeWhitelistCache()
+
 	if capacity == 0 {
 		return nil
 	}

diff --git a/web/gin/middleware/blacklist.go b/web/gin/middleware/blacklist.go
@@ -19,8 +19,10 @@ var (
 )
 
 // UseBlacklistCache 重新设置黑名单检查时缓存, 配置变化时可选再次调用, 由应用端 Start() Runtime() 调用
-func UseBlacklistCache(capacity, lifetime uint32) (err error) {
+func UseBlacklistCache(capacity, lifetime uint32) error {
 	useBlacklistLRU.Store(false)
+	PurgeBlacklistCache()
+
 	if capacity == 0 {
 		return nil
 	}

diff --git a/web/gin/middleware/whitelist.go b/web/gin/middleware/whitelist.go
@@ -26,6 +26,8 @@ var (
 // UseWhitelistCache 重新设置白名单检查时缓存, 配置变化时可选再次调用, 由应用端 Start() Runtime() 调用
 func UseWhitelistCache(capacity, lifetime uint32) error {
 	useWhitelistLRU.Store(false)
+	PurgeWhitelistCache()
+
 	if capacity == 0 {
 		return nil
 	}