111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
// 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()
|
|
}
|