Stephanie Closson
53ee72d15d
* Sent resource calls for metadata to the backend * moved resource calls to the backend * code review feedback * fixed post with body * statuscode >= 300 * cleanup * fixed tests * fixed datasource tests * code review feedback * force some other endpoints to only GET * fix linting errors * fixed tests * was able to remove section of redundant code * cleanup and code review feedback * moved query_exemplars to get request * fixed return on error * went back to resource calls, but using the backendsrv directly * moved to a resource call with fallback * fixed tests * check for proper messages * proper check for invalid calls * code review changes
115 lines
2.5 KiB
Go
115 lines
2.5 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"path"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/tsdb/prometheus/models"
|
|
)
|
|
|
|
type doer interface {
|
|
Do(req *http.Request) (*http.Response, error)
|
|
}
|
|
|
|
type Client struct {
|
|
doer doer
|
|
method string
|
|
baseUrl string
|
|
}
|
|
|
|
func NewClient(d doer, method, baseUrl string) *Client {
|
|
return &Client{doer: d, method: method, baseUrl: baseUrl}
|
|
}
|
|
|
|
func (c *Client) QueryRange(ctx context.Context, q *models.Query) (*http.Response, error) {
|
|
u, err := url.ParseRequestURI(c.baseUrl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
u.Path = path.Join(u.Path, "api/v1/query_range")
|
|
|
|
qs := u.Query()
|
|
qs.Set("query", q.Expr)
|
|
tr := q.TimeRange()
|
|
qs.Set("start", formatTime(tr.Start))
|
|
qs.Set("end", formatTime(tr.End))
|
|
qs.Set("step", strconv.FormatFloat(tr.Step.Seconds(), 'f', -1, 64))
|
|
|
|
return c.fetch(ctx, c.method, u, qs)
|
|
}
|
|
|
|
func (c *Client) QueryInstant(ctx context.Context, q *models.Query) (*http.Response, error) {
|
|
u, err := url.ParseRequestURI(c.baseUrl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
u.Path = path.Join(u.Path, "api/v1/query")
|
|
|
|
qs := u.Query()
|
|
qs.Set("query", q.Expr)
|
|
tr := q.TimeRange()
|
|
if !tr.End.IsZero() {
|
|
qs.Set("time", formatTime(tr.End))
|
|
}
|
|
|
|
return c.fetch(ctx, c.method, u, qs)
|
|
}
|
|
|
|
func (c *Client) QueryExemplars(ctx context.Context, q *models.Query) (*http.Response, error) {
|
|
u, err := url.ParseRequestURI(c.baseUrl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
u.Path = path.Join(u.Path, "api/v1/query_exemplars")
|
|
|
|
qs := u.Query()
|
|
tr := q.TimeRange()
|
|
qs.Set("query", q.Expr)
|
|
qs.Set("start", formatTime(tr.Start))
|
|
qs.Set("end", formatTime(tr.End))
|
|
|
|
return c.fetch(ctx, c.method, u, qs)
|
|
}
|
|
|
|
func (c *Client) QueryResource(ctx context.Context, method string, p string, qs url.Values) (*http.Response, error) {
|
|
u, err := url.ParseRequestURI(c.baseUrl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
u.Path = path.Join(u.Path, p)
|
|
|
|
return c.fetch(ctx, method, u, qs)
|
|
}
|
|
|
|
func (c *Client) fetch(ctx context.Context, method string, u *url.URL, qs url.Values) (*http.Response, error) {
|
|
if strings.ToUpper(method) == http.MethodGet {
|
|
u.RawQuery = qs.Encode()
|
|
}
|
|
|
|
r, err := http.NewRequestWithContext(ctx, method, u.String(), nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if strings.ToUpper(method) == http.MethodPost {
|
|
r.Body = ioutil.NopCloser(strings.NewReader(qs.Encode()))
|
|
r.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
|
}
|
|
|
|
return c.doer.Do(r)
|
|
}
|
|
|
|
func formatTime(t time.Time) string {
|
|
return strconv.FormatFloat(float64(t.Unix())+float64(t.Nanosecond())/1e9, 'f', -1, 64)
|
|
}
|