66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
package grpc
|
|
|
|
import (
|
|
"go.unistack.org/micro/v3/codec"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/encoding"
|
|
)
|
|
|
|
var (
|
|
_ encoding.Codec = &wrapMicroCodec{}
|
|
_ codec.Codec = &wrapGrpcCodec{}
|
|
)
|
|
|
|
type wrapStream struct{ grpc.ClientStream }
|
|
|
|
func (w *wrapStream) Write(d []byte) (int, error) {
|
|
n := len(d)
|
|
err := w.ClientStream.SendMsg(&codec.Frame{Data: d})
|
|
return n, err
|
|
}
|
|
|
|
func (w *wrapStream) Read(d []byte) (int, error) {
|
|
m := &codec.Frame{}
|
|
err := w.ClientStream.RecvMsg(m)
|
|
copy(d, m.Data)
|
|
return len(d), err
|
|
}
|
|
|
|
type wrapMicroCodec struct{ codec.Codec }
|
|
|
|
func (w *wrapMicroCodec) Name() string {
|
|
return w.Codec.String()
|
|
}
|
|
|
|
func (w *wrapMicroCodec) Marshal(v interface{}) ([]byte, error) {
|
|
return w.Codec.Marshal(v)
|
|
}
|
|
|
|
func (w *wrapMicroCodec) Unmarshal(d []byte, v interface{}) error {
|
|
return w.Codec.Unmarshal(d, v)
|
|
}
|
|
|
|
type wrapGrpcCodec struct{ encoding.Codec }
|
|
|
|
func (w *wrapGrpcCodec) String() string {
|
|
return w.Codec.Name()
|
|
}
|
|
|
|
func (w *wrapGrpcCodec) Marshal(v interface{}, opts ...codec.Option) ([]byte, error) {
|
|
if m, ok := v.(*codec.Frame); ok {
|
|
return m.Data, nil
|
|
}
|
|
return w.Codec.Marshal(v)
|
|
}
|
|
|
|
func (w *wrapGrpcCodec) Unmarshal(d []byte, v interface{}, opts ...codec.Option) error {
|
|
if d == nil || v == nil {
|
|
return nil
|
|
}
|
|
if m, ok := v.(*codec.Frame); ok {
|
|
m.Data = d
|
|
return nil
|
|
}
|
|
return w.Codec.Unmarshal(d, v)
|
|
}
|