2021-10-21 17:09:59 +08:00
|
|
|
|
// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
|
|
|
|
|
|
|
|
|
|
|
|
package serverconfigs
|
|
|
|
|
|
|
|
|
|
|
|
import "github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
|
|
|
|
|
|
|
2021-11-09 17:36:34 +08:00
|
|
|
|
// DefaultTrafficLimitNoticePageBody 达到流量限制时默认提示内容
|
|
|
|
|
|
const DefaultTrafficLimitNoticePageBody = `<!DOCTYPE html>
|
2021-10-21 17:09:59 +08:00
|
|
|
|
<html>
|
|
|
|
|
|
<head>
|
2023-09-06 16:31:14 +08:00
|
|
|
|
<title>Traffic Limit Exceeded Warning</title>
|
2021-10-21 17:09:59 +08:00
|
|
|
|
<body>
|
|
|
|
|
|
|
2021-12-02 14:44:56 +08:00
|
|
|
|
<h1>Traffic Limit Exceeded Warning</h1>
|
|
|
|
|
|
<p>The site traffic has exceeded the limit. Please contact with the site administrator.</p>
|
|
|
|
|
|
<address>Request ID: ${requestId}.</address>
|
2021-10-21 17:09:59 +08:00
|
|
|
|
|
|
|
|
|
|
</body>
|
|
|
|
|
|
</html>`
|
|
|
|
|
|
|
2021-11-09 17:36:34 +08:00
|
|
|
|
// TrafficLimitConfig 流量限制
|
|
|
|
|
|
type TrafficLimitConfig struct {
|
2021-10-21 17:09:59 +08:00
|
|
|
|
IsOn bool `yaml:"isOn" json:"isOn"` // 是否启用
|
|
|
|
|
|
|
|
|
|
|
|
DailySize *shared.SizeCapacity `yaml:"dailySize" json:"dailySize"` // 每日限制
|
|
|
|
|
|
MonthlySize *shared.SizeCapacity `yaml:"monthlySize" json:"monthlySize"` // 每月限制
|
|
|
|
|
|
TotalSize *shared.SizeCapacity `yaml:"totalSize" json:"totalSize"` // 总限制 TODO 需要实现
|
|
|
|
|
|
|
|
|
|
|
|
NoticePageBody string `yaml:"noticePageBody" json:"noticePageBody"` // 超出限制时的提醒，支持请求变量
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// DailyBytes 每天限制
|
|
|
|
|
|
// 不使用Init()来初始化数据，是为了让其他地方不经过Init()也能得到计算值
|
2021-11-09 17:36:34 +08:00
|
|
|
|
func (this *TrafficLimitConfig) DailyBytes() int64 {
|
2021-10-21 17:09:59 +08:00
|
|
|
|
if this.DailySize != nil {
|
|
|
|
|
|
return this.DailySize.Bytes()
|
|
|
|
|
|
}
|
|
|
|
|
|
return -1
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// MonthlyBytes 每月限制
|
2021-11-09 17:36:34 +08:00
|
|
|
|
func (this *TrafficLimitConfig) MonthlyBytes() int64 {
|
2021-10-21 17:09:59 +08:00
|
|
|
|
if this.MonthlySize != nil {
|
|
|
|
|
|
return this.MonthlySize.Bytes()
|
|
|
|
|
|
}
|
|
|
|
|
|
return -1
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// TotalBytes 总限制
|
2021-11-09 17:36:34 +08:00
|
|
|
|
func (this *TrafficLimitConfig) TotalBytes() int64 {
|
2021-10-21 17:09:59 +08:00
|
|
|
|
if this.TotalSize != nil {
|
|
|
|
|
|
return this.TotalSize.Bytes()
|
|
|
|
|
|
}
|
|
|
|
|
|
return -1
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// IsEmpty 检查是否有限制值
|
2021-11-09 17:36:34 +08:00
|
|
|
|
func (this *TrafficLimitConfig) IsEmpty() bool {
|
2021-10-21 17:09:59 +08:00
|
|
|
|
return !this.IsOn || (this.DailyBytes() <= 0 && this.MonthlyBytes() <= 0 && this.TotalBytes() <= 0)
|
|
|
|
|
|
}
|
2023-09-06 16:31:14 +08:00
|
|
|
|
|
|
|
|
|
|
func (this *TrafficLimitConfig) Equals(another *TrafficLimitConfig) bool {
|
|
|
|
|
|
if another == nil {
|
|
|
|
|
|
return false
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if this.IsOn != another.IsOn {
|
|
|
|
|
|
return false
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if !this.equalCapacity(this.DailySize, another.DailySize) {
|
|
|
|
|
|
return false
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if !this.equalCapacity(this.MonthlySize, another.MonthlySize) {
|
|
|
|
|
|
return false
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if !this.equalCapacity(this.TotalSize, another.TotalSize) {
|
|
|
|
|
|
return false
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if this.NoticePageBody != another.NoticePageBody {
|
|
|
|
|
|
return false
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
return true
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func (this *TrafficLimitConfig) equalCapacity(size1 *shared.SizeCapacity, size2 *shared.SizeCapacity) bool {
|
|
|
|
|
|
if size1 == size2 { // all are nil
|
|
|
|
|
|
return true
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if size1 != nil {
|
|
|
|
|
|
if size2 == nil {
|
|
|
|
|
|
return false
|
|
|
|
|
|
}
|
|
|
|
|
|
return size1.Bytes() == size2.Bytes()
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
return false
|
|
|
|
|
|
}
|
