-
Notifications
You must be signed in to change notification settings - Fork 0
/
influxdb.go
59 lines (50 loc) · 1.1 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
package main
import (
"log"
"github.com/influxdata/influxdb/client/v2"
)
var influxDBChan = make(chan *client.Point, 10)
var influxRevciver *InfluxDBReciver
type InfluxDBReciver struct {
address string
cache int
}
func NewInfluxDBReciver(addr string, cache int) *InfluxDBReciver {
return &InfluxDBReciver{address: addr, cache: cache}
}
func (r *InfluxDBReciver) AddBatchPoint(ps []*client.Point) error {
bp, err := client.NewBatchPoints(client.BatchPointsConfig{
Database: influxDatabase,
})
if err != nil {
log.Fatal(err)
}
bp.AddPoints(ps)
client, err := client.NewHTTPClient(client.HTTPConfig{
Addr: r.address,
})
if err != nil {
log.Fatalf("Faile to NewHTTPClient: %s", err.Error())
}
defer client.Close()
return client.Write(bp)
}
func (r *InfluxDBReciver) ListenTask() {
go func() {
var (
n = 0
points = make([]*client.Point, r.cache)
)
for {
points[n] = <-influxDBChan
if n == r.cache-1 {
if err := influxRevciver.AddBatchPoint(points); err != nil {
log.Printf("Failed AddBatchPoint: %s", err.Error())
}
n = 0
continue
}
n++
}
}()
}