config.go

Documentation: edge-infra.dev/pkg/edge/monitoring/k8s/controllers/prometheusctl

     1  package prometheusctl
     2  
     3  import (
     4  	"errors"
     5  	"flag"
     6  	"fmt"
     7  	"time"
     8  
     9  	"github.com/peterbourgon/ff/v3"
    10  
    11  	"edge-infra.dev/pkg/edge/constants/api/cluster"
    12  	"edge-infra.dev/pkg/k8s/meta"
    13  )
    14  
    15  type config struct {
    16  	prometheus        meta.NamespacedObjectReference
    17  	reconcileInterval time.Duration
    18  	clusterProvider   string
    19  }
    20  
    21  // validate checks all env variable fields are present
    22  func (cfg *config) validate() error {
    23  	if cfg.prometheus.Name == "" {
    24  		return errors.New("--prometheus-name must be provided")
    25  	}
    26  	if cfg.prometheus.Namespace == "" {
    27  		return errors.New("--prometheus-namespace must be provided")
    28  	}
    29  
    30  	return nil
    31  }
    32  
    33  // NewConfig creates a new config and flagset from os args
    34  func newConfig(args []string) (*config, error) {
    35  	cfg := &config{}
    36  
    37  	fs := flag.NewFlagSet("prometheus-stackdriver", flag.ContinueOnError)
    38  	fs.StringVar(&cfg.prometheus.Name, "prometheus-name", "prometheus",
    39  		"name of prometheus object to reconcile")
    40  	fs.StringVar(&cfg.prometheus.Namespace, "prometheus-namespace", "prometheus",
    41  		"namespace of prometheus object to reconcile")
    42  	fs.DurationVar(&cfg.reconcileInterval, "reconcile-interval", time.Second*60,
    43  		"how often to re-reconcile when an error occurs")
    44  	fs.StringVar(&cfg.clusterProvider, "cluster-provider", string(cluster.Generic),
    45  		fmt.Sprintf("K8s cluster provider: %s. Takes precedence over other provider flags.", cluster.Types))
    46  
    47  	if err := ff.Parse(fs, args[1:], ff.WithEnvVarNoPrefix()); err != nil {
    48  		return nil, err
    49  	}
    50  
    51  	if err := cfg.validate(); err != nil {
    52  		return nil, err
    53  	}
    54  
    55  	return cfg, nil
    56  }
    57
View as plain text