feat:add query func and query parameter struct for influxdb

This commit is contained in:
douxu 2024-08-02 16:16:49 +08:00
parent 500eeaf339
commit 5e42854213
2 changed files with 51 additions and 0 deletions

View File

@ -7,6 +7,7 @@ import (
influxdb2 "github.com/influxdata/influxdb-client-go/v2"
"github.com/influxdata/influxdb-client-go/v2/api"
"go.uber.org/zap"
)
// WriterPointIntoInfluxDB return the result of storing point into influxdb
@ -16,3 +17,30 @@ func WriterPointIntoInfluxDB(ctx context.Context, writeAPI api.WriteAPIBlocking,
point := influxdb2.NewPoint(measurement, tags, fields, time.Now())
return writeAPI.WritePoint(ctx, point)
}
// QueryPointFromInfluxDB return the result of query from influxdb by special parameters
// Get query client
// queryAPI := client.QueryAPI(org)
func QueryPointFromInfluxDB(ctx context.Context, queryAPI api.QueryAPI, parameters InfluxDBPara, logger *zap.Logger) ([]interface{}, error) {
dataList := make([]interface{}, 0)
// get query result
results, err := queryAPI.QueryWithParams(ctx, query, parameters)
if err != nil {
logger.Error("query data by parameters failed", zap.Error(err), zap.String("para", parameters.String()))
return nil, err
}
for results.Next() {
if results.TableChanged() {
logger.Info("find new result table", zap.String("table_metadata", results.TableMetadata().String()))
continue
}
dataList = append(dataList, results.Record().Value())
}
if results.Err() != nil {
logger.Error("query parsing error failed", zap.Error(results.Err()))
return nil, results.Err()
}
return nil, nil
}

View File

@ -0,0 +1,23 @@
// Package database define database operation functions
package database
import "fmt"
var query = `from(bucket:params.bucket)
|> range(start: duration(params.start))
|> filter(fn: (r) => r._measurement == parames.measurement)
|> filter(fn: (r) => r._field == params.field)
|> filter(fn: (r) => r._value > params.value)`
// InfluxDBPara define struct of influxdb query parameters
type InfluxDBPara struct {
Bucket string `json:"bucket"`
Measurement string `json:"measurement"`
Start string `json:"start"`
Field string `json:"field"`
Value float64 `json:"value"`
}
func (i InfluxDBPara) String() string {
return fmt.Sprintf("bucket:%s, measurement:%s, start:%s, field:%s, value:%f", i.Bucket, i.Measurement, i.Start, i.Field, i.Value)
}