mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-25 11:50:13 +01:00
108 lines
3.8 KiB
Go
108 lines
3.8 KiB
Go
package azure
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promauth"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/proxy"
|
|
)
|
|
|
|
// SDCheckInterval is check interval for Azure service discovery.
|
|
var SDCheckInterval = flag.Duration("promscrape.azureSDCheckInterval", 60*time.Second, "Interval for checking for changes in Azure. "+
|
|
"This works only if azure_sd_configs is configured in '-promscrape.config' file. "+
|
|
"See https://docs.victoriametrics.com/sd_configs.html#azure_sd_configs for details")
|
|
|
|
// SDConfig represents service discovery config for Azure.
|
|
//
|
|
// See https://prometheus.io/docs/prometheus/latest/configuration/configuration/#azure_sd_config
|
|
type SDConfig struct {
|
|
Environment string `yaml:"environment,omitempty"`
|
|
|
|
// AuthenticationMethod can be either Oauth or ManagedIdentity.
|
|
// See https://docs.microsoft.com/en-us/azure/active-directory/managed-identities-azure-resources/overview
|
|
AuthenticationMethod string `yaml:"authentication_method,omitempty"`
|
|
|
|
SubscriptionID string `yaml:"subscription_id"`
|
|
TenantID string `yaml:"tenant_id,omitempty"`
|
|
ClientID string `yaml:"client_id,omitempty"`
|
|
ClientSecret *promauth.Secret `yaml:"client_secret,omitempty"`
|
|
ResourceGroup string `yaml:"resource_group,omitempty"`
|
|
|
|
// RefreshInterval time.Duration `yaml:"refresh_interval"`
|
|
// refresh_interval is obtained from `-promscrape.azureSDCheckInterval` command-line option.
|
|
|
|
Port int `yaml:"port"`
|
|
|
|
HTTPClientConfig promauth.HTTPClientConfig `yaml:",inline"`
|
|
ProxyURL *proxy.URL `yaml:"proxy_url,omitempty"`
|
|
ProxyClientConfig promauth.ProxyClientConfig `yaml:",inline"`
|
|
}
|
|
|
|
// GetLabels returns Consul labels according to sdc.
|
|
func (sdc *SDConfig) GetLabels(baseDir string) ([]map[string]string, error) {
|
|
ac, err := getAPIConfig(sdc, baseDir)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot get API config: %w", err)
|
|
}
|
|
vms, err := getVirtualMachines(ac)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return appendMachineLabels(vms, ac.port, sdc), nil
|
|
}
|
|
|
|
// MustStop stops further usage for sdc.
|
|
func (sdc *SDConfig) MustStop() {
|
|
configMap.Delete(sdc)
|
|
}
|
|
|
|
func appendMachineLabels(vms []virtualMachine, port int, sdc *SDConfig) []map[string]string {
|
|
ms := make([]map[string]string, 0, len(vms))
|
|
for i := range vms {
|
|
vm := &vms[i]
|
|
for _, ips := range vm.ipAddresses {
|
|
if ips.privateIP == "" {
|
|
continue
|
|
}
|
|
addr := discoveryutils.JoinHostPort(ips.privateIP, port)
|
|
m := map[string]string{
|
|
"__address__": addr,
|
|
"__meta_azure_subscription_id": sdc.SubscriptionID,
|
|
"__meta_azure_machine_id": vm.ID,
|
|
"__meta_azure_machine_name": vm.Name,
|
|
"__meta_azure_machine_location": vm.Location,
|
|
"__meta_azure_machine_private_ip": ips.privateIP,
|
|
}
|
|
if sdc.TenantID != "" {
|
|
m["__meta_azure_tenant_id"] = sdc.TenantID
|
|
}
|
|
// /subscriptions/SUBSCRIPTION_ID/resourceGroups/RESOURCE_GROUP/providers/PROVIDER/TYPE/NAME
|
|
idPath := strings.Split(vm.ID, "/")
|
|
if len(idPath) > 4 {
|
|
m["__meta_azure_machine_resource_group"] = idPath[4]
|
|
}
|
|
if vm.Properties.StorageProfile.OsDisk.OsType != "" {
|
|
m["__meta_azure_machine_os_type"] = vm.Properties.StorageProfile.OsDisk.OsType
|
|
}
|
|
if vm.Properties.OsProfile.ComputerName != "" {
|
|
m["__meta_azure_machine_computer_name"] = vm.Properties.OsProfile.ComputerName
|
|
}
|
|
if ips.publicIP != "" {
|
|
m["__meta_azure_machine_public_ip"] = ips.publicIP
|
|
}
|
|
if vm.scaleSet != "" {
|
|
m["__meta_azure_machine_scale_set"] = vm.scaleSet
|
|
}
|
|
for k, v := range vm.Tags {
|
|
m[discoveryutils.SanitizeLabelName("__meta_azure_machine_tag_"+k)] = v
|
|
}
|
|
ms = append(ms, m)
|
|
}
|
|
}
|
|
return ms
|
|
}
|