实现缓存策略的部分功能

internal/web/actions/default/servers/components/cache/cacheutils/utils.go

@@ -0,0 +1,38 @@
+package cacheutils
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/errors"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+)
+
+// 查找缓存策略名称并忽略错误
+func FindCachePolicyNameWithoutError(parent *actionutils.ParentAction, cachePolicyId int64) string {
+	policy, err := FindCachePolicy(parent, cachePolicyId)
+	if err != nil {
+		return ""
+	}
+	if policy == nil {
+		return ""
+	}
+	return policy.Name
+}
+
+// 查找缓存策略配置
+func FindCachePolicy(parent *actionutils.ParentAction, cachePolicyId int64) (*serverconfigs.HTTPCachePolicy, error) {
+	resp, err := parent.RPC().HTTPCachePolicyRPC().FindEnabledHTTPCachePolicyConfig(parent.AdminContext(), &pb.FindEnabledHTTPCachePolicyConfigRequest{CachePolicyId: cachePolicyId})
+	if err != nil {
+		return nil, err
+	}
+	if len(resp.CachePolicyJSON) == 0 {
+		return nil, errors.New("cache policy not found")
+	}
+	config := &serverconfigs.HTTPCachePolicy{}
+	err = json.Unmarshal(resp.CachePolicyJSON, config)
+	if err != nil {
+		return nil, err
+	}
+	return config, nil
+}

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

@@ -7,7 +7,7 @@ type CleanAction struct {
 }
 
 func (this *CleanAction) Init() {
-	this.Nav("", "", "")
+	this.Nav("", "", "clean")
 }
 
 func (this *CleanAction) RunGet(params struct{}) {

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

@@ -1,8 +1,11 @@
 package cache
 
 import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/servers/components/cache/cacheutils"
 	"github.com/iwind/TeaGo/actions"
 	"net/http"
+	"reflect"
 )
 
 type Helper struct {
@@ -12,11 +15,23 @@ func NewHelper() *Helper {
 	return &Helper{}
 }
 
-func (this *Helper) BeforeAction(action *actions.ActionObject) {
+func (this *Helper) BeforeAction(actionPtr actions.ActionWrapper) {
+	action := actionPtr.Object()
 	if action.Request.Method != http.MethodGet {
 		return
 	}
 
 	action.Data["mainTab"] = "component"
 	action.Data["secondMenuItem"] = "cache"
+
+	cachePolicyId := action.ParamInt64("cachePolicyId")
+	action.Data["cachePolicyId"] = cachePolicyId
+
+	parentActionValue := reflect.ValueOf(actionPtr).Elem().FieldByName("ParentAction")
+	if parentActionValue.IsValid() {
+		parentAction, isOk := parentActionValue.Interface().(actionutils.ParentAction)
+		if isOk {
+			action.Data["cachePolicyName"] = cacheutils.FindCachePolicyNameWithoutError(&parentAction, cachePolicyId)
+		}
+	}
 }

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

@@ -16,13 +16,15 @@ func init() {
 			Get("", new(IndexAction)).
 			GetPost("/createPopup", new(CreatePopupAction)).
 			Get("/policy", new(PolicyAction)).
-			GetPost("/updatePopup", new(UpdatePopupAction)).
+			GetPost("/update", new(UpdateAction)).
 			GetPost("/clean", new(CleanAction)).
 			GetPost("/preheat", new(PreheatAction)).
 			GetPost("/purge", new(PurgeAction)).
 			GetPost("/stat", new(StatAction)).
 			GetPost("/test", new(TestAction)).
 			Post("/delete", new(DeleteAction)).
+			Post("/testRead", new(TestReadAction)).
+			Post("/testWrite", new(TestWriteAction)).
 			EndAll()
 	})
 }

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

@@ -1,15 +1,30 @@
 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/servers/components/cache/cacheutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+)
 
 type PolicyAction struct {
 	actionutils.ParentAction
 }
 
 func (this *PolicyAction) Init() {
-	this.Nav("", "", "")
+	this.Nav("", "", "index")
 }
 
-func (this *PolicyAction) RunGet(params struct{}) {
+func (this *PolicyAction) RunGet(params struct {
+	CachePolicyId int64
+}) {
+	cachePolicy, err := cacheutils.FindCachePolicy(this.Parent(), params.CachePolicyId)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["cachePolicy"] = cachePolicy
+
+	this.Data["typeName"] = serverconfigs.FindCachePolicyTypeName(cachePolicy.Type)
+
 	this.Show()
 }

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

@@ -7,7 +7,7 @@ type PreheatAction struct {
 }
 
 func (this *PreheatAction) Init() {
-	this.Nav("", "", "")
+	this.Nav("", "", "preheat")
 }
 
 func (this *PreheatAction) RunGet(params struct{}) {

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

@@ -7,7 +7,7 @@ type PurgeAction struct {
 }
 
 func (this *PurgeAction) Init() {
-	this.Nav("", "", "")
+	this.Nav("", "", "purge")
 }
 
 func (this *PurgeAction) RunGet(params struct{}) {

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

@@ -7,7 +7,7 @@ type StatAction struct {
 }
 
 func (this *StatAction) Init() {
-	this.Nav("", "", "")
+	this.Nav("", "", "stat")
 }
 
 func (this *StatAction) RunGet(params struct{}) {

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

@@ -1,15 +1,34 @@
 package cache
 
-import "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/maps"
+)
 
 type TestAction struct {
 	actionutils.ParentAction
 }
 
 func (this *TestAction) Init() {
-	this.Nav("", "", "")
+	this.Nav("", "", "test")
 }
 
 func (this *TestAction) RunGet(params struct{}) {
+	// 集群列表
+	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()
 }

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

@@ -0,0 +1,52 @@
+package cache
+
+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"
+)
+
+type TestReadAction struct {
+	actionutils.ParentAction
+}
+
+func (this *TestReadAction) RunPost(params struct {
+	ClusterId     int64
+	CachePolicyId int64
+	Key           string
+}) {
+	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.ReadCacheMessage{
+		CachePolicyJSON: cachePolicyJSON,
+		Key:             params.Key,
+	}
+	results, err := nodeutils.SendMessageToCluster(this.AdminContext(), params.ClusterId, messageconfigs.MessageCodeReadCache, 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()
+}

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

@@ -0,0 +1,55 @@
+package cache
+
+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"
+)
+
+type TestWriteAction struct {
+	actionutils.ParentAction
+}
+
+func (this *TestWriteAction) RunPost(params struct {
+	ClusterId     int64
+	CachePolicyId int64
+	Key           string
+	Value         string
+}) {
+	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.WriteCacheMessage{
+		CachePolicyJSON: cachePolicyJSON,
+		Key:             params.Key,
+		Value:           []byte(params.Value),
+		LifeSeconds:     3600,
+	}
+	results, err := nodeutils.SendMessageToCluster(this.AdminContext(), params.ClusterId, messageconfigs.MessageCodeWriteCache, 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()
+}

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

@@ -8,15 +8,15 @@ import (
 	"github.com/iwind/TeaGo/actions"
 )
 
-type UpdatePopupAction struct {
+type UpdateAction struct {
 	actionutils.ParentAction
 }
 
-func (this *UpdatePopupAction) Init() {
-	this.Nav("", "", "")
+func (this *UpdateAction) Init() {
+	this.Nav("", "", "update")
 }
 
-func (this *UpdatePopupAction) RunGet(params struct {
+func (this *UpdateAction) RunGet(params struct {
 	CachePolicyId int64
 }) {
 	configResp, err := this.RPC().HTTPCachePolicyRPC().FindEnabledHTTPCachePolicyConfig(this.AdminContext(), &pb.FindEnabledHTTPCachePolicyConfigRequest{CachePolicyId: params.CachePolicyId})
@@ -44,7 +44,7 @@ func (this *UpdatePopupAction) RunGet(params struct {
 	this.Show()
 }
 
-func (this *UpdatePopupAction) RunPost(params struct {
+func (this *UpdateAction) RunPost(params struct {
 	CachePolicyId int64
 
 	Name string