如何通过 RPC 监听 firestore

Question

我想收听 Firestore 中的实时变化，我也只能使用 Go。由于 Firestore SDK for Go 没有任何选项来监听实时变化，我决定使用 firestore v1beta1 sdk。

我已经编写了以下代码来做到这一点

func TestRPCHandler(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
c, err := firestore.NewClient(context.Background())

databaseName := "projects/[project_name]/databases/(default)"
if err != nil {
    panic(err)
}

stream, err := client.Listen(context.Background())
if err != nil {
    panic(err)
}

request := &firestorepb.ListenRequest{
    Database:             databaseName,
    TargetChange:         &firestorepb.ListenRequest_AddTarget{
        AddTarget: &firestorepb.Target{
            TargetType:           &firestorepb.Target_Documents{
                Documents: &firestorepb.Target_DocumentsTarget{
                    Documents:    []string{"projects/[project_name]/databases/(default)/[collection_name]"} ,
                },
            },
        },
    },
}

if err := stream.Send(request); err != nil {
    panic(err)
}

if err := stream.CloseSend(); err != nil {
    panic(err)
}

for {
    resp, err := stream.Recv()
    if err == io.EOF {
        break
    }
    if err != nil {
        panic(err)
    }
}

   }

当我这样做时，代码不会检测到我在数据库中手动带来的任何更改。 stream.Recv() 只是返回 EOF 并立即退出。我什至尝试通过添加time.Sleep() 手动等待，但这也无济于事。

Answer 1

您不需要测试版 SDK 或 hack 来实现这一点，我找到了解决方案，实际上非常简单。

https://firebase.google.com/docs/firestore/query-data/listen 文档不包含 Go 示例。

Go 的 firestore 客户端 API 的源代码有一个未导出的 watchStream，我们不能直接使用它：https://github.com/googleapis/google-cloud-go/blob/master/firestore/watch.go#L130

对存储库的深入搜索显示，这实际上用于 DocumentSnapshotIterator 和 QuerySnapshotIterator，位于：https://github.com/googleapis/google-cloud-go/blob/master/firestore/docref.go#L644 和：https://github.com/googleapis/google-cloud-go/blob/master/firestore/query.go#L716。

Collection 包含一个 Snapshots 方法，它返回我们想要的快照迭代器，之后一切都简单了，我们只需通过它的 Next 方法进行一个不定式循环。

例子：

cols, err := client.Collections(context.Background()).GetAll()

for _, col := range cols {
    iter := col.Snapshots(context.Background())
    defer iter.Stop()

    for {
        doc, err := iter.Next()
        if err != nil {
            if err == iterator.Done {
                break
            }
            return err
        }

        for _, change := range doc.Changes {
            // access the change.Doc returns the Document,
            // which contains Data() and DataTo(&p) methods.
            switch change.Kind {
            case firestore.DocumentAdded:
                // on added it returns the existing ones.
                isNew := change.Doc.CreateTime.After(l.startTime)
                // [...]
            case firestore.DocumentModified:
                // [...]
            case firestore.DocumentRemoved:
                // [...]
            }
        }
    }
}

你的，Gerasimos Maropoulos 又名 @kataras

Answer 2

Firebase 的 Get realtime updates with Cloud Firestore 文档目前表明 Go 尚不支持。

// Go 客户端库尚不支持