closes #1289 Signed-off-by: François de Metz <francois@stormz.me> Signed-off-by: Cyril Duez <cyril@stormz.me> Rename internals struct. Signed-off-by: François de Metz <francois@stormz.me> Signed-off-by: Cyril Duez <cyril@stormz.me> Update changelog. Signed-off-by: François de Metz <francois@stormz.me> Signed-off-by: Cyril Duez <cyril@stormz.me> Update READMEs and CHANGELOG. Signed-off-by: François de Metz <francois@stormz.me> Signed-off-by: Cyril Duez <cyril@stormz.me> Update SampleConfig. Update the config format. Update telegraf config. Update the webhooks README. Update changelog. Update the changelog with an upgrade path. Update default ports. Fix indent. Check for nil value on AvailableWebhooks. Check for CanInterface.
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package rollbar
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/influxdata/telegraf"
|
|
)
|
|
|
|
type RollbarWebhook struct {
|
|
Path string
|
|
acc telegraf.Accumulator
|
|
}
|
|
|
|
func (rb *RollbarWebhook) Register(router *mux.Router, acc telegraf.Accumulator) {
|
|
router.HandleFunc(rb.Path, rb.eventHandler).Methods("POST")
|
|
log.Printf("Started the webhooks_rollbar on %s\n", rb.Path)
|
|
rb.acc = acc
|
|
}
|
|
|
|
func (rb *RollbarWebhook) eventHandler(w http.ResponseWriter, r *http.Request) {
|
|
defer r.Body.Close()
|
|
data, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
dummyEvent := &DummyEvent{}
|
|
err = json.Unmarshal(data, dummyEvent)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
event, err := NewEvent(dummyEvent, data)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusOK)
|
|
return
|
|
}
|
|
|
|
rb.acc.AddFields("rollbar_webhooks", event.Fields(), event.Tags(), time.Now())
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func generateEvent(event Event, data []byte) (Event, error) {
|
|
err := json.Unmarshal(data, event)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return event, nil
|
|
}
|
|
|
|
func NewEvent(dummyEvent *DummyEvent, data []byte) (Event, error) {
|
|
switch dummyEvent.EventName {
|
|
case "new_item":
|
|
return generateEvent(&NewItem{}, data)
|
|
case "deploy":
|
|
return generateEvent(&Deploy{}, data)
|
|
default:
|
|
return nil, errors.New("Not implemented type: " + dummyEvent.EventName)
|
|
}
|
|
}
|