Update rollbar webhook plugin.

This commit is contained in:
François de Metz 2016-05-25 19:38:44 +02:00 committed by Cyril Duez
parent 2221344028
commit b09f7c89c2
1 changed files with 6 additions and 20 deletions

View File

@ -3,7 +3,6 @@ package rollbar_webhooks
import (
"encoding/json"
"errors"
"fmt"
"io/ioutil"
"log"
"net/http"
@ -20,7 +19,7 @@ func init() {
}
type RollbarWebhooks struct {
ServiceAddress string
Path string
// Lock for the struct
sync.Mutex
// Events buffer to store events between Gather calls
@ -33,8 +32,8 @@ func NewRollbarWebhooks() *RollbarWebhooks {
func (rb *RollbarWebhooks) SampleConfig() string {
return `
## Address and port to host Webhook listener on
service_address = ":1619"
## Path of the global server
path = "/rollbar"
`
}
@ -52,25 +51,12 @@ func (rb *RollbarWebhooks) Gather(acc telegraf.Accumulator) error {
return nil
}
func (rb *RollbarWebhooks) Listen() {
r := mux.NewRouter()
r.HandleFunc("/", rb.eventHandler).Methods("POST")
err := http.ListenAndServe(fmt.Sprintf("%s", rb.ServiceAddress), r)
if err != nil {
log.Printf("Error starting server: %v", err)
}
}
func (rb *RollbarWebhooks) Start(_ telegraf.Accumulator) error {
go rb.Listen()
log.Printf("Started the rollbar_webhooks service on %s\n", rb.ServiceAddress)
func (rb *RollbarWebhooks) Register(r *mux.Router, _ telegraf.Accumulator) error {
r.HandleFunc(rb.Path, rb.eventHandler).Methods("POST")
log.Printf("Register the rollbar_webhooks router on %s\n", rb.Path)
return nil
}
func (rb *RollbarWebhooks) Stop() {
log.Println("Stopping the rbWebhooks service")
}
func (rb *RollbarWebhooks) eventHandler(w http.ResponseWriter, r *http.Request) {
defer r.Body.Close()
data, err := ioutil.ReadAll(r.Body)