Revert "bug fixes and refactoring"

This reverts commit 86961cc814.
This commit is contained in:
David G. Simmons 2017-11-03 13:28:35 -04:00
parent 8e587e74f5
commit 871fae6eb3
4 changed files with 91 additions and 89 deletions

View File

@ -2,40 +2,14 @@ package particle
import ( import (
"encoding/json" "encoding/json"
"github.com/gorilla/mux"
"github.com/influxdata/telegraf"
"io/ioutil"
"log" "log"
"net/http" "net/http"
"time" "time"
"github.com/gorilla/mux"
"github.com/influxdata/telegraf"
) )
type event struct {
Name string `json:"event"`
Data data `json:"data"`
TTL int `json:"ttl"`
PublishedAt string `json:"published_at"`
Database string `json:"influx_db"`
}
type data struct {
Tags map[string]string `json:"tags"`
Fields map[string]interface{} `json:"values"`
}
func newEvent() *event {
return &event{
Data: data{
Tags: make(map[string]string),
Fields: make(map[string]interface{}),
},
}
}
func (e *event) Time() (time.Time, error) {
return time.Parse("2006-01-02T15:04:05Z", e.PublishedAt)
}
type ParticleWebhook struct { type ParticleWebhook struct {
Path string Path string
acc telegraf.Accumulator acc telegraf.Accumulator
@ -49,19 +23,26 @@ func (rb *ParticleWebhook) Register(router *mux.Router, acc telegraf.Accumulator
func (rb *ParticleWebhook) eventHandler(w http.ResponseWriter, r *http.Request) { func (rb *ParticleWebhook) eventHandler(w http.ResponseWriter, r *http.Request) {
defer r.Body.Close() defer r.Body.Close()
e := newEvent() data, err := ioutil.ReadAll(r.Body)
if err := json.NewDecoder(r.Body).Decode(e); err != nil { if err != nil {
log.Println(err)
w.WriteHeader(http.StatusBadRequest) w.WriteHeader(http.StatusBadRequest)
return return
} }
dummy := &DummyData{}
pTime, err := e.Time() if err := json.Unmarshal(data, dummy); err != nil {
if err != nil { w.WriteHeader(http.StatusBadRequest)
pTime = time.Now() return
log.Printf("error parsing particle event time: %s. Using telegraf host time instead: %s", e.PublishedAt, pTime)
} }
pd := &ParticleData{}
rb.acc.AddFields(e.Name, e.Data.Fields, e.Data.Tags, pTime) if err := json.Unmarshal([]byte(dummy.Data), pd); err != nil {
w.WriteHeader(http.StatusBadRequest)
return
}
pTime, err := dummy.Time()
if err != nil {
log.Printf("Time Conversion Error")
pTime = time.Now()
}
rb.acc.AddFields(dummy.InfluxDB, pd.Fields, pd.Tags, pTime)
w.WriteHeader(http.StatusOK) w.WriteHeader(http.StatusOK)
} }

View File

@ -0,0 +1,22 @@
package particle
import (
"time"
)
type DummyData struct {
Event string `json:"event"`
Data string `json:"data"`
Ttl int `json:"ttl"`
PublishedAt string `json:"published_at"`
InfluxDB string `json:"influx_db"`
}
type ParticleData struct {
Event string `json:"event"`
Tags map[string]string `json:"tags"`
Fields map[string]interface{} `json:"values"`
}
func (d *DummyData) Time() (time.Time, error) {
return time.Parse("2006-01-02T15:04:05Z", d.PublishedAt)
}

View File

@ -0,0 +1,39 @@
package particle
func NewItemJSON() string {
return `
{
"event": "temperature",
"data": "{
"tags": {
"id": "230035001147343438323536",
"location": "TravelingWilbury"
},
"values": {
"temp_c": 26.680000,
"temp_f": 80.024001,
"humidity": 44.937500,
"pressure": 998.998901,
"altitude": 119.331436,
"broadband": 1266,
"infrared": 528,
"lux": 0
}
}",
"ttl": 60,
"published_at": "2017-09-28T21:54:10.897Z",
"coreid": "123456789938323536",
"userid": "1234ee123ac8e5ec1231a123d",
"version": 10,
"public": false,
"productID": 1234,
"name": "sensor"
"influx_db": "mydata"
}`
}
func UnknowJSON() string {
return `
{
"event": "roger"
}`
}

View File

@ -1,16 +1,17 @@
package particle package particle
import ( import (
"github.com/influxdata/telegraf/testutil"
"log"
"net/http" "net/http"
"net/http/httptest" "net/http/httptest"
"strings" "strings"
"testing" "testing"
"github.com/influxdata/telegraf/testutil"
) )
func postWebhooks(rb *ParticleWebhook, eventBody string) *httptest.ResponseRecorder { func postWebhooks(rb *ParticleWebhook, eventBody string) *httptest.ResponseRecorder {
req, _ := http.NewRequest("POST", "/", strings.NewReader(eventBody)) req, _ := http.NewRequest("POST", "/", strings.NewReader(eventBody))
log.Printf("eventBody: %s\n", eventBody)
w := httptest.NewRecorder() w := httptest.NewRecorder()
w.Code = 500 w.Code = 500
@ -20,10 +21,10 @@ func postWebhooks(rb *ParticleWebhook, eventBody string) *httptest.ResponseRecor
} }
func TestNewItem(t *testing.T) { func TestNewItem(t *testing.T) {
t.Parallel()
var acc testutil.Accumulator var acc testutil.Accumulator
rb := &ParticleWebhook{Path: "/particle", acc: &acc} rb := &ParticleWebhook{Path: "/particle", acc: &acc}
resp := postWebhooks(rb, NewItemJSON()) resp := postWebhooks(rb, NewItemJSON())
log.Printf("Respnse: %s\n", resp.Body)
if resp.Code != http.StatusOK { if resp.Code != http.StatusOK {
t.Errorf("POST new_item returned HTTP status code %v.\nExpected %v", resp.Code, http.StatusOK) t.Errorf("POST new_item returned HTTP status code %v.\nExpected %v", resp.Code, http.StatusOK)
} }
@ -31,12 +32,12 @@ func TestNewItem(t *testing.T) {
fields := map[string]interface{}{ fields := map[string]interface{}{
"temp_c": 26.680000, "temp_c": 26.680000,
"temp_f": 80.024001, "temp_f": 80.024001,
"infrared": 528.0,
"lux": 0.0,
"humidity": 44.937500, "humidity": 44.937500,
"pressure": 998.998901, "pressure": 998.998901,
"altitude": 119.331436, "altitude": 119.331436,
"broadband": 1266.0, "broadband": 1266,
"infrared": 528,
"lux": 0,
} }
tags := map[string]string{ tags := map[string]string{
@ -44,54 +45,13 @@ func TestNewItem(t *testing.T) {
"location": "TravelingWilbury", "location": "TravelingWilbury",
} }
acc.AssertContainsTaggedFields(t, "temperature", fields, tags) acc.AssertContainsTaggedFields(t, "particle_webhooks", fields, tags)
} }
func TestUnknowItem(t *testing.T) { func TestUnknowItem(t *testing.T) {
t.Parallel() rb := &ParticleWebhook{Path: "/particle"}
var acc testutil.Accumulator
rb := &ParticleWebhook{Path: "/particle", acc: &acc}
resp := postWebhooks(rb, UnknowJSON()) resp := postWebhooks(rb, UnknowJSON())
log.Printf("Response: %s\n", resp.Body)
if resp.Code != http.StatusOK { if resp.Code != http.StatusOK {
t.Errorf("POST unknown returned HTTP status code %v.\nExpected %v", resp.Code, http.StatusOK) t.Errorf("POST unknown returned HTTP status code %v.\nExpected %v", resp.Code, http.StatusOK)
} }
} }
func NewItemJSON() string {
return `
{
"event": "temperature",
"data": {
"tags": {
"id": "230035001147343438323536",
"location": "TravelingWilbury"
},
"values": {
"temp_c": 26.680000,
"temp_f": 80.024001,
"humidity": 44.937500,
"pressure": 998.998901,
"altitude": 119.331436,
"broadband": 1266,
"infrared": 528,
"lux": 0
}
},
"ttl": 60,
"published_at": "2017-09-28T21:54:10.897Z",
"coreid": "123456789938323536",
"userid": "1234ee123ac8e5ec1231a123d",
"version": 10,
"public": false,
"productID": 1234,
"name": "sensor",
"influx_db": "mydata"
}`
}
func UnknowJSON() string {
return `
{
"event": "roger"
}`
}