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

问题描述

使用以下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}
}

任何提示将不胜感激。

解决方法

我在Golang中尝试将Firestore和protobuf一起使用时遇到了类似的问题。似乎不支持使用原型场掩码和结构进行更新。

这是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..