使用结构（或 protobuf 消息）作为数据更新 Golang 中的 Firestore 文档

Question

给出以下 API：

syntax = "proto3";

service FooService {
  rpc UpdateFoo(UpdateFooRequest) returns (Foo) {}
}

message Foo {
  string name = 1;
  string description = 2;
  string action = 3;
}

message UpdateFooRequest {
  Foo foo = 1;
  // The list of fields to be updated. For the `FieldMask` definition,
  // see https://developers.google.com/protocol-buffers/docs/reference/google.protobuf#fieldmask
  google.protobuf.FieldMask update_mask = 2;
}

此 API 与后端 Firestore“foos”集合对话。
实现在 Golang 中。 firestore package 有一个 Update() 方法，但它需要提供一个 firestore.Update 结构列表：

type Update struct {
    Path      string // Will be split on dots, and must not contain any of "˜*/[]".
    FieldPath FieldPath
    Value     interface{}
}

我在编写 RPC 的实现时遇到问题 UpdateFoo：我无法轻松构建 firestore.Update 结构。同时，我要确保update_mask中提供的路径有效。

这就是我正在尝试的：

func (s *Server) UpdateFoo(ctx context.Context, req *pb.UpdateFooRequest) (*pb.Foo, error) {

var updates []firestore.Update
// Loop all paths in the update_mask
for _, p := range req.GetUpdateMask().GetPaths() {
   // Check that path is valid field of Foo message.
   // How to do this?
   ...
   ...
   if path is not valid {
       return nil, status.Errorf(codes.InvalidArgument, "invalid path %s", p)
   }
   // When path is valid, I need to take the value from the req.GetFoo() message
   // How to do this?
   newValue := req.GetFoo().Get????<this is p variable>

   update = append(update, &firestore.Update{Path: p, Value: newValue}
}

如有任何提示，我们将不胜感激。

Answer 1

我运行遇到了一个类似的问题，试图在 Golang 中使用 firestore 和 protobuf。似乎不支持使用 proto 字段掩码和结构进行更新。

这是 Set 和 fieldmask-utils 的解决方法：

import (
  "github.com/golang/protobuf/protoc-gen-go/generator"
  fieldmask_utils "github.com/mennanov/fieldmask-utils"
)

var (
  id      string
  request UpdateFooRequest
)

mask, err: = fieldmask_utils.MaskFromProtoFieldMask(request.FieldMask, generator.CamelCase)
// handle err...

m := make(map[string]interface{}) // a map to copy to
err := fieldmask_utils.StructToMap(mask, request.Foo, m)
// handle err..

_, err = r.client.Collection(r.collection).Doc(id).Set(ctx, m, firestore.MergeAll)
// handle err..

使用结构（或 protobuf 消息）作为数据更新 Golang 中的 Firestore 文档

Updating Firestore documents in Golang using structs (or protobuf messages) as data

google-cloud-firestore

protoreflect