Perf: add in-process timing instrumentation to replace py-spy dependency
メンテナーはふだん 1 日以内に返信
まだ誰も着手していません。
評価
- 難易度
- 4/5
- 見積もり時間
- 3〜5日
- 初心者へのやさしさ
- 46/100
- issue の種類
- 機能追加
- 明瞭さ
- おおむね明確
- 活発さ
- 停滞
- 技術スタック
- python
- 領域
- performance
調査の方向性
Start by reading read_xarray_table and _iteration_callback in the reader, then inspect perf_tests/profile.sh and perf_tests/query_large_synthetic.py to understand the current profiling workflow. Choose and implement a timing approach that reports per-partition costs across the relevant execution stages. Confirm the measurements distinguish I/O, conversion, and DataFusion work without requiring py-spy or root access.
索引モデルが issue の本文から書いたものです。
説明
Problem
The current profiling workflow in perf_tests/ requires py-spy, which needs sudo on macOS:
sudo ./profile.sh groupby_air_full.py
This makes profiling unavailable in many CI environments and for users without root access. Additionally, cProfile only sees the Python side of query execution — the entire DataFusion execution appears as a single opaque to_arrow_table() call (0.66s for 320 MB of data in our benchmarks), making it impossible to distinguish:
- Time spent in xarray I/O (
ds.isel(block),.values) - Time spent in numpy → Arrow conversion (
iter_record_batches) - Time spent in DataFusion (aggregation, hash-grouping, row filtering)
- Time spent waiting for GIL between partitions
Proposed solution
Add a lightweight per-partition timing mode to the _iteration_callback mechanism already in read_xarray_table:
# reader.py — extend make_stream to record timing
def make_stream(projected_names=None) -> pa.RecordBatchReader:
if _timing_callback is not None:
t0 = time.perf_counter()
result = pa.RecordBatchReader.from_batches(...)
if _timing_callback is not None:
_timing_callback({"elapsed_s": time.perf_counter() - t0, "block": block})
return result
Or alternatively: add a verbose=True flag to read_xarray_table that prints per-partition timings to stderr, giving insight into whether execution is I/O-bound or compute-bound without requiring external tools.
A lower-overhead option is to expose a PartitionTimingHook via LazyArrowStreamTable that fires before/after each factory call.
Current profiling workaround
For the query_large_synthetic.py benchmark (320 MB, 8 partitions):
Full scan AVG: 0.662s (341 MB peak) ← all in Rust's to_arrow_table()
Filter pushdown: 0.005s (0.7 MB peak) ← 7/8 partitions pruned
GROUP BY lat/lon: 0.839s (341 MB peak) ← +27% vs. AVG for hash aggregation
cProfile shows 246 Python calls for the full scan, but 0.660s of those are attributed to {method 'to_arrow_table' of 'datafusion.DataFrame' objects} — a Rust builtin that cProfile cannot introspect.
Related
_iteration_callbackinXarrayRecordBatchReaderandread_xarray_table— already used for testingperf_tests/profile.sh— current py-spy based approachperf_tests/query_large_synthetic.py— new benchmark script added alongside this issue
- 主要言語
- Python
- スター
- 142
- フォーク
- 19
- 平均マージ
- 1日 6時間
- マージ済み PR(30日)
- 10
環境構築
- Dockerfile・Docker Compose ファイルなし
- プルリクエストのテンプレートなし
- コントリビューションガイドを読む
はじめの一歩
- issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
- 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
- リポジトリをフォークし、ブランチを切って変更します。
- issue 番号を参照したプルリクエストを送ります。
xqlsystems/xarray-sql のほかの issue
-
documentation
難易度 1/5 1〜3時間 初心者へのやさしさ 85/100
xqlsystems/xarray-sql#268 ·
メンテナーはふだん 1 日以内に返信
-
enhancement
難易度 2/5 1〜3時間 初心者へのやさしさ 64/100
xqlsystems/xarray-sql#267 ·
メンテナーはふだん 1 日以内に返信
-
難易度 1/5 1時間未満 初心者へのやさしさ 74/100
xqlsystems/xarray-sql#243 ·
メンテナーはふだん 1 日以内に返信
-
publish.yml: the TestPyPI verification step is unreachable (and installs the wrong package name)オープン
難易度 2/5 1〜3時間 初心者へのやさしさ 62/100
xqlsystems/xarray-sql#242 ·
メンテナーはふだん 1 日以内に返信
-
難易度 2/5 1〜3時間 初心者へのやさしさ 72/100
xqlsystems/xarray-sql#237 ·
メンテナーはふだん 1 日以内に返信
xqlsystems/xarray-sql の issue をすべて見る
似ている issue
-
bug good first issue
難易度 2/5 1〜3時間 初心者へのやさしさ 78/100
yunaremaia/proto-drift#22 ·
-
bug good first issue
難易度 2/5 1〜3時間 初心者へのやさしさ 82/100
yunaremaia/memwatch#96 ·
-
bug good first issue
難易度 1/5 1時間未満 初心者へのやさしさ 91/100
yunaremaia/ci-gate-watch#34 ·
-
bug good first issue
難易度 2/5 1〜3時間 初心者へのやさしさ 85/100
yunaremaia/agent-capability-attestation#64 ·
メンテナーはふだん 1 日以内に返信
-
bug good first issue
難易度 1/5 1時間未満 初心者へのやさしさ 86/100
yunaremaia/agent-call-graph#43 ·
メンテナーはふだん 1 日以内に返信