Go语言 API-立哥开发

//Copy right 2020 Jacky Zong . All  rights  reserved .

package server

import (

	"encoding/json"

	“mySQL”

）

// Server will hold connection to the db as well as handlers

type Server struct {

	GqlSchema *graphql.Schema

}

type reqBody struct {

	Query string json:"query"

	}

// GraphQL returns an http.HandlerFunc for our /graphql endpoint

func (s *Server) GraphQL() http.HandlerFunc {

	return func(w http.ResponseWriter, r *http.Request) {

		// Check to ensure query was provided in the request body

		if r.Body == nil {

			http.Error(w, "Must provide graphql query in request body", 400)

			return

		}

		var rBody reqBody

		// Decode the request body into rBody

		err := json.NewDecoder(r.Body).Decode(&rBody)

		if err != nil {

			http.Error(w, "Error parsing JSON request body", 400)

		}

		// Execute graphql query

		result := gql.ExecuteQuery(rBody.Query, *s.GqlSchema)

		// render.JSON comes from the chi/render package and handles

		// marshalling to json, automatically escaping HTML and setting

		// the Content-Type as application/json.

		render.JSON(w, r, result)

	}

}