Synchronous Firestore call unbinds the caller thread's `contextvars` Context (breaks Flask request context) under gunicorn sync worker

未關閉
#17,821 2 則留言 0 個 reaction 已指派 0 人 在 GitHub 檢視

維護者通常 1 天內回覆

還沒有人認領這個 Issue。

評估

難度
4/5
預估耗時
3-5 天
新手友好度
45/100
Issue 類型
缺陷
描述清晰度
基本清楚
活躍度
冷清
技術堆疊
flask, grpc, python
領域
backend, databases

研究方向

首先,在 Flask 2.3 和 gunicorn sync workers 下重現同步 get()/batch.commit() 流程,然後檢查 grpc/_channel.py 中 _run_channel_spin_thread、ForkManagedThread 和 segregated_call 附近的程式碼。將其行為與 grpc/grpc#40123 以及 google/cloud/firestore_v1 中的 Firestore 用戶端呼叫點進行比較;完成標準是確認 context 變更是 gRPC 全域性的還是 Firestore 特有的,並記錄受支援的修正或 workaround。

由索引模型根據 Issue 內容生成。

描述

api: firestore priority: p3 type: question
Determine this is the right repository
  • I determined this is the correct repository in which to report this bug.
Summary of the issue

A synchronous Firestore call (gRPC transport) made from inside a request-handling
thread switches/unbinds that thread's contextvars Context partway through. Any
framework that stores per-request state in a ContextVar (Flask 2.3+ keeps its request
context in _cv_request) sees that state disappear on the same thread immediately
after the Firestore call. In our Flask app this surfaces as:

RuntimeError: Working outside of request context

raised during response finalization — a 500 to the client, even though the view code
completed successfully.

Firebase support directed us here, and pointed to a similar regression in the gRPC repo:
grpc/grpc#40123 ("contextvar loss when using thread pool executors in newer gRPC
versions"). Filing so the Cloud Firestore team can determine whether this is global gRPC
behavior or specific to the Firestore client's transport.

Environment

  • Python: 3.9
  • google-cloud-firestore: 2.18.0
  • firebase-admin: 6.5.0 (thin wrapper over the client above)
  • google-api-core: 2.19.2
  • grpcio / grpcio-status: 1.66.1
  • Transport: default (gRPC)
  • Web stack: Flask 2.3.0 / Werkzeug 3.0.3, served by gunicorn sync worker
    (one request at a time per worker process; stable OS thread)
  • Client init: module-level firestore.client() (channel created lazily on first RPC)

Observed behavior

We instrumented contextvars.copy_context and logged has_request_context() (hrc)
with a full stack at each call. Within a single request, on the same OS thread:

(A) First Firestore call — hrc is True (context healthy). Fires at channel creation:

grpc/_channel.py:1761  _run_channel_spin_thread -> cygrpc.ForkManagedThread(target=channel_spin)
grpc/_channel.py:1821  create -> _run_channel_spin_thread(state)
grpc/_channel.py:1414  __call__ -> self._managed_call(...)
google/api_core/grpc_helpers.py:164  error_remapped_callable
google/api_core/timeout.py / retry_unary.py / gapic_v1/method.py
google/cloud/firestore_v1/services/firestore/client.py:1154  batch_get_documents
google/cloud/firestore_v1/document.py:402  get()          # <- our doc_ref.get()
... application view ... gunicorn sync worker.handle_request
[hrc=True]

(B) A later Firestore call in the SAME request — hrc is now False (context gone):

grpc/_channel.py:1146  _blocking -> self._channel.segregated_call(...)
grpc/_channel.py:1178  __call__ -> self._blocking(...)
google/api_core/grpc_helpers.py:76  error_remapped_callable
google/api_core/timeout.py / retry_unary.py / gapic_v1/method.py
google/cloud/firestore_v1/services/firestore/client.py:1372  commit
google/cloud/firestore_v1/batch.py:59  commit()           # <- our batch.commit()
... same application/gunicorn stack ...
[hrc=False]

Between (A) and (B) our code runs only in-process work (SQLAlchemy queries + a list
comprehension) — nothing that touches contextvars, spawns threads, or uses
asyncio/greenlets. The only machinery that manipulates context between the two points
is the Firestore/gRPC transport (_run_channel_spin_thread / cygrpc.ForkManagedThread
at channel creation, and segregated_call on subsequent RPCs).

Expected vs actual

  • Expected: a synchronous Firestore RPC leaves the calling thread's contextvars
    Context unchanged.
  • Actual: after the RPC, the calling thread is running in a different Context; any
    ContextVar set before the call (e.g. Flask's _cv_request) reads as unset.

Reproduction

Flask 2.3 app served by gunicorn sync worker. In a view: log has_request_context(),
do a synchronous Firestore read+write (doc_ref.get() then batch.commit()), log
has_request_context() again. On the first Firestore-touching request after a worker
starts (fresh gRPC channel), the second check is False and Flask raises "Working outside
of request context" during response finalization. Minimal standalone repro available on
request.

Questions

  1. Is the Firestore Python client (or the underlying gRPC transport) expected to alter
    the calling thread's contextvars Context during a synchronous RPC?
  2. Is this the same root cause as grpc/grpc#40123, or specific to the Firestore client's
    ForkManagedThread / segregated_call usage?
  3. Supported ways to avoid it:
    • Pre-creating/warming the gRPC channel at process (post-fork) startup — does that
      help, or does segregated_call disturb context on every RPC regardless of channel
      age?
    • Using the REST transport (transport="rest") — is that a supported path to avoid
      the gRPC thread/context machinery entirely?
    • A known-good grpcio version to pin below the regression, if this is version-specific?

Current workaround

We run all Firestore calls on a short-lived daemon thread so the context churn lands on
a disposable thread and the request thread's context stays intact. This resolves the
500s, but we would prefer a supported fix rather than thread-isolating every call.

Impact

Intermittent 500s (correlated with the first Firestore call per worker / gRPC channel
creation) on any endpoint that makes a synchronous Firestore call, in production.

主要語言
Python
星號
5.4k
分支
1.8k
平均合併
2 天 7 小時
30 天內合併 PR
173

環境準備

從這裡開始

  1. 先讀完整個 Issue,再讀專案的貢獻指南。
  2. 在 Issue 下留言說明你要接手 —— 這能避免兩個人做同樣的事。
  3. Fork 儲存庫,在一個分支上完成修改。
  4. 送出 Pull Request,並在描述裡引用這個 Issue 編號。

googleapis/google-cloud-python 的其他 Issue

查看 googleapis/google-cloud-python 的全部 Issue

相似的 Issue

更多 Python Issue

把新 issue 寄到你的電子郵件信箱

精選適合新手參與的 GitHub issue 摘要。