ebd94723c1
- Each provider interface can now fetch updated servers data - Rename each provider updater subpackage name to `updater` - Updater constructor does not take a settings struct - Updater update method takes in a slice of provider strings
43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
package updater
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/qdm12/gluetun/internal/models"
|
|
)
|
|
|
|
type Provider interface {
|
|
Name() string
|
|
FetchServers(ctx context.Context, minServers int) (servers []models.Server, err error)
|
|
}
|
|
|
|
func (u *Updater) updateProvider(ctx context.Context, provider Provider) (err error) {
|
|
providerName := provider.Name()
|
|
existingServersCount := u.storage.GetServersCount(providerName)
|
|
minServers := getMinServers(existingServersCount)
|
|
servers, err := provider.FetchServers(ctx, minServers)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot get servers: %w", err)
|
|
}
|
|
|
|
if u.storage.ServersAreEqual(providerName, servers) {
|
|
return nil
|
|
}
|
|
|
|
// Note the servers variable must NOT BE MUTATED after this call,
|
|
// since the implementation does not deep copy the servers.
|
|
// TODO set in storage in provider updater directly, server by server,
|
|
// to avoid accumulating server data in memory.
|
|
err = u.storage.SetServers(providerName, servers)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot set servers to storage: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func getMinServers(existingServersCount int) (minServers int) {
|
|
const minRatio = 0.8
|
|
return int(minRatio * float64(existingServersCount))
|
|
}
|