[cache]清理所有缓存时，会将此动作同步到集群

Author: iwind

teacluster/action_run.go

@@ -23,6 +23,8 @@ func (this *RunAction) Execute() error {
 	switch this.Cmd {
 	case "cache.refresh":
 		this.runCacheRefresh()
+	case "cache.clean":
+		this.runCacheClean()
 	}
 	return nil
 }
@@ -43,6 +45,9 @@ func (this *RunAction) runCacheRefresh() {
 	manager := teacache.FindCachePolicyManager(filename)
 	if manager == nil {
 		manager = teacache.NewManagerFromConfig(policy)
+		defer func() {
+			_ = manager.Close()
+		}()
 	}
 	prefixes := this.Data.GetSlice("prefixes")
 	prefixStrings := []string{}
@@ -51,7 +56,30 @@ func (this *RunAction) runCacheRefresh() {
 	}
 	_, err := manager.DeletePrefixes(prefixStrings)
 	if err != nil {
-		logs.Println("[cluster][cache.refresh]delete prefixes")
+		logs.Println("[cluster][cache.refresh]delete prefixes: " + err.Error())
+		return
+	}
+}
+
+// clean cache
+func (this *RunAction) runCacheClean() {
+	filename := this.Data.GetString("filename")
+	policy := shared.NewCachePolicyFromFile(filename)
+	if policy == nil {
+		logs.Println("[cluster][cache.clean]can not find policy with '" + filename + "'")
+		return
+	}
+
+	manager := teacache.FindCachePolicyManager(filename)
+	if manager == nil {
+		manager = teacache.NewManagerFromConfig(policy)
+		defer func() {
+			_ = manager.Close()
+		}()
+	}
+	err := manager.Clean()
+	if err != nil {
+		logs.Println("[cluster][cache.clean]clean: " + err.Error())
 		return
 	}
 }

teaweb/actions/default/cache/cleanPolicy.go

@@ -2,8 +2,10 @@ package cache
 
 import (
 	"github.com/TeaWeb/code/teacache"
+	"github.com/TeaWeb/code/teacluster"
 	"github.com/TeaWeb/code/teaconfigs/shared"
 	"github.com/TeaWeb/code/teaweb/actions/default/actionutils"
+	"github.com/iwind/TeaGo/maps"
 )
 
 type CleanPolicyAction struct {
@@ -53,6 +55,17 @@ func (this *CleanPolicyAction) RunPost(params struct {
 		this.Fail()
 	}
 
+	// 同步到集群
+	action := new(teacluster.RunAction)
+	action.Cmd = "cache.clean"
+	action.Data = maps.Map{
+		"filename": params.Filename,
+	}
+	err = teacluster.SharedManager.Write(action)
+	if err != nil {
+		this.Fail("同步到集群失败：" + err.Error())
+	}
+
 	this.Data["result"] = "清理完成"
 	this.Success()
 }

teaweb/actions/default/cache/refreshPolicy.go

@@ -75,7 +75,7 @@ func (this *RefreshPolicyAction) RunPost(params struct {
 
 	this.Data["count"] = count
 
-	// 清除节点
+	// 同步到集群
 	action := new(teacluster.RunAction)
 	action.Cmd = "cache.refresh"
 	action.Data = maps.Map{
@@ -84,7 +84,7 @@ func (this *RefreshPolicyAction) RunPost(params struct {
 	}
 	err = teacluster.SharedManager.Write(action)
 	if err != nil {
-		this.Fail("刷新集群失败：" + err.Error())
+		this.Fail("同步到集群失败：" + err.Error())
 	}
 
 	this.Success()