google.protobuf.Any 带有 grpc 的字段始终为零

时间:2021-07-13 04:06:56

标签: go protocol-buffers grpc grpc-go google-protocol-buffer

我在 golang 中使用 gRPC。我有一个非常简单的 proto 定义和一个 gRPC 服务。 proto 定义在 Endorsement 中有一个 google/protobuf/any 类型的字段。 gRPC 服务无法将此字段映射到输入值,并且始终初始化为 nil

原型定义

syntax = "proto3";

option go_package = "service";
option java_multiple_files = true;
option java_package = "io.grpc.consensus";

import "google/protobuf/any.proto";

package service;

service MyService {
  rpc Verify (Payload) returns (Response) {}
}

message Response {
  string policyId =1;
  string txnId =2;
}

message Endorsement {
  string endorserId=1;
  // This is being initialise to nil by gRPC
  google.protobuf.Any data = 2;
  string signature=3;
  bool isVerified=4;
}


message Payload {
  string policyId =1;
  string txnId =2;
  repeated Endorsement endorsements=3;
}

使用它,实现了一个简单的 gRPC 服务:

package service

import (
    "log"

    "golang.org/x/net/context"
)

type ServiceServerImpl struct {
}

func NewServiceServerImpl() *ServiceServerImpl {
    return &ServiceServerImpl{}
}

func (s *ServiceServerImpl) Verify(ctx context.Context, txnPayload *Payload) (*Response, error) {
    log.Printf("Got verification request: %s", txnPayload.TxnId)
    for _, endorsement := range txnPayload.Endorsements {
        j, err := endorsement.Data.UnmarshalNew()
        if err != nil {
            log.Print("Error while unmarshaling the endorsement")
        }
        if j==nil {
       //This gets printed as payload's endorsement data is always null for google/protobuf/any type
            log.Print("Data is null for endorsement")
        }
    }
    return &Response{TxnId: txnPayload.TxnId,  PolicyId: txnPayload.PolicyId}, nil
}

输入数据:

{
  "policyId": "9dd97b1e-b76f-4c49-b067-22143c954e75",
  "txnId": "231-4dc0-8e54-58231df6f0ce",
  "endorsements": [
    {
      "endorserId": "67e1dfbd-1716-4d91-94ec-83dde64e4b80",
      "data": {
        "type": "issueTx",
        "userId": 1,
        "transaction": {
            "amount": 10123.50
        }
    },
      "signature": "MEUCIBkooxG2uFZeSEeaf5Xh5hWLxcKGMxCZzfnPshOh22y2AiEAwVLAaGhccUv8UhgC291qNWtxrGawX2pPsI7UUA/7QLM=",
      "isVerified": false
    }
  ]
}

客户:

type Data struct {
    Type        string      `json:"type"`
    UserId      int16       `json:"userId"`
    Transaction Transaction `json:"transaction"`
}

type Transaction struct {
    Amount float32 `json:"amount"`
}

data := &Data{Type: "Buffer", UserId: 1, Transaction: Transaction{Amount: 10123.50}}
    byteData, err := json.Marshal(data)
    if err != nil {
        log.Printf("Could not convert data input to bytes")
    }

    e := &any.Any{
        TypeUrl: "anything",
        Value:   byteData,
    }

    endosement := &Endorsement{EndorserId: "1", Signature: "MEUCIBkooxG2uFZeSEeaf5Xh5hWLxcKGMxCZzfnPshOh22y2AiEAwVLAaGhccUv8UhgC291qNWtxrGawX2pPsI7UUA/7QLM=", Data: e, IsVerified: false}
    var endosements = make([]*Endorsement, 1)
    endosements[0] = endosement
    t := &Payload{TxnId: "123", PolicyId: "456", Endorsements: endosements}

    response, err := c.Verify(context.Background(), t)
    if err != nil {
        log.Fatalf("Error when calling SayHello: %s", err)
    }

对于未知类型,google/protobuf/any 类型应该如何解组?

m, err := e.Data.UnmarshalNew()
    if err != nil {
        log.Print("Error while unmarshaling the endorsement")
    }
<块引用>

抛出错误:s:"not found"

2 个答案:

答案 0 :(得分:0)

如您所知,google.protobuf.Any 是 messageType,因此当您没有为其设置任何值时,它将为零。

您必须使用结构解组数据,请参阅下面 protobuf

示例中的代码
      // marshal any
      foo := &pb.Foo{...}
      any, err := anypb.New(foo)
      if err != nil {
        ...
      }
      // unmarshal any 
      foo := &pb.Foo{}
      if err := any.UnmarshalTo(foo); err != nil {
        ...
      }

或者我认为你可以将它与指针接口一起使用(&interface{}) 像这样:

      d := &interface{}{}
      if err := endorsement.Data.UnmarshalTo(d); err != nil {
        ...
      }

答案 1 :(得分:0)

来自包1文档Unmarshaling an Any

<块引用>

UnmarshalNew 使用全局类型注册表来解析消息类型并构造该消息的新实例以进行解组。为了使消息类型出现在全局注册表中,必须将表示该 protobuf 消息类型的 Go 类型链接到 Go 二进制文件中。对于 protoc-gen-go 生成的消息,这是通过导入表示 .proto 文件的生成的 Go 包来实现的。

类型注册表是 protoregistry.GlobalTypes。类型查找是使用 Any.TypeUrl 字段完成的,该字段在编组原始消息时由 gRPC 客户端设置为具体类型的 url。

关于 Any 的令人困惑的细节是它可以是任何 protobuf 消息,但该 protobuf 消息必须在某处定义

您的 .proto 文件没有与输入中的 data 对象匹配的消息定义。可能是此 Data 消息是在其他地方定义的(不是在您自己的 proto 文件中),但无论如何您必须导入生成消息所在的 Go 包。 >

否则,如果输入不是来自已定义的 proto 消息,您可以自己向 proto 添加消息定义,然后使用 UnmarshalTo

// proto file
message Data {
    string type = 1;
    int user_id = 2;
    Transaction transaction = 3;
}

message Transaction {
    float amount = 1;
}

然后:

    for _, endorsement := range txnPayload.Endorsements {
        data := generated.Data{}
        err := endorsement.Data.UnmarshalTo(&data)
        if err != nil {
            log.Print("Error while unmarshaling the endorsement")
        }
    }

如果您只需要任意字节序列,即真正未知的类型,则使用原型 bytes 并将其视为 JSON 负载。

将其建模为 Go 结构体:

type Data struct {
    Type     string `json:"type"`
    UserID   int    `json:"userId"`
    Transaction struct{
        Amount float64 `json:"amount"`
    } `json:"transaction"`

}

或者作为 map[string]interface{},如果客户可以发送任何东西。

然后在您的处理程序函数中:

    for _, endorsement := range txnPayload.Endorsements {
        data := Data{} // or `map[string]interface{}`
        err := json.Unmarshal(endorsement.Data, &data)
        if err != nil {
            log.Print("Error while unmarshaling the endorsement")
        }
    }