TRKJ/EdgeCommon
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeCommon.git synced 2025-11-07 23:30:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

5秒盾增加匹配条件

Browse Source
This commit is contained in:
刘祥超
2024-01-12 17:13:34 +08:00
parent 64c3d76fb2
commit 3e8fc126b8
1 changed files with 23 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
pkg/serverconfigs/uam_config.go
View File

@@ -2,7 +2,9 @@
package serverconfigs package serverconfigs
import "github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared" import (
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
)
// UAMConfig UAM配置 // UAMConfig UAM配置
type UAMConfig struct { type UAMConfig struct {
@@ -13,6 +15,7 @@ type UAMConfig struct {
OnlyURLPatterns []*shared.URLPattern `yaml:"onlyURLPatterns" json:"onlyURLPatterns"` // 仅限的URL OnlyURLPatterns []*shared.URLPattern `yaml:"onlyURLPatterns" json:"onlyURLPatterns"` // 仅限的URL
ExceptURLPatterns []*shared.URLPattern `yaml:"exceptURLPatterns" json:"exceptURLPatterns"` // 排除的URL ExceptURLPatterns []*shared.URLPattern `yaml:"exceptURLPatterns" json:"exceptURLPatterns"` // 排除的URL
MinQPSPerIP int `yaml:"minQPSPerIP" json:"minQPSPerIP"` // 启用要求的单IP最低平均QPS MinQPSPerIP int `yaml:"minQPSPerIP" json:"minQPSPerIP"` // 启用要求的单IP最低平均QPS
Conds *shared.HTTPRequestCondsConfig `yaml:"conds" json:"conds"` // 匹配条件
} }
func NewUAMConfig() *UAMConfig { func NewUAMConfig() *UAMConfig {
@@ -38,6 +41,14 @@ func (this *UAMConfig) Init() error {
} }
} }
// conds
if this.Conds != nil {
err := this.Conds.Init()
if err != nil {
return err
}
}
return nil return nil
} }
@@ -62,3 +73,10 @@ func (this *UAMConfig) MatchURL(url string) bool {
return true return true
} }
func (this *UAMConfig) MatchRequest(formatter func(s string) string) bool {
if this.Conds == nil {
return true
}
return this.Conds.MatchRequest(formatter)
}