Delete references to json.RawMessage (#12)

This commit is contained in:
Peter Edge 2018-02-20 21:03:05 +01:00 committed by Joshua Humphries
parent 620796528e
commit ceba64b971
3 changed files with 6 additions and 8 deletions

View File

@ -427,7 +427,7 @@ func (*handler) OnSendHeaders(md metadata.MD) {
}
}
func (h *handler) getRequestData() (json.RawMessage, error) {
func (h *handler) getRequestData() ([]byte, error) {
// we don't use a mutex, though this methods will be called from different goroutine
// than other methods for bidi calls, because this method does not share any state
// with the other methods.

View File

@ -11,7 +11,6 @@ import (
"crypto/tls"
"crypto/x509"
"encoding/base64"
"encoding/json"
"errors"
"fmt"
"io"
@ -275,7 +274,7 @@ type InvocationEventHandler interface {
// RequestMessageSupplier is a function that is called to retrieve request
// messages for a GRPC operation. The message contents must be valid JSON. If
// the supplier has no more messages, it should return nil, io.EOF.
type RequestMessageSupplier func() (json.RawMessage, error)
type RequestMessageSupplier func() ([]byte, error)
// InvokeRpc uses te given GRPC connection to invoke the given method. The given descriptor source
// is used to determine the type of method and the type of request and response message. The given
@ -406,7 +405,7 @@ func invokeClientStream(ctx context.Context, stub grpcdynamic.Stub, md *desc.Met
// Upload each request message in the stream
var resp proto.Message
for err == nil {
var data json.RawMessage
var data []byte
data, err = requestData()
if err == io.EOF {
resp, err = str.CloseAndReceive()
@ -528,7 +527,7 @@ func invokeBidi(ctx context.Context, cancel context.CancelFunc, stub grpcdynamic
// Concurrently upload each request message in the stream
var err error
var data json.RawMessage
var data []byte
for err == nil {
data, err = requestData()

View File

@ -1,7 +1,6 @@
package grpcurl_test
import (
"encoding/json"
"fmt"
"io"
"net"
@ -568,7 +567,7 @@ type handler struct {
respTrailersCount int
}
func (h *handler) getRequestData() (json.RawMessage, error) {
func (h *handler) getRequestData() ([]byte, error) {
// we don't use a mutex, though this method will be called from different goroutine
// than other methods for bidi calls, because this method does not share any state
// with the other methods.
@ -580,7 +579,7 @@ func (h *handler) getRequestData() (json.RawMessage, error) {
// insert delay between messages in request stream
time.Sleep(time.Millisecond * 50)
}
return json.RawMessage(h.reqMessages[h.reqMessagesCount-1]), nil
return []byte(h.reqMessages[h.reqMessagesCount-1]), nil
}
func (h *handler) OnResolveMethod(md *desc.MethodDescriptor) {