Initial commit (code only without large binaries)

EdgeAdmin/internal/web/actions/default/users/setting/server.go

@@ -0,0 +1,110 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build plus
+
+package setting
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/systemconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
+	"github.com/iwind/TeaGo/actions"
+	"github.com/iwind/TeaGo/maps"
+)
+
+type ServerAction struct {
+	actionutils.ParentAction
+}
+
+func (this *ServerAction) Init() {
+	this.Nav("", "", "server")
+}
+
+func (this *ServerAction) RunGet(params struct{}) {
+	// 当前配置
+	resp, err := this.RPC().SysSettingRPC().ReadSysSetting(this.AdminContext(), &pb.ReadSysSettingRequest{Code: systemconfigs.SettingCodeUserServerConfig})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	var config = userconfigs.DefaultUserServerConfig()
+	if len(resp.ValueJSON) > 0 {
+		err = json.Unmarshal(resp.ValueJSON, config)
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+	}
+
+	this.Data["config"] = config
+
+	// 分组
+	groupsResp, err := this.RPC().ServerGroupRPC().FindAllEnabledServerGroups(this.AdminContext(), &pb.FindAllEnabledServerGroupsRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var groupMaps = []maps.Map{}
+	for _, group := range groupsResp.ServerGroups {
+		groupMaps = append(groupMaps, maps.Map{
+			"id":   group.Id,
+			"name": group.Name,
+		})
+	}
+	this.Data["groups"] = groupMaps
+
+	// 缓存
+	this.Data["defaultMaxCacheKeysPerTask"] = userconfigs.MaxCacheKeysPerTask
+	this.Data["defaultMaxCacheKeysPerDay"] = userconfigs.MaxCacheKeysPerDay
+
+	this.Show()
+}
+
+func (this *ServerAction) RunPost(params struct {
+	GroupId     int64
+	RequirePlan bool
+	EnableStat  bool
+
+	// 缓存相关
+	PurgeMaxKeysPerTask int32
+	PurgeMaxKeysPerDay  int32
+
+	FetchMaxKeysPerTask int32
+	FetchMaxKeysPerDay  int32
+
+	Must *actions.Must
+	CSRF *actionutils.CSRF
+}) {
+	var config = userconfigs.DefaultUserServerConfig()
+	config.GroupId = params.GroupId
+	config.RequirePlan = params.RequirePlan
+	config.EnableStat = params.EnableStat
+
+	if config.HTTPCacheTaskPurgeConfig != nil {
+		config.HTTPCacheTaskPurgeConfig.MaxKeysPerTask = params.PurgeMaxKeysPerTask
+		config.HTTPCacheTaskPurgeConfig.MaxKeysPerDay = params.PurgeMaxKeysPerDay
+	}
+
+	if config.HTTPCacheTaskFetchConfig != nil {
+		config.HTTPCacheTaskFetchConfig.MaxKeysPerTask = params.FetchMaxKeysPerTask
+		config.HTTPCacheTaskFetchConfig.MaxKeysPerDay = params.FetchMaxKeysPerDay
+	}
+
+	configJSON, err := json.Marshal(config)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	_, err = this.RPC().SysSettingRPC().UpdateSysSetting(this.AdminContext(), &pb.UpdateSysSettingRequest{
+		Code:      systemconfigs.SettingCodeUserServerConfig,
+		ValueJSON: configJSON,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}