package nsq import ( "fmt" "github.com/nsqio/go-nsq" "github.com/influxdata/telegraf" "github.com/influxdata/telegraf/plugins/outputs" "github.com/influxdata/telegraf/plugins/serializers" ) type NSQ struct { Server string Topic string producer *nsq.Producer serializer serializers.Serializer } var sampleConfig = ` ### Location of nsqd instance listening on TCP server = "localhost:4150" ### NSQ topic for producer messages topic = "telegraf" ### Data format to output. This can be "influx" or "graphite" ### Each data format has it's own unique set of configuration options, read ### more about them here: ### https://github.com/influxdata/telegraf/blob/master/docs/DATA_FORMATS_OUTPUT.md data_format = "influx" ` func (n *NSQ) SetSerializer(serializer serializers.Serializer) { n.serializer = serializer } func (n *NSQ) Connect() error { config := nsq.NewConfig() producer, err := nsq.NewProducer(n.Server, config) if err != nil { return err } n.producer = producer return nil } func (n *NSQ) Close() error { n.producer.Stop() return nil } func (n *NSQ) SampleConfig() string { return sampleConfig } func (n *NSQ) Description() string { return "Send telegraf measurements to NSQD" } func (n *NSQ) Write(metrics []telegraf.Metric) error { if len(metrics) == 0 { return nil } for _, metric := range metrics { values, err := n.serializer.Serialize(metric) if err != nil { return err } var pubErr error for _, value := range values { err = n.producer.Publish(n.Topic, []byte(value)) if err != nil { pubErr = err } } if pubErr != nil { return fmt.Errorf("FAILED to send NSQD message: %s", err) } } return nil } func init() { outputs.Add("nsq", func() telegraf.Output { return &NSQ{} }) }