2021-01-03 18:15:44 +05:30

98 lines
1.5 KiB
Go

package godruid
import (
"bytes"
"encoding/json"
"fmt"
"io/ioutil"
"net/http"
"time"
"go.uber.org/zap"
)
const (
DefaultEndPoint = "/druid/v2"
)
type Client struct {
Url string
EndPoint string
Timeout time.Duration
Debug bool
LastRequest string
LastResponse string
}
func (c *Client) Query(query Query) (err error) {
query.setup()
var reqJson []byte
if c.Debug {
reqJson, err = json.MarshalIndent(query, "", " ")
} else {
reqJson, err = json.Marshal(query)
}
if err != nil {
return
}
// log request made to druid
zap.S().Info(string(reqJson))
result, err := c.QueryRaw(reqJson)
if err != nil {
return
}
return query.onResponse(result)
}
func (c *Client) QueryRaw(req []byte) (result []byte, err error) {
if c.EndPoint == "" {
c.EndPoint = DefaultEndPoint
}
endPoint := c.EndPoint
if c.Debug {
endPoint += "?pretty"
c.LastRequest = string(req)
}
if err != nil {
return
}
// By default, use 60 second timeout unless specified otherwise
// by the caller
clientTimeout := 60 * time.Second
if c.Timeout != 0 {
clientTimeout = c.Timeout
}
httpClient := &http.Client{
Timeout: clientTimeout,
}
resp, err := httpClient.Post(c.Url+endPoint, "application/json", bytes.NewBuffer(req))
if err != nil {
return
}
defer func() {
resp.Body.Close()
}()
result, err = ioutil.ReadAll(resp.Body)
if err != nil {
return
}
if c.Debug {
c.LastResponse = string(result)
}
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("%s: %s", resp.Status, string(result))
}
return
}