dataset-manager
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.
来源信息
- 仓库
- matrixorigin/sirius
- 最近来源活动
- 2026年4月10日 21:17
- 检测到的 SKILL.md 语言
- 英语
- 星标
- 0
- 分支
- 1
安装方式
默认使用会先检查来源的 Prompt；你也可以切换为直接命令，或下载本地副本。
检查来源文件
决定是否安装前，请先阅读 SKILL.md，以及 SkillsMP 当前展示的配套文件。
正在显示 SKILL.md
SKILL.md
来源说明 · 只读预览- 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 <path>]
# 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.
<!-- To add a new benchmark: copy an existing entry, fill in the fields, and update the description frontmatter. -->
### 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` |
```bash
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` |
```bash
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:
```bash
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
在 GitHub 查看