如何在gRPC调用中直接传递原始二进制文件？

Question

我已经有了protobuf消息对象的proto二进制文件。

现在要进行gRPC调用，我需要将其解组并通过message对象发送到gRPC客户端（该客户端将再次封送相同的对象），这会浪费延迟。

如何通过将二进制文件传递给客户端来避免这种情况？

Answer 1

如果要发送原始二进制数据，请使用字节的protobuf类型，例如。

// .proto file
message MyRawData {
  bytes rawbin = 1;
}

这将在go中转换为字节片，非常适合存储原始二进制：

// go-code
type MyRawData struct {   
    Rawbin []byte   `protobuf:"bytes,1,opt,name=rawbin,proto3" json:"rawbin,omitempty"`
}

Answer 2

这可以使用自定义编解码器。自定义编解码器可以定义为

type StubbedCodec struct{}

func (cb StubbedCodec) Marshal(v interface{}) ([]byte, error) {
    return v.([]byte), nil
}

func (cb StubbedCodec) Unmarshal(data []byte, v interface{}) error {
    ba, _ := v.([]byte)

    for index, byte := range data {
        ba[index] = byte
    }
    return nil
}

一旦我们有了这个，我们就可以将编解码器作为拨号选项传递给

grpc.Dial(grpcServer, grpc.WithInsecure(), grpc.WithBlock(), grpc.WithDefaultCallOptions(grpc.ForceCodec(StubbedCodec{})))

这将迫使grpc使用您的编解码器，它基本上什么都不做（如上所述）。