是否应该根据 Google App Engine 的请求创建 Firestore 客户端？

Question

我对如何解决这个问题感到困惑。

GAE 似乎希望每个客户端库都使用范围为 http.Request 的 context.Context。

我以前有过做这样的事情的经验：

main.go

type server struct {
    db *firestore.Client
}

func main() {
    // Setup server
    s := &server{db: NewFirestoreClient()}

    // Setup Router
    http.HandleFunc("/people", s.peopleHandler())

    // Starts the server to receive requests
    appengine.Main()
}

func (s *server) peopleHandler() http.HandlerFunc {
    // pass context in this closure from main?
    return func(w http.ResponseWriter, r *http.Request) {
        ctx := r.Context() // appengine.NewContext(r) but should it inherit from background somehow?
        s.person(ctx, 1)
        // ...
    }
}

func (s *server) person(ctx context.Context, id int) {
    // what context should this be?
    _, err := s.db.Client.Collection("people").Doc(uid).Set(ctx, p)
    // handle client results
}

火力基地

// Firestore returns a warapper for client
type Firestore struct {
    Client *firestore.Client
}

// NewFirestoreClient returns a firestore struct client to use for firestore db access
func NewFirestoreClient() *Firestore {
    ctx := context.Background()
    client, err := firestore.NewClient(ctx, os.Getenv("GOOGLE_PROJECT_ID"))
    if err != nil {
        log.Fatal(err)
    }

    return &Firestore{
        Client: client,
    }
}

这对如何确定项目范围的客户有很大的影响。例如，挂断 aserver{db: client}并在该结构上附加处理程序，或者必须通过请求中的依赖注入将其传递出去。

我确实注意到使用客户端的调用需要另一个上下文。所以也许它应该是这样的：

1. main.go 创建一个 ctx := context.Background()
2. main.go 将其传递给新客户
3. handler ctx := appengine.NewContext(r)

基本上初始设置无关紧要，context.Background()因为新请求与 App Engine 具有不同的上下文？

我可以将 ctx 从 main 传入处理程序，然后将 NewContext 传入 + 请求？

对此的惯用方法是什么？

注意：Firestore在之前的迭代中，我也从结构中删除了 firestore 方法......

Answer 1

您应该firestore.Client为多次调用重用该实例。但是，这在 GAE 标准的旧 Go 运行时中是不可能的。因此，在这种情况下，您必须为firestore.Client每个请求创建一个新的。

但是，如果您使用新的 Golang 1.11 运行时 GAE 标准，那么您可以随意使用您喜欢的任何上下文。在这种情况下，您可以firestore.Client在main()函数或init()使用后台上下文的函数中进行初始化。然后，您可以使用请求上下文在请求处理程序中进行 API 调用。

package main

var client *firestore.Client

func init() {
  var err error
  client, err = firestore.NewClient(context.Background())
  // handle errors as needed
}

func handleRequest(w http.ResponseWriter, r *http.Request) {
  doc := client.Collection("cities").Doc("Mountain View")
  doc.Set(r.Context(), someData)
  // rest of the handler logic
}

这是我使用 Go 1.11 和 Firestore 实现的示例 GAE 应用程序，它演示了上述模式：https : //github.com/hiranya911/firecloud/blob/master/crypto-fire-alert/cryptocron/web/main.go

有关 GAE 中 Go 1.11 支持的更多信息：https : //cloud.google.com/appengine/docs/standard/go111/go-differences