51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
|
package gather
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
|
||
|
|
"github.com/influxdata/platform"
|
||
|
|
"github.com/influxdata/platform/nats"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
// handler implents nats Handler interface.
|
||
|
|
type handler struct {
|
||
|
|
Scraper Scraper
|
||
|
|
Publisher nats.Publisher
|
||
|
|
Logger *zap.Logger
|
||
|
|
}
|
||
|
|
|
||
|
|
// Process consumes scraper target from scraper target queue,
|
||
|
|
// call the scraper to gather, and publish to metrics queue.
|
||
|
|
func (h *handler) Process(s nats.Subscription, m nats.Message) {
|
||
|
|
defer m.Ack()
|
||
|
|
|
||
|
|
req := new(platform.ScraperTarget)
|
||
|
|
err := json.Unmarshal(m.Data(), req)
|
||
|
|
if err != nil {
|
||
|
|
h.Logger.Error("unable to unmarshal json", zap.Error(err))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
ms, err := h.Scraper.Gather(context.TODO(), *req)
|
||
|
|
if err != nil {
|
||
|
|
h.Logger.Error("unable to gather", zap.Error(err))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// send metrics to storage queue
|
||
|
|
buf := new(bytes.Buffer)
|
||
|
|
if err := json.NewEncoder(buf).Encode(ms); err != nil {
|
||
|
|
h.Logger.Error("unable to marshal json", zap.Error(err))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := h.Publisher.Publish(MetricsSubject, buf); err != nil {
|
||
|
|
h.Logger.Error("unable to publish scraper metrics", zap.Error(err))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|