You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
package internal
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/fullstorydev/grpcurl"
|
|
"github.com/golang/protobuf/jsonpb"
|
|
"github.com/zeromicro/go-zero/rest/httpx"
|
|
"github.com/zeromicro/go-zero/rest/pathvar"
|
|
)
|
|
|
|
// NewRequestParser creates a new request parser from the given http.Request and resolver.
|
|
func NewRequestParser(r *http.Request, resolver jsonpb.AnyResolver) (grpcurl.RequestParser, error) {
|
|
vars := pathvar.Vars(r)
|
|
params, err := httpx.GetFormValues(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for k, v := range vars {
|
|
params[k] = v
|
|
}
|
|
|
|
body, ok := getBody(r)
|
|
if !ok {
|
|
return buildJsonRequestParser(params, resolver)
|
|
}
|
|
|
|
if len(params) == 0 {
|
|
return grpcurl.NewJSONRequestParser(body, resolver), nil
|
|
}
|
|
|
|
m := make(map[string]interface{})
|
|
if err := json.NewDecoder(body).Decode(&m); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for k, v := range params {
|
|
m[k] = v
|
|
}
|
|
|
|
return buildJsonRequestParser(m, resolver)
|
|
}
|
|
|
|
func buildJsonRequestParser(m map[string]interface{}, resolver jsonpb.AnyResolver) (
|
|
grpcurl.RequestParser, error) {
|
|
var buf bytes.Buffer
|
|
if err := json.NewEncoder(&buf).Encode(m); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return grpcurl.NewJSONRequestParser(&buf, resolver), nil
|
|
}
|
|
|
|
func getBody(r *http.Request) (io.Reader, bool) {
|
|
if r.Body == nil {
|
|
return nil, false
|
|
}
|
|
|
|
if r.ContentLength == 0 {
|
|
return nil, false
|
|
}
|
|
|
|
if r.ContentLength > 0 {
|
|
return r.Body, true
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
if _, err := io.Copy(&buf, r.Body); err != nil {
|
|
return nil, false
|
|
}
|
|
|
|
if buf.Len() > 0 {
|
|
return &buf, true
|
|
}
|
|
|
|
return nil, false
|
|
}
|