influxdb/cmd/telemetryd/main.go

package main

import (
	"fmt"
	"net/http"
	"os"
	"time"

	"github.com/influxdata/influxdb/kit/cli"
	influxlogger "github.com/influxdata/influxdb/logger"
	"github.com/influxdata/influxdb/prometheus"
	"github.com/influxdata/influxdb/telemetry"
	"go.uber.org/zap"
)

var (
	log  = influxlogger.New(os.Stdout)
	addr string
)

func main() {
	prog := &cli.Program{
		Run:  run,
		Name: "telemetryd",
		Opts: []cli.Opt{
			{
				DestP:   &addr,
				Flag:    "bind-addr",
				Default: ":8080",
				Desc:    "binding address for telemetry server",
			},
		},
	}
	cmd := cli.NewCommand(prog)

	var exitCode int
	if err := cmd.Execute(); err != nil {
		exitCode = 1
		log.Error("Command returned error", zap.Error(err))
	}

	if err := log.Sync(); err != nil {
		exitCode = 1
		fmt.Fprintf(os.Stderr, "Error syncing logs: %v\n", err)
	}
	time.Sleep(10 * time.Millisecond)
	os.Exit(exitCode)
}

func run() error {
	log := log.With(zap.String("service", "telemetryd"))
	store := telemetry.NewLogStore(log)
	svc := telemetry.NewPushGateway(log, store)
	// Print data as line protocol
	svc.Encoder = &prometheus.LineProtocol{}

	handler := http.HandlerFunc(svc.Handler)
	log.Info("Starting telemetryd server", zap.String("addr", addr))

	srv := http.Server{
		Addr:     addr,
		Handler:  handler,
		ErrorLog: zap.NewStdLog(log),
	}
	return srv.ListenAndServe()
}