telegraf/services/enterprise/enterprise.go

package enterprise

import (
	"log"
	"os"

	"github.com/influxdata/enterprise-client/v2"
)

type Config struct {
	Hosts []*client.Host
}

type Service struct {
	hosts    []*client.Host
	logger   *log.Logger
	hostname string
}

func NewEnterprise(c Config, hostname string) *Service {
	return &Service{
		hosts:    c.Hosts,
		hostname: hostname,
		logger:   log.New(os.Stdout, "[enterprise]", log.Ldate|log.Ltime),
	}
}

func (s *Service) Open() {
	cl, err := client.New(s.hosts)
	if err != nil {
		s.logger.Printf("Unable to contact one or more Enterprise hosts. err: %s", err.Error())
		return
	}
	go s.registerProduct(cl)
}

func (s *Service) registerProduct(cl *client.Client) {
	p := client.Product{
		ProductID: "telegraf",
		Host:      s.hostname,
	}

	_, err := cl.Register(&p)
	if err != nil {
		s.logger.Println("Unable to register Telegraf with Enterprise")
	}
}