lywsvip/cloudpods
1
0
Fork 0
mirror of https://github.com/yunionio/cloudpods.git synced 2026-05-13 18:43:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

region-dns: priority, weight for srv records

Browse Source
This commit is contained in:
Yousong Zhou
2018-10-18 15:37:38 +00:00
parent 1c64800f39
commit e4bb883c60
1 changed files with 26 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
pkg/dns/dns.go
View File

@@ -308,17 +308,35 @@ func (r *SRegionDNS) queryLocalDnsRecords(req *recordRequest) (recs []msg.Servic
ttl = defaultTTL
}
if req.IsSRV() {
parts := strings.SplitN(ip.Addr, ":", 2)
if len(parts) != 2 {
parts := strings.SplitN(ip.Addr, ":", 4)
if len(parts) < 2 {
ylog.Errorf("Invalid SRV records: %q", ip.Addr)
return
continue
}
port, e := strconv.Atoi(parts[1])
if e != nil {
ylog.Errorf("Invalid SRV records: %q", ip.Addr)
return
host := parts[0]
port, err := strconv.Atoi(parts[1])
if err != nil {
ylog.Errorf("SRV: invalid port: %s", ip.Addr)
continue
}
s = msg.Service{Host: parts[0], Port: port, TTL: ttl}
priority := 0
weight := 100
if len(parts) >= 3 {
var err error
weight, err = strconv.Atoi(parts[2])
if err != nil {
ylog.Errorf("SRV: invalid weight: %s", ip.Addr)
continue
}
if len(parts) >= 4 {
priority, err = strconv.Atoi(parts[3])
if err != nil {
ylog.Errorf("SRV: invalid priority: %s", ip.Addr)
continue
}
}
}
s = msg.Service{Host: host, Port: port, Weight: weight, Priority: priority, TTL: ttl}
} else {
s = msg.Service{Host: ip.Addr, TTL: ttl}
}