99 lines
2.3 KiB
Go
99 lines
2.3 KiB
Go
|
package grpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/golang/protobuf/proto"
|
||
|
|
"github.com/json-iterator/go"
|
||
|
|
"github.com/micro/go-micro/codec"
|
||
|
|
"github.com/micro/go-micro/codec/jsonrpc"
|
||
|
|
"github.com/micro/go-micro/codec/protorpc"
|
||
|
|
"google.golang.org/grpc/encoding"
|
||
|
|
)
|
||
|
|
|
||
|
|
type jsonCodec struct{}
|
||
|
|
type protoCodec struct{}
|
||
|
|
type bytesCodec struct{}
|
||
|
|
type wrapCodec struct{ encoding.Codec }
|
||
|
|
|
||
|
|
var (
|
||
|
|
defaultGRPCCodecs = map[string]encoding.Codec{
|
||
|
|
"application/json": jsonCodec{},
|
||
|
|
"application/proto": protoCodec{},
|
||
|
|
"application/protobuf": protoCodec{},
|
||
|
|
"application/octet-stream": protoCodec{},
|
||
|
|
"application/grpc+json": jsonCodec{},
|
||
|
|
"application/grpc+proto": protoCodec{},
|
||
|
|
"application/grpc+bytes": bytesCodec{},
|
||
|
|
}
|
||
|
|
|
||
|
|
defaultRPCCodecs = map[string]codec.NewCodec{
|
||
|
|
"application/json": jsonrpc.NewCodec,
|
||
|
|
"application/json-rpc": jsonrpc.NewCodec,
|
||
|
|
"application/protobuf": protorpc.NewCodec,
|
||
|
|
"application/proto-rpc": protorpc.NewCodec,
|
||
|
|
"application/octet-stream": protorpc.NewCodec,
|
||
|
|
}
|
||
|
|
|
||
|
|
json = jsoniter.ConfigCompatibleWithStandardLibrary
|
||
|
|
)
|
||
|
|
|
||
|
|
// UseNumber fix unmarshal Number(8234567890123456789) to interface(8.234567890123457e+18)
|
||
|
|
func UseNumber() {
|
||
|
|
json = jsoniter.Config{
|
||
|
|
UseNumber: true,
|
||
|
|
EscapeHTML: true,
|
||
|
|
SortMapKeys: true,
|
||
|
|
ValidateJsonRawMessage: true,
|
||
|
|
}.Froze()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w wrapCodec) String() string {
|
||
|
|
return w.Codec.Name()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (protoCodec) Marshal(v interface{}) ([]byte, error) {
|
||
|
|
return proto.Marshal(v.(proto.Message))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (protoCodec) Unmarshal(data []byte, v interface{}) error {
|
||
|
|
return proto.Unmarshal(data, v.(proto.Message))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (protoCodec) Name() string {
|
||
|
|
return "proto"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (bytesCodec) Marshal(v interface{}) ([]byte, error) {
|
||
|
|
b, ok := v.(*[]byte)
|
||
|
|
if !ok {
|
||
|
|
return nil, fmt.Errorf("failed to marshal: %v is not type of *[]byte", v)
|
||
|
|
}
|
||
|
|
return *b, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (bytesCodec) Unmarshal(data []byte, v interface{}) error {
|
||
|
|
b, ok := v.(*[]byte)
|
||
|
|
if !ok {
|
||
|
|
return fmt.Errorf("failed to unmarshal: %v is not type of *[]byte", v)
|
||
|
|
}
|
||
|
|
*b = data
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (bytesCodec) Name() string {
|
||
|
|
return "bytes"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (jsonCodec) Marshal(v interface{}) ([]byte, error) {
|
||
|
|
return json.Marshal(v)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (jsonCodec) Unmarshal(data []byte, v interface{}) error {
|
||
|
|
return json.Unmarshal(data, v)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (jsonCodec) Name() string {
|
||
|
|
return "json"
|
||
|
|
}
|