chronograf/influx/influx.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
}