mirror of https://github.com/prometheus/prometheus
129 lines
3.7 KiB
Go
129 lines
3.7 KiB
Go
// Copyright 2022 The Prometheus Authors
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package ionos
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/go-kit/log"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/common/config"
|
|
"github.com/prometheus/common/model"
|
|
|
|
"github.com/prometheus/prometheus/discovery"
|
|
"github.com/prometheus/prometheus/discovery/refresh"
|
|
)
|
|
|
|
const (
|
|
// metaLabelPrefix is the meta prefix used for all meta labels in this
|
|
// discovery.
|
|
metaLabelPrefix = model.MetaLabelPrefix + "ionos_"
|
|
metaLabelSeparator = ","
|
|
)
|
|
|
|
func init() {
|
|
discovery.RegisterConfig(&SDConfig{})
|
|
}
|
|
|
|
// Discovery periodically performs IONOS Cloud target discovery. It implements
|
|
// the Discoverer interface.
|
|
type Discovery struct{}
|
|
|
|
// NewDiscovery returns a new refresh.Discovery for IONOS Cloud.
|
|
func NewDiscovery(conf *SDConfig, logger log.Logger, metrics discovery.DiscovererMetrics) (*refresh.Discovery, error) {
|
|
m, ok := metrics.(*ionosMetrics)
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid discovery metrics type")
|
|
}
|
|
|
|
if conf.ionosEndpoint == "" {
|
|
conf.ionosEndpoint = "https://api.ionos.com"
|
|
}
|
|
|
|
d, err := newServerDiscovery(conf, logger)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return refresh.NewDiscovery(
|
|
refresh.Options{
|
|
Logger: logger,
|
|
Mech: "ionos",
|
|
Interval: time.Duration(conf.RefreshInterval),
|
|
RefreshF: d.refresh,
|
|
MetricsInstantiator: m.refreshMetrics,
|
|
},
|
|
), nil
|
|
}
|
|
|
|
// DefaultSDConfig is the default IONOS Cloud service discovery configuration.
|
|
var DefaultSDConfig = SDConfig{
|
|
HTTPClientConfig: config.DefaultHTTPClientConfig,
|
|
RefreshInterval: model.Duration(60 * time.Second),
|
|
Port: 80,
|
|
}
|
|
|
|
// SDConfig configuration to use for IONOS Cloud Discovery.
|
|
type SDConfig struct {
|
|
// DatacenterID: IONOS Cloud data center ID used to discover targets.
|
|
DatacenterID string `yaml:"datacenter_id"`
|
|
|
|
HTTPClientConfig config.HTTPClientConfig `yaml:",inline"`
|
|
|
|
RefreshInterval model.Duration `yaml:"refresh_interval"`
|
|
Port int `yaml:"port"`
|
|
|
|
ionosEndpoint string // For tests only.
|
|
}
|
|
|
|
// NewDiscovererMetrics implements discovery.Config.
|
|
func (*SDConfig) NewDiscovererMetrics(reg prometheus.Registerer, rmi discovery.RefreshMetricsInstantiator) discovery.DiscovererMetrics {
|
|
return &ionosMetrics{
|
|
refreshMetrics: rmi,
|
|
}
|
|
}
|
|
|
|
// Name returns the name of the IONOS Cloud service discovery.
|
|
func (c SDConfig) Name() string {
|
|
return "ionos"
|
|
}
|
|
|
|
// NewDiscoverer returns a new discovery.Discoverer for IONOS Cloud.
|
|
func (c SDConfig) NewDiscoverer(options discovery.DiscovererOptions) (discovery.Discoverer, error) {
|
|
return NewDiscovery(&c, options.Logger, options.Metrics)
|
|
}
|
|
|
|
// UnmarshalYAML implements the yaml.Unmarshaler interface.
|
|
func (c *SDConfig) UnmarshalYAML(unmarshal func(interface{}) error) error {
|
|
*c = DefaultSDConfig
|
|
type plain SDConfig
|
|
err := unmarshal((*plain)(c))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if c.DatacenterID == "" {
|
|
return errors.New("datacenter id can't be empty")
|
|
}
|
|
|
|
return c.HTTPClientConfig.Validate()
|
|
}
|
|
|
|
// SetDirectory joins any relative file paths with dir.
|
|
func (c *SDConfig) SetDirectory(dir string) {
|
|
c.HTTPClientConfig.SetDirectory(dir)
|
|
}
|