1.4.5.2

EdgeCommon/pkg/serverconfigs/shared/http_request_conds_config.go

@@ -0,0 +1,94 @@
+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
+}