forked from mrichman/hargo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
influxdb.go
103 lines (79 loc) · 1.93 KB
/
influxdb.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package hargo
import (
"fmt"
"net/url"
"strings"
"time"
log "github.com/Sirupsen/logrus"
client "github.com/influxdata/influxdb/client/v2"
)
var db string
// NewInfluxDBClient returns a new InfluxDB client
func NewInfluxDBClient(u url.URL) (client.Client, error) {
log.Print("Connecting to InfluxDB: ", u.String())
c, err := client.NewHTTPClient(client.HTTPConfig{
Addr: u.String(),
})
if err != nil {
log.Fatal("Error: ", err)
return c, err
}
db = strings.Replace(u.Path, "/", "", -1)
log.Info("DB: ", db)
cmd := fmt.Sprintf("CREATE DATABASE %s", db)
log.Debug("Query: ", cmd)
_, err = queryDB(c, cmd)
if err != nil {
log.Warn("Could not connect to InfluxDB: ", err)
return nil, err
}
return c, nil
}
// WritePoints inserts data to InfluxDB
func WritePoints(c client.Client, tr []TestResult) error {
bp, err := client.NewBatchPoints(client.BatchPointsConfig{
Database: db,
Precision: "us",
})
if err != nil {
log.Fatalln("Error: ", err)
}
//spew.Dump("BatchPoint:", bp)
for i := 0; i < len(tr); i++ {
// Create a point and add to batch
//tags := map[string]string{"test": "test-results"}
fields := map[string]interface{}{
"URL": tr[i].URL,
"Status": tr[i].Status,
"StartTime": tr[i].StartTime,
"EndTime": tr[i].EndTime,
"Latency": tr[i].Latency}
pt, err := client.NewPoint("test_result", nil, fields, time.Now())
if err != nil {
log.Fatalln("Error: ", err)
}
bp.AddPoint(pt)
}
// Write the batch
err = c.Write(bp)
if err != nil {
log.Fatalln("Error: ", err)
}
return nil
}
// queryDB convenience function to query the database
func queryDB(clnt client.Client, cmd string) (res []client.Result, _ error) {
q := client.Query{
Command: cmd,
Database: db,
}
if response, err := clnt.Query(q); err == nil {
if response.Error() != nil {
return res, response.Error()
}
res = response.Results
} else {
return res, err
}
return res, nil
}