mirror of
https://github.com/lukaszraczylo/go-simple-graphql.git
synced 2026-10-14 00:58:32 +00:00
86 lines
2.5 KiB
Go
86 lines
2.5 KiB
Go
// Package / library or rather wrapper for GraphQL queries execution in the painless way.
|
|
// Using it is as easy as copy / paste the query itself and set appropriate variables in.
|
|
//
|
|
// Library supports basic error reporting on unsuccessful queries and setting appropriate headers.
|
|
package gql
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/lukaszraczylo/zero"
|
|
"github.com/tidwall/gjson"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
// Endpoint of your GraphQL server to query
|
|
// this variable can be overwritten by setting env variable, for example:
|
|
// GRAPHQL_ENDPOINT=http://hasura.local/v1/graphql
|
|
var GraphQLUrl = "http://127.0.0.1:9090/v1/graphql"
|
|
|
|
type requestBase struct {
|
|
Query string `json:"query"`
|
|
Variables interface{} `json:"variables"`
|
|
}
|
|
|
|
func init() {
|
|
if !zero.IsZero(os.Getenv("GRAPHQL_ENDPOINT")) {
|
|
GraphQLUrl = os.Getenv("GRAPHQL_ENDPOINT")
|
|
}
|
|
}
|
|
|
|
// queryBuilder takes query data (string) and variables (interface) as a parameter and assembles
|
|
// graphQL query. It also compacts the JSON result. Function returns query as []byte and error if anything went wrong.
|
|
func queryBuilder(data string, variables interface{}) ([]byte, error) {
|
|
var err error
|
|
var qb = &requestBase{
|
|
Query: data,
|
|
Variables: variables,
|
|
}
|
|
j := new(bytes.Buffer)
|
|
j2, err := json.Marshal(qb)
|
|
if err != nil {
|
|
return []byte{}, err
|
|
}
|
|
if err = json.Compact(j, j2); err != nil {
|
|
return []byte{}, err
|
|
}
|
|
return j.Bytes(), err
|
|
}
|
|
|
|
// Query allows you to execute the GraphQL query.
|
|
// Query is a string ( copy paste from Hasura or any other query builder )
|
|
// Variables and Headers are maps of strings ( see the example )
|
|
// Function returns whatever specified query returns and/or error.
|
|
func Query(query string, variables interface{}, headers map[string]interface{}) (string, error) {
|
|
var err error
|
|
readyQuery, err := queryBuilder(query, variables)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
req := fasthttp.AcquireRequest()
|
|
defer fasthttp.ReleaseRequest(req)
|
|
req.Header.SetContentType("application/json")
|
|
for header, value := range headers {
|
|
req.Header.Set(fmt.Sprintf("%v", header), fmt.Sprintf("%v", value))
|
|
}
|
|
req.Header.SetMethodBytes([]byte("POST"))
|
|
req.SetBody(readyQuery)
|
|
req.SetRequestURI(GraphQLUrl)
|
|
res := fasthttp.AcquireResponse()
|
|
defer fasthttp.ReleaseResponse(res)
|
|
if err := fasthttp.Do(req, res); err != nil {
|
|
return "", err
|
|
}
|
|
body := res.Body()
|
|
toReturn := gjson.Get(string(body), "data")
|
|
if toReturn.String() == "" {
|
|
err = errors.New(string(body))
|
|
return "", err
|
|
}
|
|
return toReturn.String(), err
|
|
}
|