package shared // HTTPRequestCondsConfig 条件配置 // 数据结构: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 } // Init 初始化 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 } // MatchRequest 判断请求是否匹配 func (this *HTTPRequestCondsConfig) MatchRequest(formatter Formatter) 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 } // MatchResponse 判断响应是否匹配 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 } // HasRequestConds 判断是否有请求条件 func (this *HTTPRequestCondsConfig) HasRequestConds() bool { return this.hasRequestConds } // HasResponseConds 判断是否有响应条件 func (this *HTTPRequestCondsConfig) HasResponseConds() bool { return this.hasResponseConds }