TRKJ/EdgeCommon
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeCommon.git synced 2025-11-04 21:50:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化源站有专属域名情况下的算法

Browse Source
This commit is contained in:
刘祥超
2022-05-23 19:59:05 +08:00
parent 24f8ccb166
commit 85553bf5d9
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
pkg/serverconfigs/reverse_proxy_config.go
View File

@@ -65,6 +65,7 @@ func (this *ReverseProxyConfig) Init() error {
// 将源站分组 // 将源站分组
this.schedulingGroupMap = map[string]*SchedulingGroup{} this.schedulingGroupMap = map[string]*SchedulingGroup{}
var hasDomainGroups = false
for _, origin := range this.PrimaryOrigins { for _, origin := range this.PrimaryOrigins {
if len(origin.Domains) == 0 { if len(origin.Domains) == 0 {
group, ok := this.schedulingGroupMap[""] group, ok := this.schedulingGroupMap[""]
@@ -77,6 +78,7 @@ func (this *ReverseProxyConfig) Init() error {
} }
group.PrimaryOrigins = append(group.PrimaryOrigins, origin) group.PrimaryOrigins = append(group.PrimaryOrigins, origin)
} else { } else {
hasDomainGroups = true
for _, domain := range origin.Domains { for _, domain := range origin.Domains {
group, ok := this.schedulingGroupMap[domain] group, ok := this.schedulingGroupMap[domain]
if !ok { if !ok {
@@ -102,6 +104,7 @@ func (this *ReverseProxyConfig) Init() error {
} }
group.BackupOrigins = append(group.BackupOrigins, origin) group.BackupOrigins = append(group.BackupOrigins, origin)
} else { } else {
hasDomainGroups = true
for _, domain := range origin.Domains { for _, domain := range origin.Domains {
group, ok := this.schedulingGroupMap[domain] group, ok := this.schedulingGroupMap[domain]
if !ok { if !ok {
@@ -116,6 +119,23 @@ func (this *ReverseProxyConfig) Init() error {
} }
} }
// 再次分解
if hasDomainGroups {
defaultGroup, ok := this.schedulingGroupMap[""]
if ok {
delete(this.schedulingGroupMap, "")
for _, group := range this.schedulingGroupMap {
for _, origin := range defaultGroup.PrimaryOrigins {
group.PrimaryOrigins = append(group.PrimaryOrigins, origin)
}
for _, origin := range defaultGroup.BackupOrigins {
group.BackupOrigins = append(group.BackupOrigins, origin)
}
}
}
}
// 初始化分组 // 初始化分组
for _, group := range this.schedulingGroupMap { for _, group := range this.schedulingGroupMap {
err := group.Init() err := group.Init()