mirror of
https://github.com/prometheus/prometheus.git
synced 2024-12-28 15:09:39 -08:00
127 lines
3.3 KiB
Go
127 lines
3.3 KiB
Go
package hcloud
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"github.com/hetznercloud/hcloud-go/hcloud/schema"
|
|
)
|
|
|
|
// Location represents a location in the Hetzner Cloud.
|
|
type Location struct {
|
|
ID int
|
|
Name string
|
|
Description string
|
|
Country string
|
|
City string
|
|
Latitude float64
|
|
Longitude float64
|
|
NetworkZone NetworkZone
|
|
}
|
|
|
|
// LocationClient is a client for the location API.
|
|
type LocationClient struct {
|
|
client *Client
|
|
}
|
|
|
|
// GetByID retrieves a location by its ID. If the location does not exist, nil is returned.
|
|
func (c *LocationClient) GetByID(ctx context.Context, id int) (*Location, *Response, error) {
|
|
req, err := c.client.NewRequest(ctx, "GET", fmt.Sprintf("/locations/%d", id), nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
var body schema.LocationGetResponse
|
|
resp, err := c.client.Do(req, &body)
|
|
if err != nil {
|
|
if IsError(err, ErrorCodeNotFound) {
|
|
return nil, resp, nil
|
|
}
|
|
return nil, resp, err
|
|
}
|
|
return LocationFromSchema(body.Location), resp, nil
|
|
}
|
|
|
|
// GetByName retrieves an location by its name. If the location does not exist, nil is returned.
|
|
func (c *LocationClient) GetByName(ctx context.Context, name string) (*Location, *Response, error) {
|
|
if name == "" {
|
|
return nil, nil, nil
|
|
}
|
|
locations, response, err := c.List(ctx, LocationListOpts{Name: name})
|
|
if len(locations) == 0 {
|
|
return nil, response, err
|
|
}
|
|
return locations[0], response, err
|
|
}
|
|
|
|
// Get retrieves a location by its ID if the input can be parsed as an integer, otherwise it
|
|
// retrieves a location by its name. If the location does not exist, nil is returned.
|
|
func (c *LocationClient) Get(ctx context.Context, idOrName string) (*Location, *Response, error) {
|
|
if id, err := strconv.Atoi(idOrName); err == nil {
|
|
return c.GetByID(ctx, int(id))
|
|
}
|
|
return c.GetByName(ctx, idOrName)
|
|
}
|
|
|
|
// LocationListOpts specifies options for listing location.
|
|
type LocationListOpts struct {
|
|
ListOpts
|
|
Name string
|
|
}
|
|
|
|
func (l LocationListOpts) values() url.Values {
|
|
vals := l.ListOpts.values()
|
|
if l.Name != "" {
|
|
vals.Add("name", l.Name)
|
|
}
|
|
return vals
|
|
}
|
|
|
|
// List returns a list of locations for a specific page.
|
|
//
|
|
// Please note that filters specified in opts are not taken into account
|
|
// when their value corresponds to their zero value or when they are empty.
|
|
func (c *LocationClient) List(ctx context.Context, opts LocationListOpts) ([]*Location, *Response, error) {
|
|
path := "/locations?" + opts.values().Encode()
|
|
req, err := c.client.NewRequest(ctx, "GET", path, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
var body schema.LocationListResponse
|
|
resp, err := c.client.Do(req, &body)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
locations := make([]*Location, 0, len(body.Locations))
|
|
for _, i := range body.Locations {
|
|
locations = append(locations, LocationFromSchema(i))
|
|
}
|
|
return locations, resp, nil
|
|
}
|
|
|
|
// All returns all locations.
|
|
func (c *LocationClient) All(ctx context.Context) ([]*Location, error) {
|
|
allLocations := []*Location{}
|
|
|
|
opts := LocationListOpts{}
|
|
opts.PerPage = 50
|
|
|
|
_, err := c.client.all(func(page int) (*Response, error) {
|
|
opts.Page = page
|
|
locations, resp, err := c.List(ctx, opts)
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
allLocations = append(allLocations, locations...)
|
|
return resp, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return allLocations, nil
|
|
}
|