open-consul/agent/hcp/discover/discover.go

77 lines
1.3 KiB
Go

package discover
import (
"context"
"fmt"
"log"
"time"
"github.com/hashicorp/consul/agent/hcp"
"github.com/hashicorp/consul/agent/hcp/config"
)
type Provider struct {
}
var (
defaultTimeout = 5 * time.Second
)
type providerConfig struct {
config.CloudConfig
timeout time.Duration
}
func (p *Provider) Addrs(args map[string]string, l *log.Logger) ([]string, error) {
cfg, err := parseArgs(args)
if err != nil {
return nil, err
}
client, err := hcp.NewClient(cfg.CloudConfig)
if err != nil {
return nil, err
}
ctx, cancel := context.WithTimeout(context.Background(), cfg.timeout)
defer cancel()
servers, err := client.DiscoverServers(ctx)
if err != nil {
return nil, err
}
return servers, nil
}
func (p *Provider) Help() string {
return ""
}
func parseArgs(args map[string]string) (cfg providerConfig, err error) {
cfg.timeout = defaultTimeout
if id, ok := args["resource_id"]; ok {
cfg.ResourceID = id
} else {
err = fmt.Errorf("'resource_id' was not found and is required")
}
if cid, ok := args["client_id"]; ok {
cfg.ClientID = cid
}
if csec, ok := args["client_secret"]; ok {
cfg.ClientSecret = csec
}
if timeoutRaw, ok := args["timeout"]; ok {
timeout, err := time.ParseDuration(timeoutRaw)
if err != nil {
return cfg, err
}
cfg.timeout = timeout
}
return
}