96 lines
1.8 KiB
Go
96 lines
1.8 KiB
Go
package updater
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/qdm12/gluetun/internal/models"
|
|
"github.com/qdm12/gluetun/internal/provider/common"
|
|
)
|
|
|
|
func (u *Updater) FetchServers(ctx context.Context, minServers int) (
|
|
servers []models.Server, err error) {
|
|
nts := make(nameToServer)
|
|
|
|
noChangeCounter := 0
|
|
const maxNoChange = 10
|
|
const betweenDuration = 200 * time.Millisecond
|
|
const maxDuration = time.Minute
|
|
|
|
maxTimer := time.NewTimer(maxDuration)
|
|
|
|
for {
|
|
data, err := fetchAPI(ctx, u.client)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
change := addData(data.Regions, nts)
|
|
|
|
if !change {
|
|
noChangeCounter++
|
|
if noChangeCounter == maxNoChange {
|
|
break
|
|
}
|
|
} else {
|
|
noChangeCounter = 0
|
|
}
|
|
|
|
timer := time.NewTimer(betweenDuration)
|
|
maxTimeout := false
|
|
select {
|
|
case <-ctx.Done():
|
|
if !timer.Stop() {
|
|
<-timer.C
|
|
}
|
|
if !maxTimer.Stop() {
|
|
<-timer.C
|
|
}
|
|
return nil, ctx.Err()
|
|
case <-timer.C:
|
|
case <-maxTimer.C:
|
|
if !timer.Stop() {
|
|
<-timer.C
|
|
}
|
|
maxTimeout = true
|
|
}
|
|
|
|
if maxTimeout {
|
|
break
|
|
}
|
|
}
|
|
|
|
servers = nts.toServersSlice()
|
|
|
|
if len(servers) < minServers {
|
|
return nil, fmt.Errorf("%w: %d and expected at least %d",
|
|
common.ErrNotEnoughServers, len(servers), minServers)
|
|
}
|
|
|
|
sort.Sort(models.SortableServers(servers))
|
|
|
|
return servers, nil
|
|
}
|
|
|
|
func addData(regions []regionData, nts nameToServer) (change bool) {
|
|
for _, region := range regions {
|
|
for _, server := range region.Servers.UDP {
|
|
const tcp, udp = false, true
|
|
if nts.add(server.CN, region.DNS, region.Name, tcp, udp, region.PortForward, server.IP) {
|
|
change = true
|
|
}
|
|
}
|
|
|
|
for _, server := range region.Servers.TCP {
|
|
const tcp, udp = true, false
|
|
if nts.add(server.CN, region.DNS, region.Name, tcp, udp, region.PortForward, server.IP) {
|
|
change = true
|
|
}
|
|
}
|
|
}
|
|
|
|
return change
|
|
}
|