package graphite import ( "errors" "fmt" "github.com/influxdb/influxdb/client/v2" "github.com/influxdb/telegraf/plugins/outputs" "log" "math/rand" "net" "strings" "time" ) type Graphite struct { // URL is only for backwards compatability Servers []string Prefix string Timeout int conns []net.Conn } var sampleConfig = ` # TCP endpoint for your graphite instance. servers = ["localhost:2003"] # Prefix metrics name prefix = "" # timeout in seconds for the write connection to graphite timeout = 2 ` func (g *Graphite) Connect() error { // Set default values if g.Timeout <= 0 { g.Timeout = 2 } if len(g.Servers) == 0 { g.Servers = append(g.Servers, "localhost:2003") } // Get Connections var conns []net.Conn for _, server := range g.Servers { conn, err := net.DialTimeout("tcp", server, time.Duration(g.Timeout)*time.Second) if err == nil { conns = append(conns, conn) } } g.conns = conns return nil } func (g *Graphite) Close() error { // Closing all connections for _, conn := range g.conns { conn.Close() } return nil } func (g *Graphite) SampleConfig() string { return sampleConfig } func (g *Graphite) Description() string { return "Configuration for Graphite server to send metrics to" } // Choose a random server in the cluster to write to until a successful write // occurs, logging each unsuccessful. If all servers fail, return error. func (g *Graphite) Write(points []*client.Point) error { // Prepare data var bp []string for _, point := range points { // Get name name := point.Name() // Convert UnixNano to Unix timestamps timestamp := point.UnixNano() / 1000000000 for field_name, value := range point.Fields() { // Convert value value_str := fmt.Sprintf("%#v", value) // Write graphite point var graphitePoint string if name == field_name { graphitePoint = fmt.Sprintf("%s.%s %s %d\n", strings.Replace(point.Tags()["host"], ".", "_", -1), strings.Replace(name, ".", "_", -1), value_str, timestamp) } else { graphitePoint = fmt.Sprintf("%s.%s.%s %s %d\n", strings.Replace(point.Tags()["host"], ".", "_", -1), strings.Replace(name, ".", "_", -1), strings.Replace(field_name, ".", "_", -1), value_str, timestamp) } if g.Prefix != "" { graphitePoint = fmt.Sprintf("%s.%s", g.Prefix, graphitePoint) } bp = append(bp, graphitePoint) //fmt.Printf(graphitePoint) } } graphitePoints := strings.Join(bp, "") // This will get set to nil if a successful write occurs err := errors.New("Could not write to any Graphite server in cluster\n") // Send data to a random server p := rand.Perm(len(g.conns)) for _, n := range p { if _, e := fmt.Fprintf(g.conns[n], graphitePoints); e != nil { // Error log.Println("ERROR: " + err.Error()) // Let's try the next one } else { // Success err = nil break } } // try to reconnect if err != nil { g.Connect() } return err } func init() { outputs.Add("graphite", func() outputs.Output { return &Graphite{} }) }