Archived
95 lines
2.0 KiB
Go
95 lines
2.0 KiB
Go
package shared
|
|
|
|
// 条件配置
|
|
// 数据结构：conds -> []groups -> []cond
|
|
type HTTPRequestCondsConfig struct {
|
|
IsOn bool `yaml:"isOn" json:"isOn"`
|
|
Connector string `yaml:"connector" json:"connector"`
|
|
Groups []*HTTPRequestCondGroup `yaml:"groups" json:"groups"`
|
|
|
|
hasRequestConds bool
|
|
hasResponseConds bool
|
|
}
|
|
|
|
// 初始化
|
|
func (this *HTTPRequestCondsConfig) Init() error {
|
|
if len(this.Connector) == 0 {
|
|
this.Connector = "or"
|
|
}
|
|
|
|
for _, group := range this.Groups {
|
|
err := group.Init()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// 是否有请求条件
|
|
for _, group := range this.Groups {
|
|
if group.IsOn {
|
|
if group.HasRequestConds() {
|
|
this.hasRequestConds = true
|
|
}
|
|
if group.HasResponseConds() {
|
|
this.hasResponseConds = true
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// 判断请求是否匹配
|
|
func (this *HTTPRequestCondsConfig) MatchRequest(formatter func(s string) string) bool {
|
|
if !this.IsOn && len(this.Groups) == 0 {
|
|
return true
|
|
}
|
|
ok := false
|
|
for _, group := range this.Groups {
|
|
b := group.MatchRequest(formatter)
|
|
if !b && this.Connector == "and" {
|
|
return false
|
|
}
|
|
if b && this.Connector == "or" {
|
|
return true
|
|
}
|
|
if b {
|
|
// 对于 or 来说至少有一个分组要返回 true
|
|
ok = true
|
|
}
|
|
}
|
|
return ok
|
|
}
|
|
|
|
// 判断响应是否匹配
|
|
func (this *HTTPRequestCondsConfig) MatchResponse(formatter func(s string) string) bool {
|
|
if !this.IsOn && len(this.Groups) == 0 {
|
|
return true
|
|
}
|
|
ok := false
|
|
for _, group := range this.Groups {
|
|
b := group.MatchResponse(formatter)
|
|
if !b && this.Connector == "and" {
|
|
return false
|
|
}
|
|
if b && this.Connector == "or" {
|
|
return true
|
|
}
|
|
if b {
|
|
// 对于 or 来说至少有一个分组要返回 true
|
|
ok = true
|
|
}
|
|
}
|
|
return ok
|
|
}
|
|
|
|
// 判断是否有请求条件
|
|
func (this *HTTPRequestCondsConfig) HasRequestConds() bool {
|
|
return this.hasRequestConds
|
|
}
|
|
|
|
// 判断是否有响应条件
|
|
func (this *HTTPRequestCondsConfig) HasResponseConds() bool {
|
|
return this.hasResponseConds
|
|
}
|