sigmasternchen/iot-relay
1
0
Fork 0
mirror of https://github.com/sigmasternchen/iot-relay synced 2025-03-15 06:28:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixed multi value problem

Browse source
This commit is contained in:
overflowerror 2021-11-14 14:16:31 +01:00
parent 6ac99dc2af
commit c8dc8a1338
1 changed files with 31 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
internal/client/client.go
View file

@ -11,7 +11,7 @@ import (
"time"
)
func lineProtocol(request types.Request, config config.Config) string {
func lineProtocolPrefix(request types.Request, config config.Config) string {
var builder strings.Builder
builder.WriteString(config.Client.Measurement)
builder.WriteString(",host=")
@ -24,40 +24,47 @@ func lineProtocol(request types.Request, config config.Config) string {
builder.WriteString(request.Location)
builder.WriteString(" ")
for key, value := range request.Data {
builder.WriteString(key)
builder.WriteString("=")
builder.WriteString(value)
builder.WriteString(" ")
}
builder.WriteString(strconv.FormatInt(time.Now().UnixNano(), 10))
return builder.String()
}
func GetHandler(config config.Config) types.Callback {
return func(request types.Request) error {
line := lineProtocol(request, config)
prefix := lineProtocolPrefix(request, config)
timeString := strconv.FormatInt(time.Now().UnixNano(), 10)
url := config.Client.Address + "/write?db=" + config.Client.DB
client := &http.Client{}
req, err := http.NewRequest("POST", url, strings.NewReader(line))
if len(config.Client.Username) != 0 && len(config.Client.Password) != 0 {
req.SetBasicAuth(config.Client.Username, config.Client.Password)
}
var builder strings.Builder
resp, err := client.Do(req)
if err != nil {
log.Println(err)
return err
}
if resp.StatusCode != 204 {
err = fmt.Errorf("influxdb responded with %d", resp.StatusCode)
log.Println(err)
return err
for key, value := range request.Data {
builder.Reset()
builder.WriteString(prefix)
builder.WriteString(key)
builder.WriteString("=")
builder.WriteString(value)
builder.WriteString(" ")
builder.WriteString(timeString)
req, err := http.NewRequest("POST", url, strings.NewReader(builder.String()))
if len(config.Client.Username) != 0 && len(config.Client.Password) != 0 {
req.SetBasicAuth(config.Client.Username, config.Client.Password)
}
resp, err := client.Do(req)
if err != nil {
log.Println(err)
return err
}
if resp.StatusCode != 204 {
err = fmt.Errorf("influxdb responded with %d", resp.StatusCode)
log.Println(err)
return err
}
}
return nil