57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
package influx
|
|
|
|
import (
|
|
ixClient "github.com/influxdata/influxdb/client/v2"
|
|
"github.com/influxdata/mrfusion"
|
|
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
// Client is a device for retrieving time series data from an InfluxDB instance
|
|
type Client struct {
|
|
ix ixClient.Client
|
|
}
|
|
|
|
// NewClient initializes an HTTP Client for InfluxDB. UDP, although supported
|
|
// for querying InfluxDB, is not supported here to remove the need to
|
|
// explicitly Close the client.
|
|
func NewClient(host string) (*Client, error) {
|
|
cl, err := ixClient.NewHTTPClient(ixClient.HTTPConfig{
|
|
Addr: host,
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Client{
|
|
ix: cl,
|
|
}, nil
|
|
}
|
|
|
|
// Query issues a request to a configured InfluxDB instance for time series
|
|
// information specified by query. Queries must be "fully-qualified," and
|
|
// include both the database and retention policy. In-flight requests can be
|
|
// cancelled using the provided context.
|
|
func (c *Client) Query(ctx context.Context, query mrfusion.Query) (mrfusion.Response, error) {
|
|
q := ixClient.NewQuery(query.Command, query.DB, query.RP)
|
|
resps := make(chan (response))
|
|
go func() {
|
|
resp, err := c.ix.Query(q)
|
|
resps <- response{resp, err}
|
|
}()
|
|
|
|
select {
|
|
case resp := <-resps:
|
|
return resp, resp.err
|
|
case <-ctx.Done():
|
|
return nil, mrfusion.ErrUpstreamTimeout
|
|
}
|
|
}
|
|
|
|
// MonitoredServices returns all services for which this instance of InfluxDB
|
|
// has time series information stored for.
|
|
func (c *Client) MonitoredServices(ctx context.Context) ([]mrfusion.MonitoredService, error) {
|
|
return []mrfusion.MonitoredService{}, nil
|
|
}
|