68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
package influx
|
|
|
|
import (
|
|
"log"
|
|
"time"
|
|
|
|
client "github.com/influxdata/influxdb/client/v2"
|
|
"github.com/spf13/viper"
|
|
"myschools.me/wyh/influx-demo.git/model"
|
|
)
|
|
|
|
//ConnInflux 连接influxDB
|
|
func ConnInflux() client.Client {
|
|
cli, err := client.NewHTTPClient(client.HTTPConfig{
|
|
Addr: viper.GetString("influx.Addr"),
|
|
Username: viper.GetString("influx.UserName"),
|
|
Password: viper.GetString("influx.Password"),
|
|
})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
return cli
|
|
}
|
|
|
|
//WritesPoints 写入数据
|
|
func WritesPoints(points *model.Influx, cli client.Client) {
|
|
bp, err := client.NewBatchPoints(client.BatchPointsConfig{
|
|
Database: viper.GetString("influx.DbName"),
|
|
Precision: "s",
|
|
})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
pt, err := client.NewPoint(
|
|
// "cpu_usage",
|
|
points.TableName,
|
|
points.Tags,
|
|
points.Fields,
|
|
time.Now(),
|
|
)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
bp.AddPoint(pt)
|
|
|
|
if err := cli.Write(bp); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
//QueryDB 查询数据库
|
|
func QueryDB(cli client.Client, cmd string) (res []client.Result, err error) {
|
|
q := client.Query{
|
|
Command: cmd,
|
|
Database: viper.GetString("influx.DbName"),
|
|
}
|
|
if response, err := cli.Query(q); err == nil {
|
|
if response.Error() != nil {
|
|
return res, response.Error()
|
|
}
|
|
res = response.Results
|
|
} else {
|
|
return res, err
|
|
}
|
|
return res, nil
|
|
}
|