query.go

package druid

import (
	"net/http"

	"github.com/h2oai/go-druid/builder/query"

	"github.com/h2oai/go-druid/builder"
)

const (
	NativeQueryEndpoint = "druid/v2"
	SQLQueryEndpoint    = "druid/v2/sql"
)

type QueryService struct {
	client *Client
}

func (q *QueryService) Execute(qry builder.Query, result interface{}, headers ...http.Header) (*Response, error) {
	var path string
	switch qry.Type() {
	case "sql":
		path = SQLQueryEndpoint
	default:
		path = NativeQueryEndpoint
	}
	r, err := q.client.NewRequest("POST", path, qry)
	if err != nil {
		return nil, err
	}
	if len(headers) >= 1 {
		for k, v := range headers[0] {
			for _, vv := range v {
				r.Header.Set(k, vv)
			}
		}
	}
	resp, err := q.client.Do(r, result)
	if err != nil {
		return nil, err
	}
	return resp, nil
}

//func (q *QueryService) Cancel(query builder.Query) () {}

//func (q *QueryService) Candidates(query builder.Query, result interface{}) (*Response, error) {}

func (q *QueryService) Load(data []byte) (builder.Query, error) {
	return query.Load(data)
}