Skip to content

Add Parquet bulk load to quickwit tool local-ingest - #6843

Open
fmassot wants to merge 1 commit into
mainfrom
fmassot/parquet-local-ingest
Open

fmassot wants to merge 1 commit into
mainfrom
fmassot/parquet-local-ingest

Conversation

@fmassot

@fmassot fmassot commented Sep 29, 2026 •

Copy link
Copy Markdown
Collaborator

Motivation

There is no good way to index very fast data from parquet files (or equivalent well compressed file format). This can really help some use cases like large migrations or benchmark.

Solution

Add a CLI command to bulk load of a local Parquet file with N indexing pipelines, bypassing the ingest API and WAL:

quickwit tool local-ingest --index idx --input-path file:///data/x.parquet --num-pipelines 16

Behind the parquet Cargo feature, off by default. No config or proto change.

  • CLI-injected sources share row groups and emit JSON to the regular DocProcessor.
  • No merge pipeline. Requires an empty index; cleanup on failure is best effort.
  • Uploader lifecycle fixes are separate: Wait for uploader tasks before actor shutdown #6868. This PR does not depend on it.

Details: docs/internals/parquet-bulk-load.md.

Tests

309 CLI/indexing tests passed; workspace clippy with and without Parquet, formatting and license checks passed.

Used for the https://github.com/serenedb/searchbench and managed to divide by >2 the time to index.

@fmassot
fmassot force-pushed the fmassot/parquet-local-ingest branch 2 times, most recently from 7e71326 to a0bcbe1 Compare October 2, 2026 13:20
@fmassot
fmassot force-pushed the fmassot/parquet-local-ingest branch 6 times, most recently from d5ec73f to 0277a97 Compare October 4, 2026 16:29
Feature-gated, benchmark-only local Parquet ingestion with parallel indexing
pipelines, bypassing the ingest API and WAL. CLI-injected sources share a
row-group plan and emit JSON documents; servers keep the default source loader.

No merge pipeline is spawned. Loads require an empty index, reject invalid
documents, and attempt index cleanup on failure. Uploader lifecycle changes
are kept in a separate independent PR.
@fmassot
fmassot force-pushed the fmassot/parquet-local-ingest branch from 0277a97 to 70b38e2 Compare October 4, 2026 17:27
@fmassot
fmassot marked this pull request as ready for review October 4, 2026 18:31
@fmassot
fmassot requested a review from a team as a code owner October 4, 2026 18:31
@chatgpt-codex-connector

chatgpt-codex-connector Bot commented Oct 4, 2026 •

Copy link
Copy Markdown

Codex Review Summary

This comment shows the latest Codex review activity on this pull request.

Review Status Commit Review trigger
📝 Code Review ✅ Completed 2026-10-04T18:36:27.888770Z 70b38e2 Draft marked ready
ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review" or "@codex security review".

Codex reacts with 👀 while any review is running, comments if it has suggestions, and reacts with 👍 once all reviews finish with no findings.

@chatgpt-codex-connector chatgpt-codex-connector Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 70b38e24a2

ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

Codex can also answer questions or update the PR. Try commenting "@codex address that feedback".

itertools = { workspace = true }
libz-sys = { workspace = true, optional = true }
oneshot = { workspace = true }
parquet = { workspace = true, optional = true, features = ["arrow", "snap", "zstd", "lz4"] }

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

P2 Badge Enable standard GZIP and Brotli Parquet codecs

When the input file uses GZIP or Brotli compression—both standard Parquet codecs—the reader will reject the file because default-features = false disables those decoders and this feature list enables only Snappy, Zstd, and LZ4. Since the CLI and documentation advertise generic .parquet input without a codec restriction, enable the flate2 and brotli Parquet features (or explicitly reject/document those otherwise-valid files).

Useful? React with 👍 / 👎.

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant