83b4a3fe55
- Exported `Fetcher` interface - Inject `Fetcher` to publicip loop and updaters - Get public IP and information at the same time - Only query ipinfo.io - Make `MultiInfo` part of the `Fetch` object
31 lines
695 B
Go
31 lines
695 B
Go
package updater
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
|
|
"github.com/qdm12/gluetun/internal/models"
|
|
"github.com/qdm12/gluetun/internal/provider/common"
|
|
)
|
|
|
|
func setLocationInfo(ctx context.Context, fetcher common.IPFetcher, servers []models.Server) (err error) {
|
|
// Get public IP address information
|
|
ipsToGetInfo := make([]net.IP, 0, len(servers))
|
|
for _, server := range servers {
|
|
ipsToGetInfo = append(ipsToGetInfo, server.IPs...)
|
|
}
|
|
ipsInfo, err := fetcher.FetchMultiInfo(ctx, ipsToGetInfo)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for i := range servers {
|
|
ipInfo := ipsInfo[i]
|
|
servers[i].Country = ipInfo.Country
|
|
servers[i].Region = ipInfo.Region
|
|
servers[i].City = ipInfo.City
|
|
}
|
|
|
|
return nil
|
|
}
|