95 lines
2.6 KiB
Go
95 lines
2.6 KiB
Go
// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
|
|
//go:build plus
|
|
|
|
package trafficstats
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeAdmin/internal/configloaders"
|
|
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/systemconfigs"
|
|
"github.com/iwind/TeaGo/actions"
|
|
"github.com/iwind/TeaGo/maps"
|
|
)
|
|
|
|
type SettingAction struct {
|
|
actionutils.ParentAction
|
|
}
|
|
|
|
func (this *SettingAction) Init() {
|
|
this.Nav("", "", "setting")
|
|
}
|
|
|
|
func (this *SettingAction) RunGet(params struct{}) {
|
|
this.Data["config"] = maps.Map{
|
|
"bandwidthPercentile": systemconfigs.DefaultBandwidthPercentile,
|
|
"defaultBandwidthDateRange": systemconfigs.DefaultBandwidthDateRangeCode,
|
|
"bandwidthAlgo": systemconfigs.BandwidthAlgoSecondly,
|
|
}
|
|
userConfig, _ := configloaders.LoadUserUIConfig()
|
|
if userConfig != nil {
|
|
var config = userConfig.TrafficStats
|
|
if config.BandwidthPercentile <= 0 {
|
|
config.BandwidthPercentile = systemconfigs.DefaultBandwidthPercentile
|
|
}
|
|
if len(config.DefaultBandwidthDateRange) == 0 {
|
|
config.DefaultBandwidthDateRange = systemconfigs.DefaultBandwidthDateRangeCode
|
|
}
|
|
if len(config.BandwidthAlgo) == 0 {
|
|
config.BandwidthAlgo = systemconfigs.BandwidthAlgoSecondly
|
|
}
|
|
this.Data["config"] = config
|
|
}
|
|
|
|
this.Data["dateRanges"] = systemconfigs.FindAllBandwidthDateRanges()
|
|
|
|
this.Show()
|
|
}
|
|
|
|
func (this *SettingAction) RunPost(params struct {
|
|
BandwidthPercentile int32
|
|
DefaultBandwidthDateRange string
|
|
BandwidthAlgo string
|
|
|
|
Must *actions.Must
|
|
CSRF *actionutils.CSRF
|
|
}) {
|
|
config, err := configloaders.LoadUserUIConfig()
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
if params.BandwidthPercentile < 0 {
|
|
params.BandwidthPercentile = 0
|
|
}
|
|
if params.BandwidthPercentile > 100 {
|
|
params.BandwidthPercentile = 95
|
|
}
|
|
|
|
config.TrafficStats.BandwidthPercentile = params.BandwidthPercentile
|
|
config.TrafficStats.DefaultBandwidthDateRange = params.DefaultBandwidthDateRange
|
|
config.TrafficStats.BandwidthAlgo = params.BandwidthAlgo
|
|
|
|
err = configloaders.UpdateUserUIConfig(config)
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
// 财务配置
|
|
priceConfig, err := configloaders.LoadUserPriceConfig()
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
if priceConfig != nil && priceConfig.DefaultBandwidthPriceConfig != nil && priceConfig.DefaultBandwidthPriceConfig.BandwidthAlgo != params.BandwidthAlgo {
|
|
priceConfig.DefaultBandwidthPriceConfig.BandwidthAlgo = params.BandwidthAlgo
|
|
err = configloaders.UpdateUserPriceConfig(priceConfig)
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
}
|
|
|
|
this.Success()
|
|
}
|