mirror of
https://github.com/TeaOSLab/EdgeCommon.git
synced 2025-11-03 04:10:25 +08:00
251 lines
7.6 KiB
Go
251 lines
7.6 KiB
Go
package serverconfigs
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/ossconfigs"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/sslconfigs"
|
||
)
|
||
|
||
// OriginConfig 源站服务配置
|
||
type OriginConfig struct {
|
||
Id int64 `yaml:"id" json:"id"` // ID
|
||
IsOn bool `yaml:"isOn" json:"isOn"` // 是否启用
|
||
Version int `yaml:"version" json:"version"` // 版本
|
||
Name string `yaml:"name" json:"name"` // 名称
|
||
Addr *NetworkAddressConfig `yaml:"addr" json:"addr"` // 地址
|
||
HTTP2Enabled bool `yaml:"http2Enabled" json:"http2Enabled"` // 是否支持HTTP2
|
||
OSS *ossconfigs.OSSConfig `yaml:"oss" json:"oss"` // 对象存储配置
|
||
Description string `yaml:"description" json:"description"` // 描述
|
||
Code string `yaml:"code" json:"code"` // 代号
|
||
|
||
Weight uint `yaml:"weight" json:"weight"` // 权重
|
||
ConnTimeout *shared.TimeDuration `yaml:"connTimeout" json:"connTimeout"` // 连接失败超时
|
||
ReadTimeout *shared.TimeDuration `yaml:"readTimeout" json:"readTimeout"` // 读取超时时间
|
||
IdleTimeout *shared.TimeDuration `yaml:"idleTimeout" json:"idleTimeout"` // 空闲连接超时时间
|
||
MaxFails int `yaml:"maxFails" json:"maxFails"` // 最多失败次数
|
||
MaxConns int `yaml:"maxConns" json:"maxConns"` // 最大并发连接数
|
||
MaxIdleConns int `yaml:"idleConns" json:"idleConns"` // 最大空闲连接数
|
||
|
||
Domains []string `yaml:"domains" json:"domains"` // 所属域名
|
||
|
||
StripPrefix string `yaml:"stripPrefix" json:"stripPrefix"` // 去除URL前缀
|
||
RequestURI string `yaml:"requestURI" json:"requestURI"` // 转发后的请求URI
|
||
RequestHost string `yaml:"requestHost" json:"requestHost"` // 自定义主机名
|
||
FollowPort bool `yaml:"followPort" json:"followPort"` // 端口跟随
|
||
FollowProtocol *FollowProtocolConfig `yaml:"followProtocol" json:"followProtocol"` // 协议跟随
|
||
|
||
RequestHeaderPolicyRef *shared.HTTPHeaderPolicyRef `yaml:"requestHeaderPolicyRef" json:"requestHeaderPolicyRef"` // 请求Header
|
||
RequestHeaderPolicy *shared.HTTPHeaderPolicy `yaml:"requestHeaderPolicy" json:"requestHeaderPolicy"` // 请求Header策略
|
||
ResponseHeaderPolicyRef *shared.HTTPHeaderPolicyRef `yaml:"responseHeaderPolicyRef" json:"responseHeaderPolicyRef"` // 响应Header`
|
||
ResponseHeaderPolicy *shared.HTTPHeaderPolicy `yaml:"responseHeaderPolicy" json:"responseHeaderPolicy"` // 响应Header策略
|
||
|
||
// 健康检查URL，目前支持：
|
||
// - http|https 返回2xx-3xx认为成功
|
||
HealthCheck *HealthCheckConfig `yaml:"healthCheck" json:"healthCheck"` // TODO
|
||
|
||
CertRef *sslconfigs.SSLCertRef `yaml:"certRef" json:"certRef"` // 证书的引用
|
||
Cert *sslconfigs.SSLCertConfig `yaml:"cert" json:"cert"` // 请求源服务器用的证书
|
||
|
||
// ftp
|
||
FTPServerRef *FTPServerRef `yaml:"ftpServerRef" json:"ftpServerRef"` // TODO
|
||
FTPServer *FTPServerConfig `yaml:"ftpServer" json:"ftpServer"` // TODO
|
||
|
||
IsOk bool `yaml:"isOk" json:"isOk"` // 是否可以正常访问，仅仅用于运行时检查
|
||
|
||
connTimeoutDuration time.Duration
|
||
readTimeoutDuration time.Duration
|
||
idleTimeoutDuration time.Duration
|
||
|
||
hasRequestURI bool
|
||
requestPath string
|
||
requestArgs string
|
||
|
||
//hasRequestHeaders bool
|
||
//hasResponseHeaders bool
|
||
|
||
uniqueKey string
|
||
|
||
requestHostHasVariables bool
|
||
requestURIHasVariables bool
|
||
}
|
||
|
||
// Init 校验
|
||
func (this *OriginConfig) Init(ctx context.Context) error {
|
||
this.IsOk = true
|
||
|
||
// URL
|
||
this.requestHostHasVariables = configutils.HasVariables(this.RequestHost)
|
||
this.requestURIHasVariables = configutils.HasVariables(this.RequestURI)
|
||
|
||
// unique key
|
||
this.uniqueKey = strconv.FormatInt(this.Id, 10) + "@" + strconv.Itoa(this.Version) + "@" + fmt.Sprintf("%p", this)
|
||
|
||
// addr
|
||
if this.Addr != nil {
|
||
err := this.Addr.Init()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
// 证书
|
||
if this.Cert != nil {
|
||
err := this.Cert.Init(ctx)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
// connTimeout
|
||
if this.ConnTimeout != nil {
|
||
this.connTimeoutDuration = this.ConnTimeout.Duration()
|
||
}
|
||
|
||
// readTimeout
|
||
if this.ReadTimeout != nil {
|
||
this.readTimeoutDuration = this.ReadTimeout.Duration()
|
||
}
|
||
|
||
// idleTimeout
|
||
if this.IdleTimeout != nil {
|
||
this.idleTimeoutDuration = this.IdleTimeout.Duration()
|
||
}
|
||
|
||
// Headers
|
||
if this.RequestHeaderPolicyRef != nil {
|
||
err := this.RequestHeaderPolicyRef.Init()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
if this.RequestHeaderPolicy != nil {
|
||
err := this.RequestHeaderPolicy.Init()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
if this.ResponseHeaderPolicyRef != nil {
|
||
err := this.ResponseHeaderPolicyRef.Init()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
if this.ResponseHeaderPolicy != nil {
|
||
err := this.ResponseHeaderPolicy.Init()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
// request uri
|
||
if len(this.RequestURI) == 0 || this.RequestURI == "${requestURI}" {
|
||
this.hasRequestURI = false
|
||
} else {
|
||
this.hasRequestURI = true
|
||
|
||
if strings.Contains(this.RequestURI, "?") {
|
||
pieces := strings.SplitN(this.RequestURI, "?", -1)
|
||
this.requestPath = pieces[0]
|
||
this.requestArgs = pieces[1]
|
||
} else {
|
||
this.requestPath = this.RequestURI
|
||
}
|
||
}
|
||
|
||
// health check
|
||
if this.HealthCheck != nil {
|
||
err := this.HealthCheck.Init()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
// follow protocol
|
||
if this.FollowProtocol != nil {
|
||
err := this.FollowProtocol.Init()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
// oss
|
||
if this.OSS != nil {
|
||
err := this.OSS.Init()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// CandidateCodes 候选对象代号
|
||
func (this *OriginConfig) CandidateCodes() []string {
|
||
codes := []string{strconv.FormatInt(this.Id, 10)}
|
||
if len(this.Code) > 0 {
|
||
codes = append(codes, this.Code)
|
||
}
|
||
return codes
|
||
}
|
||
|
||
// CandidateWeight 候选对象权重
|
||
func (this *OriginConfig) CandidateWeight() uint {
|
||
return this.Weight
|
||
}
|
||
|
||
// ConnTimeoutDuration 连接超时时间
|
||
func (this *OriginConfig) ConnTimeoutDuration() time.Duration {
|
||
return this.connTimeoutDuration
|
||
}
|
||
|
||
// ReadTimeoutDuration 读取超时时间
|
||
func (this *OriginConfig) ReadTimeoutDuration() time.Duration {
|
||
return this.readTimeoutDuration
|
||
}
|
||
|
||
// IdleTimeoutDuration 休眠超时时间
|
||
func (this *OriginConfig) IdleTimeoutDuration() time.Duration {
|
||
return this.idleTimeoutDuration
|
||
}
|
||
|
||
// RequestHostHasVariables 判断RequestHost是否有变量
|
||
func (this *OriginConfig) RequestHostHasVariables() bool {
|
||
return this.requestHostHasVariables
|
||
}
|
||
|
||
// RequestURIHasVariables 判断RequestURI是否有变量
|
||
func (this *OriginConfig) RequestURIHasVariables() bool {
|
||
return this.requestURIHasVariables
|
||
}
|
||
|
||
// UniqueKey 唯一Key
|
||
func (this *OriginConfig) UniqueKey() string {
|
||
return this.uniqueKey
|
||
}
|
||
|
||
// IsOSS 判断当前源站是否为OSS
|
||
func (this *OriginConfig) IsOSS() bool {
|
||
return this.Addr != nil && ossconfigs.IsOSSProtocol(this.Addr.Protocol.String())
|
||
}
|
||
|
||
// AddrSummary 地址描述
|
||
func (this *OriginConfig) AddrSummary() string {
|
||
if this.Addr == nil {
|
||
return ""
|
||
}
|
||
|
||
// OSS
|
||
if ossconfigs.IsOSSProtocol(this.Addr.Protocol.String()) && this.OSS != nil {
|
||
return this.OSS.Summary()
|
||
}
|
||
|
||
return this.Addr.Protocol.String() + "://" + this.Addr.Host + ":" + this.Addr.PortRange
|
||
}
|