By name: dataset-manager
description: >
Use this skill to generate benchmark datasets (TPC-H, TPC-DS, etc.). Trigger when the user
needs test data at a specific scale factor for benchmarking or testing. Supports parquet and
duckdb output formats.
argument-hint: "[benchmark] [scale_factor] [--format duckdb|parquet] [--output ]"
Dataset Generator
Generate benchmark datasets by running the corresponding shell script under test/<benchmark>_performance/.
Gather Parameters
Parse $ARGUMENTS for:
- Benchmark: name from the registry below (first positional arg)
- Scale factor: integer (second positional arg)
- Format:
--format duckdb|parquet(optional — each benchmark has a default) - Output path:
--output <path>(optional — scripts have sensible defaults)
If any required parameter (benchmark, scale factor) is missing, ask the user.
Benchmark Registry
Each entry follows the same structure: script location, command template, supported formats, defaults, and prerequisites.
TPC-H
| Field | Value |
|---|---|
| Script | test/tpch_performance/generate_tpch_data.sh |
| Default format | parquet |
| Formats | parquet (tpchgen-rs), duckdb (DuckDB dbgen()) |
| Default output (parquet) | test_datasets/tpch_parquet_sf<SF> |
| Default output (duckdb) | test_datasets/tpch_sf<SF>.duckdb |
| Prerequisites | Parquet: pixi env (rust, python, pyarrow). DuckDB: build/release/duckdb |
cd test/tpch_performance && pixi run bash generate_tpch_data.sh <SF> --format <FORMAT> [--output <path>]
Notes:
- If the parquet output directory already exists, the script skips generation
TPC-DS
| Field | Value |
|---|---|
| Script | test/tpcds_performance/generate_tpcds_data.sh |
| Default format | duckdb |
| Formats | duckdb, parquet |
| Default output (duckdb) | test_datasets/tpcds_sf<SF>.duckdb |
| Default output (parquet) | test_datasets/tpcds_parquet_sf<SF> |
| Prerequisites | build/release/duckdb |
cd test/tpcds_performance && bash generate_tpcds_data.sh <SF> --format <FORMAT> [--output <path>]
Notes:
- Also extracts TPC-DS query files to
test/tpcds_performance/queries/q{1..99}.sql
Prerequisites
For any benchmark that requires the DuckDB binary, check before running:
test -x build/release/duckdb
If missing, tell the user to build first: CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) make
Report Results
- Benchmark name
- Output path
- Format used
- Whether generation was skipped (output already existed) or completed