2020-11-20 11:38:12 +00:00
|
|
|
package eureka
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/xml"
|
|
|
|
"fmt"
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
|
|
|
|
)
|
|
|
|
|
|
|
|
var configMap = discoveryutils.NewConfigMap()
|
|
|
|
|
|
|
|
type apiConfig struct {
|
|
|
|
client *discoveryutils.Client
|
|
|
|
}
|
|
|
|
|
|
|
|
func newAPIConfig(sdc *SDConfig, baseDir string) (*apiConfig, error) {
|
2021-04-02 18:17:43 +00:00
|
|
|
ac, err := sdc.HTTPClientConfig.NewConfig(baseDir)
|
2020-11-20 11:38:12 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("cannot parse auth config: %w", err)
|
|
|
|
}
|
|
|
|
apiServer := sdc.Server
|
|
|
|
if apiServer == "" {
|
|
|
|
apiServer = "localhost:8080/eureka/v2"
|
|
|
|
}
|
|
|
|
if !strings.Contains(apiServer, "://") {
|
2021-04-02 18:17:43 +00:00
|
|
|
scheme := "http"
|
|
|
|
if sdc.HTTPClientConfig.TLSConfig != nil {
|
|
|
|
scheme = "https"
|
2020-11-20 11:38:12 +00:00
|
|
|
}
|
|
|
|
apiServer = scheme + "://" + apiServer
|
|
|
|
}
|
2021-04-03 21:40:08 +00:00
|
|
|
proxyAC, err := sdc.ProxyClientConfig.NewConfig(baseDir)
|
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("cannot parse proxy auth config: %w", err)
|
|
|
|
}
|
2023-07-06 22:59:56 +00:00
|
|
|
client, err := discoveryutils.NewClient(apiServer, ac, sdc.ProxyURL, proxyAC, &sdc.HTTPClientConfig)
|
2020-11-20 11:38:12 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("cannot create HTTP client for %q: %w", apiServer, err)
|
|
|
|
}
|
|
|
|
cfg := &apiConfig{
|
|
|
|
client: client,
|
|
|
|
}
|
|
|
|
return cfg, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func getAPIConfig(sdc *SDConfig, baseDir string) (*apiConfig, error) {
|
|
|
|
v, err := configMap.Get(sdc, func() (interface{}, error) { return newAPIConfig(sdc, baseDir) })
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return v.(*apiConfig), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func getAPIResponse(cfg *apiConfig, path string) ([]byte, error) {
|
|
|
|
return cfg.client.GetAPIResponse(path)
|
|
|
|
}
|
|
|
|
|
|
|
|
func parseAPIResponse(data []byte) (*applications, error) {
|
|
|
|
var apps applications
|
|
|
|
if err := xml.Unmarshal(data, &apps); err != nil {
|
|
|
|
return nil, fmt.Errorf("failed parse eureka api response: %q, err: %w", data, err)
|
|
|
|
}
|
|
|
|
return &apps, nil
|
|
|
|
}
|