实现缓存策略部分管理功能

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

@@ -0,0 +1,15 @@
+package cache
+
+import "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+
+type CleanAction struct {
+	actionutils.ParentAction
+}
+
+func (this *CleanAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *CleanAction) RunGet(params struct{}) {
+	this.Show()
+}

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

@@ -0,0 +1,82 @@
+package cache
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/actions"
+)
+
+type CreatePopupAction struct {
+	actionutils.ParentAction
+}
+
+func (this *CreatePopupAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *CreatePopupAction) RunGet(params struct{}) {
+	this.Data["types"] = serverconfigs.AllCachePolicyTypes
+	this.Show()
+}
+
+func (this *CreatePopupAction) RunPost(params struct {
+	Name string
+	Type string
+
+	// file
+	FileDir string
+
+	CapacityJSON []byte
+	MaxSizeJSON  []byte
+	MaxKeys      int64
+
+	Description string
+	IsOn        bool
+
+	Must *actions.Must
+}) {
+	params.Must.
+		Field("name", params.Name).
+		Require("请输入策略名称")
+
+	// 校验选项
+	var options interface{}
+	switch params.Type {
+	case serverconfigs.CachePolicyTypeFile:
+		params.Must.
+			Field("fileDir", params.FileDir).
+			Require("请输入缓存目录")
+		options = &serverconfigs.HTTPFileCacheConfig{
+			Dir: params.FileDir,
+		}
+	case serverconfigs.CachePolicyTypeMemory:
+		options = &serverconfigs.HTTPMemoryCacheConfig{
+		}
+	default:
+		this.Fail("请选择正确的缓存类型")
+	}
+
+	optionsJSON, err := json.Marshal(options)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	_, err = this.RPC().HTTPCachePolicyRPC().CreateHTTPCachePolicy(this.AdminContext(), &pb.CreateHTTPCachePolicyRequest{
+		IsOn:         params.IsOn,
+		Name:         params.Name,
+		Description:  params.Description,
+		CapacityJSON: params.CapacityJSON,
+		MaxKeys:      params.MaxKeys,
+		MaxSizeJSON:  params.MaxSizeJSON,
+		Type:         params.Type,
+		OptionsJSON:  optionsJSON,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}

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

@@ -0,0 +1,32 @@
+package cache
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+type DeleteAction struct {
+	actionutils.ParentAction
+}
+
+func (this *DeleteAction) RunPost(params struct {
+	CachePolicyId int64
+}) {
+	// 检查是否被引用
+	countResp, err := this.RPC().ServerRPC().CountAllEnabledServersWithCachePolicyId(this.AdminContext(), &pb.CountAllEnabledServersWithCachePolicyIdRequest{CachePolicyId: params.CachePolicyId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	if countResp.Count > 0 {
+		this.Fail("此缓存策略正在被别的服务引用，请修改后再删除。")
+	}
+
+	_, err = this.RPC().HTTPCachePolicyRPC().DeleteHTTPCachePolicy(this.AdminContext(), &pb.DeleteHTTPCachePolicyRequest{CachePolicyId: params.CachePolicyId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}

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

@@ -1,7 +1,11 @@
 package cache
 
 import (
+	"encoding/json"
 	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/maps"
 )
 
 type IndexAction struct {
@@ -13,6 +17,47 @@ func (this *IndexAction) Init() {
 }
 
 func (this *IndexAction) RunGet(params struct{}) {
+	countResp, err := this.RPC().HTTPCachePolicyRPC().CountAllEnabledHTTPCachePolicies(this.AdminContext(), &pb.CountAllEnabledHTTPCachePoliciesRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	count := countResp.Count
+	page := this.NewPage(count)
+	this.Data["page"] = page.AsHTML()
+
+	listResp, err := this.RPC().HTTPCachePolicyRPC().ListEnabledHTTPCachePolicies(this.AdminContext(), &pb.ListEnabledHTTPCachePoliciesRequest{
+		Offset: page.Offset,
+		Size:   page.Size,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	cachePoliciesJSON := listResp.CachePoliciesJSON
+	cachePolicies := []*serverconfigs.HTTPCachePolicy{}
+	err = json.Unmarshal(cachePoliciesJSON, &cachePolicies)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["cachePolicies"] = cachePolicies
+
+	infos := []maps.Map{}
+	for _, cachePolicy := range cachePolicies {
+		countServersResp, err := this.RPC().ServerRPC().CountAllEnabledServersWithCachePolicyId(this.AdminContext(), &pb.CountAllEnabledServersWithCachePolicyIdRequest{CachePolicyId: cachePolicy.Id})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		countServers := countServersResp.Count
+
+		infos = append(infos, maps.Map{
+			"typeName": serverconfigs.FindCachePolicyTypeName(cachePolicy.Type),
+			"countServers": countServers,
+		})
+	}
+	this.Data["infos"] = infos
 
 	this.Show()
 }

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

@@ -14,6 +14,15 @@ func init() {
 			Helper(componentutils.NewComponentHelper()).
 			Prefix("/servers/components/cache").
 			Get("", new(IndexAction)).
+			GetPost("/createPopup", new(CreatePopupAction)).
+			Get("/policy", new(PolicyAction)).
+			GetPost("/updatePopup", new(UpdatePopupAction)).
+			GetPost("/clean", new(CleanAction)).
+			GetPost("/preheat", new(PreheatAction)).
+			GetPost("/purge", new(PurgeAction)).
+			GetPost("/stat", new(StatAction)).
+			GetPost("/test", new(TestAction)).
+			Post("/delete", new(DeleteAction)).
 			EndAll()
 	})
 }

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

@@ -0,0 +1,15 @@
+package cache
+
+import "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+
+type PolicyAction struct {
+	actionutils.ParentAction
+}
+
+func (this *PolicyAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *PolicyAction) RunGet(params struct{}) {
+	this.Show()
+}

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

@@ -0,0 +1,15 @@
+package cache
+
+import "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+
+type PreheatAction struct {
+	actionutils.ParentAction
+}
+
+func (this *PreheatAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *PreheatAction) RunGet(params struct{}) {
+	this.Show()
+}

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

@@ -0,0 +1,15 @@
+package cache
+
+import "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+
+type PurgeAction struct {
+	actionutils.ParentAction
+}
+
+func (this *PurgeAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *PurgeAction) RunGet(params struct{}) {
+	this.Show()
+}

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

@@ -0,0 +1,15 @@
+package cache
+
+import "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+
+type StatAction struct {
+	actionutils.ParentAction
+}
+
+func (this *StatAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *StatAction) RunGet(params struct{}) {
+	this.Show()
+}

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

@@ -0,0 +1,15 @@
+package cache
+
+import "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+
+type TestAction struct {
+	actionutils.ParentAction
+}
+
+func (this *TestAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *TestAction) RunGet(params struct{}) {
+	this.Show()
+}

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

@@ -0,0 +1,108 @@
+package cache
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/actions"
+)
+
+type UpdatePopupAction struct {
+	actionutils.ParentAction
+}
+
+func (this *UpdatePopupAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *UpdatePopupAction) RunGet(params struct {
+	CachePolicyId int64
+}) {
+	configResp, err := this.RPC().HTTPCachePolicyRPC().FindEnabledHTTPCachePolicyConfig(this.AdminContext(), &pb.FindEnabledHTTPCachePolicyConfigRequest{CachePolicyId: params.CachePolicyId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	configJSON := configResp.CachePolicyJSON
+	if len(configJSON) == 0 {
+		this.NotFound("cachePolicy", params.CachePolicyId)
+		return
+	}
+
+	cachePolicy := &serverconfigs.HTTPCachePolicy{}
+	err = json.Unmarshal(configJSON, cachePolicy)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["cachePolicy"] = cachePolicy
+
+	// 其他选项
+	this.Data["types"] = serverconfigs.AllCachePolicyTypes
+
+	this.Show()
+}
+
+func (this *UpdatePopupAction) RunPost(params struct {
+	CachePolicyId int64
+
+	Name string
+	Type string
+
+	// file
+	FileDir string
+
+	CapacityJSON []byte
+	MaxSizeJSON  []byte
+	MaxKeys      int64
+
+	Description string
+	IsOn        bool
+
+	Must *actions.Must
+}) {
+	params.Must.
+		Field("name", params.Name).
+		Require("请输入策略名称")
+
+	// 校验选项
+	var options interface{}
+	switch params.Type {
+	case serverconfigs.CachePolicyTypeFile:
+		params.Must.
+			Field("fileDir", params.FileDir).
+			Require("请输入缓存目录")
+		options = &serverconfigs.HTTPFileCacheConfig{
+			Dir: params.FileDir,
+		}
+	case serverconfigs.CachePolicyTypeMemory:
+		options = &serverconfigs.HTTPMemoryCacheConfig{
+		}
+	default:
+		this.Fail("请选择正确的缓存类型")
+	}
+
+	optionsJSON, err := json.Marshal(options)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	_, err = this.RPC().HTTPCachePolicyRPC().UpdateHTTPCachePolicy(this.AdminContext(), &pb.UpdateHTTPCachePolicyRequest{
+		CachePolicyId: params.CachePolicyId,
+		IsOn:          params.IsOn,
+		Name:          params.Name,
+		Description:   params.Description,
+		CapacityJSON:  params.CapacityJSON,
+		MaxKeys:       params.MaxKeys,
+		MaxSizeJSON:   params.MaxSizeJSON,
+		Type:          params.Type,
+		OptionsJSON:   optionsJSON,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}