实现缓存策略若干功能

internal/web/actions/default/servers/components/cache/stat.go

@@ -1,6 +1,16 @@
 package cache
 
-import "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/nodes/nodeutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/messageconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/actions"
+	"github.com/iwind/TeaGo/maps"
+	"github.com/iwind/TeaGo/types"
+	"net/http"
+	"strconv"
+)
 
 type StatAction struct {
 	actionutils.ParentAction
@@ -11,5 +21,72 @@ func (this *StatAction) Init() {
 }
 
 func (this *StatAction) RunGet(params struct{}) {
+	// 默认的集群ID
+	cookie, err := this.Request.Cookie("cache_cluster_id")
+	if cookie != nil && err == nil {
+		this.Data["clusterId"] = types.Int64(cookie.Value)
+	}
+
+	// 集群列表
+	clustersResp, err := this.RPC().NodeClusterRPC().FindAllEnabledNodeClusters(this.AdminContext(), &pb.FindAllEnabledNodeClustersRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	clusterMaps := []maps.Map{}
+	for _, cluster := range clustersResp.Clusters {
+		clusterMaps = append(clusterMaps, maps.Map{
+			"id":   cluster.Id,
+			"name": cluster.Name,
+		})
+	}
+	this.Data["clusters"] = clusterMaps
+
 	this.Show()
 }
+
+func (this *StatAction) RunPost(params struct {
+	CachePolicyId int64
+	ClusterId     int64
+
+	Must *actions.Must
+}) {
+	// 记录clusterId
+	this.AddCookie(&http.Cookie{
+		Name:  "cache_cluster_id",
+		Value: strconv.FormatInt(params.ClusterId, 10),
+	})
+
+	cachePolicyResp, err := this.RPC().HTTPCachePolicyRPC().FindEnabledHTTPCachePolicyConfig(this.AdminContext(), &pb.FindEnabledHTTPCachePolicyConfigRequest{CachePolicyId: params.CachePolicyId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	cachePolicyJSON := cachePolicyResp.CachePolicyJSON
+	if len(cachePolicyJSON) == 0 {
+		this.Fail("找不到要操作的缓存策略")
+	}
+
+	// 发送命令
+	msg := &messageconfigs.StatCacheMessage{
+		CachePolicyJSON: cachePolicyJSON,
+	}
+	results, err := nodeutils.SendMessageToCluster(this.AdminContext(), params.ClusterId, messageconfigs.MessageCodeStatCache, msg, 10)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	isAllOk := true
+	for _, result := range results {
+		if !result.IsOK {
+			isAllOk = false
+			break
+		}
+	}
+
+	this.Data["isAllOk"] = isAllOk
+	this.Data["results"] = results
+
+	this.Success()
+}