如何在Go中的运行时创建“鸭子类型”grpc原型消息？

Question

我正在使用grpc protobuf消息定义，并在Go中实现它们。

我的最终目标是让我的rpc在用户上检索一些json并返回带有可选嵌套消息的ProfileMessage，用于解组json的子集。

使用此rpc：

  rpc GetUser (GetUserRequest) returns (Profile) {
    option (google.api.http) = {
      get: "/user/{id=*}"
    };
  }

假设如下：

{
  "profile": {
     "foo": {
        "name": "tim"
        "age": 22
      },
     "bar": {
       "level": 5
     }
  }
}

我想返回一个Profile消息，该消息仅包含“foo”、“bar”或两者，作为基于grpc请求的传入运行时scope参数的嵌套消息（目前，scope将是包含用于将json子设置为相应消息的消息名称的字符串列表，例如["Foo"，"Bar"]）。

给定这些Message定义：

message Profile {
  //both Foo & Bar are optional by default
  Foo foo = 1 [json_name="foo"];
  Bar bar = 2 [json_name="bar"];
}

message Foo {
  string name = 1 [json_name="name"];
  int32 age = 2 [json_name="age"];
}

message Bar {
  string level = 1 [json_name="level"];
}

然后，如果范围是，我希望rpc返回：

Profile{
  Foo: // Foo Message unmarshalled from json
}

或者如果"scope"是["Foo"，"Bar"]那么：

Profile{
  Foo: 
  Bar:
}

问题似乎可以归结为“duck-typing”消息类型。

我几乎用protoreflect

import(
    "google.golang.org/protobuf/reflect/protoregistry"
    "google.golang.org/protobuf/reflect/protoreflect"
)

var scope protoreflect.FullName = "Foo"
var types = new(protoregistry.Types)
var message, errs = types.FindMessageByName(scope)
var almost_foo = message.New()

// using `myJson` object without top level "profile" key to make testing more simple at the moment
var myJson = `{ "foo": { .. }, "bar", { ... } }`
err = json.Unmarshal([]byte(myJson), almost_foo)

但当我尝试使用几乎_foo创建配置文件消息时：

var profile = &pb.Profile{almost_foo}

我得到错误：不能使用几乎_foo（protoreflect.Message类型）作为类型*package_名称。字段值中的Foo

使用

import(
  "github.com/jhump/protoreflect/desc"
  "github.com/jhump/protoreflect/dynamic"
)

我尝试再次动态创建消息：

var fd, errs = desc.LoadFileDescriptor("github.com/package/path/..")
var message_desc = fd.FindMessage("Foo")
var almost_foo = dynamic.NewMessage(message_desc)

出现了一个类似的错误：不能使用几乎_foo（type*dynamic.Message）作为type*package_名称。字段值中的Foo

两次尝试几乎都会创建一条消息，但类型系统仍然不允许实际使用这两种消息。

感谢您的帮助。

Answer 1

我设法解决了这个问题，首先将json缩减到我想要的大小，然后将其解组。

我创建了一条顶级消息，将Profile作为一个字段：

message GetUserResponse {
  Profile profile = 1;
}

并重新定义了反应：

rpc GetUser (GetUserRequest) returns (GetUserResponse)

要构建允许/限定范围的json，我使用：

"github.com/tidwall/gjson"
"github.com/tidwall/sjson"

我将作用域路径（以点分隔，例如[“profile.foo”]）指定的json复制到一个最初为空的输出响应_json字符串中，并只构建允许返回的json。

var response_json string

var scopes = []string{"profile.foo"}

for _, scope := range scopes {
    // copy allowed json subset in raw form
    value := gjson.Get(myJson, scope + "|@ugly")

    // if scope didn't copy a value, skip
    if value.String() == "" {
        continue
    } else {
        // write value to output json using SetRaw since value can be an object
        response_json, _ = sjson.SetRaw(response_json, scope, value.Raw)
    }
}

响应如下所示：

{
  "profile": {
     "foo": {
        "name": "tim"
        "age": 22
      }
  }
}

然后解开玛莎：

response = &pb.VerifyUserResponse{}
err = json.Unmarshal([]byte(response_json), response)