实现服务的缓存策略设置

pkg/serverconfigs/http_cache_ref.go

@@ -1,18 +1,80 @@
 package serverconfigs
 
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
+	"github.com/iwind/TeaGo/lists"
+	"strings"
+	"time"
+)
+
+var DefaultSkippedResponseCacheControlValues = []string{"private", "no-cache", "no-store"}
+
 type HTTPCacheRef struct {
-	IsPrior       bool           `yaml:"isPrior" json:"isPrior"`             // 是否覆盖
-	IsOn          bool           `yaml:"isOn" json:"isOn"`                   // 是否开启
-	CachePolicyId int64          `yaml:"cachePolicyId" json:"cachePolicyId"` // 缓存策略ID
-	Cond          *HTTPCacheCond `yaml:"cond" json:"cond"`                   // 条件
+	IsOn          bool  `yaml:"isOn" json:"isOn"`
+	CachePolicyId int64 `yaml:"cachePolicyId" json:"cachePolicyId"`
+
+	Key     string               `yaml:"key" json:"key"`         // 每个缓存的Key规则，里面可以有变量
+	Life    *shared.TimeDuration `yaml:"life" json:"life"`       // 时间
+	Status  []int                `yaml:"status" json:"status"`   // 缓存的状态码列表
+	MaxSize *shared.SizeCapacity `yaml:"maxSize" json:"maxSize"` // 能够请求的最大尺寸
+
+	SkipResponseCacheControlValues []string `yaml:"skipCacheControlValues" json:"skipCacheControlValues"`     // 可以跳过的响应的Cache-Control值
+	SkipResponseSetCookie          bool     `yaml:"skipSetCookie" json:"skipSetCookie"`                       // 是否跳过响应的Set-Cookie Header
+	EnableRequestCachePragma       bool     `yaml:"enableRequestCachePragma" json:"enableRequestCachePragma"` // 是否支持客户端的Pragma: no-cache
+
+	Conds *shared.HTTPRequestCondsConfig `yaml:"conds" json:"conds"` // 请求条件
+
+	CachePolicy *HTTPCachePolicy `yaml:"cachePolicy" json:"cachePolicy"`
+
+	life                            time.Duration
+	maxSize                         int64
+	uppercaseSkipCacheControlValues []string
 }
 
 func (this *HTTPCacheRef) Init() error {
-	if this.Cond != nil {
-		err := this.Cond.Init()
+	if this.MaxSize != nil {
+		this.maxSize = this.MaxSize.Bytes()
+	}
+	if this.Life != nil {
+		this.life = this.Life.Duration()
+	}
+
+	// control-values
+	this.uppercaseSkipCacheControlValues = []string{}
+	for _, value := range this.SkipResponseCacheControlValues {
+		this.uppercaseSkipCacheControlValues = append(this.uppercaseSkipCacheControlValues, strings.ToUpper(value))
+	}
+
+	// conds
+	if this.Conds != nil {
+		err := this.Conds.Init()
 		if err != nil {
 			return err
 		}
 	}
+
+	// policy
+	if this.CachePolicy != nil {
+		err := this.CachePolicy.Init()
+		if err != nil {
+			return err
+		}
+	}
+
 	return nil
 }
+
+// 最大数据尺寸
+func (this *HTTPCacheRef) MaxDataSize() int64 {
+	return this.maxSize
+}
+
+// 生命周期
+func (this *HTTPCacheRef) LifeDuration() time.Duration {
+	return this.life
+}
+
+// 是否包含某个Cache-Control值
+func (this *HTTPCacheRef) ContainsCacheControl(value string) bool {
+	return lists.ContainsString(this.uppercaseSkipCacheControlValues, strings.ToUpper(value))
+}