main.go

package main

import (
	"flag"
	"fmt"
	"github.com/Andywugh/prometheus-pusher/scrape"
	"github.com/gin-contrib/pprof"
	"github.com/gin-gonic/gin"
	"github.com/prometheus/prometheus/config"
	"github.com/prometheus/prometheus/retrieval"
	"github.com/prometheus/prometheus/storage"
	"strings"
)

var cfg = struct {
	configFile        string
	customLabels      string
	customLabelValues string
}{}

var (
	labels, values []string
)

func init() {
	flag.StringVar(
		&cfg.configFile, "config.file", "prometheus_pusher.yml",
		"Prometheus configuration file name.",
	)
	flag.StringVar(
		&cfg.customLabels, "config.customLabels", "", "custom metrics labels",
	)
	flag.StringVar(
		&cfg.customLabelValues, "config.customLabelValues", "", "custom mertics label values",
	)
}

func main() {

	flag.Parse()

	var (
		sampleAppender = storage.Fanout{}
		targetManager  = retrieval.NewTargetManager(sampleAppender)
		jobTargets     = scrape.NewJobTargets(targetManager)
	)

	fmt.Println("Loading prometheus config file: " + cfg.configFile)
	fmt.Println("Custom labels: " + cfg.customLabels + "\t Custom label values: " + cfg.customLabelValues)

	if cfg.customLabels == "" {
		labels = []string{}
		values = []string{}
	} else {
		labels = strings.Split(cfg.customLabels, ",")
		values = strings.Split(cfg.customLabelValues, ",")
	}

	var (
		scrapeManager = scrape.NewExporterScrape(jobTargets, labels, values)
	)

	conf, err := config.LoadFile(cfg.configFile)
	if err != nil {
		fmt.Println(err.Error())
		return
	}

	targetManager.ApplyConfig(conf)

	go targetManager.Run()
	defer targetManager.Stop()

	scrapeManager.AppConfig(conf)

	go scrapeManager.Run()
	defer scrapeManager.Stop()

	r := gin.Default()
	pprof.Register(r, nil) // NOQA

	r.GET("/ping", func(c *gin.Context) {
		c.JSON(200, gin.H{
			"message": "pong",
		})
	})
	r.GET("/targets", func(c *gin.Context) {
		c.JSON(200, jobTargets.Targets())
	})
	r.Run()

}