GitHub - rocky-data/rocky: The typed graph between your code and whichever warehouse, table format, or query engine you've chosen — typed compiler, branches, replay, column-level lineage, compile-time contracts, per-model cost. Adapters: Databricks, Snowflake, BigQuery, DuckDB. Single static Rust binary. Apache 2.0. · GitHub

Source: original

Rocky

License: Apache 2.0

Rocky is the typed graph between your code and whichever warehouse, table format, or query engine you've chosen.

It is a typed compiler that runs over your existing Databricks, Snowflake, BigQuery, or DuckDB and owns the graph between your code and your data: named branches, content-addressed run records, column-level lineage, compile-time contracts, and per-model cost. Storage and compute stay where they are, and Rocky works on the SQL you already have. The .rocky DSL is there when you want it. Apache 2.0.

The failures that cost data teams the most are invisible to the warehouse and out of scope for the templating layer above it: schema drift, column-rename blast radius, dialect divergence, cost spikes nobody can attribute. Rocky turns them into compile errors and blocked PRs.

Rocky quickstart — create a project, compile, and run 3 models in under 15s

Try it in 60 seconds

macOS / Linux

curl -fsSL https://raw.githubusercontent.com/rocky-data/rocky/main/engine/install.sh | bash

Windows (PowerShell)

irm https://raw.githubusercontent.com/rocky-data/rocky/main/engine/install.ps1 | iex

rocky playground my-first-project cd my-first-project rocky compile && rocky test && rocky run

No credentials needed; the playground runs end-to-end on local DuckDB.

rocky run is the one-step path for local iteration and automation. For production or PR-gated deploys, split it into rocky plan (builds and persists an auditable plan to .rocky/plans/<id>.json) followed by rocky apply <plan-id>.

Who Rocky is for

Rocky is built first for data platform engineers running production-critical, multi-tenant pipelines on Databricks , where silent failures cost real money and Dagster is already the orchestrator. That is the launch wedge, and where Rocky is most battle-tested.

The next ring out: Snowflake and BigQuery shops evaluating SQLMesh, who want correctness in the compiler rather than the planner and prefer SQL by default. Adapters are Beta today; see Where Rocky is today below.

See it in action

Each demo below is a self-contained POC in examples/playground/pocs/: cd in, run ./run.sh, and reproduce it locally.

Detects schema drift the moment it happens

A source column type changes upstream. On the next run, Rocky diffs source against target, drops the target, and recreates it. No silent data corruption.

rocky run detects source type change and recreates the target

POC: 02-performance/06-schema-drift-recover

Enforces data contracts at compile time

Missing required columns, protected columns being removed, or unsafe type changes surface as diagnostic codes (E010, E013) before a single row is written.

rocky compile flags E010 and E013 contract violations on broken_metrics

POC: 01-quality/01-data-contracts-strict

Named branches for risk-free experiments

Create a branch, run against it in an isolated schema, inspect, then drop or promote. Column-level lineage shows the downstream blast radius before you ship.

rocky branch create, run on branch, and trace column lineage downstream

POC: 00-foundations/06-branches-replay-lineage

Column-level lineage, not table-level

Trace a single column from a downstream fact back through its aggregations, all the way to the seed. Blast-radius analysis without reading every model.

rocky lineage --column traces fct_revenue.total back to seeds.orders.amount

POC: 06-developer-experience/01-lineage-column-level

AI model generation with a compile-validate loop

Describe what you want in plain English. Rocky generates a Rocky DSL model, compiles it, and retries on parse failure. The Attempts: 2 line shows the loop catching a first-pass error.

rocky ai generates a .rocky model from natural language intent, Attempts: 2

POC: 03-ai/01-model-generation

PR-time blast-radius with rocky lineage-diff

Compare two git refs and get a per-changed-column readout of downstream consumers; the pre-rendered Markdown drops straight into a GitHub PR comment. CODEOWNERS-style review tooling can't reach this granularity without a compiled engine.

rocky lineage-diff main lists added and removed columns across two models with downstream consumers per change

POC: 06-developer-experience/11-lineage-diff

Classify columns, mask by environment, gate CI

Tag PII columns in the model sidecar, and bind tags to mask strategies in [mask] / [mask.<env>]. rocky compliance --env prod --fail-on exception exits 1 the moment a classified column has no resolved strategy: a one-line CI gate against accidentally unmasked data.

rocky compliance rolls up classification tags to mask strategies; --fail-on exception exits 1, gating CI on unmasked PII

POC: 04-governance/05-classification-masking-compliance

Incremental loads with persistent watermark state

strategy = "incremental" plus a timestamp_column is all it takes. Rocky writes the high-water mark to the embedded state store, and subsequent runs only INSERT … WHERE timestamp > watermark. Append 25 rows after a 500-row load, and run 2 still finishes in 0.2s.

rocky run with incremental strategy: run 1 copies 500 rows; appended 25 rows; run 2 only copies the delta in 0.2s

POC: 02-performance/01-incremental-watermark

Native BigQuery: materialize live, cost to the byte

Swap the adapter to BigQuery and the same project materializes a full-refresh CREATE TABLE AS against the live warehouse. Rocky's run receipt reports bytes_scanned and cost_usd, and a cross-check confirms that bytes_scanned matches BigQuery's own totalBytesBilled for the job, to the byte. The same models run against Snowflake or Databricks by changing only the adapter.

rocky run materializes a full-refresh table in BigQuery with a cost receipt, then a cross-check shows bytes_scanned equals BigQuery's totalBytesBilled

POC: 07-adapters/05-bigquery-native-queries — the live path requires BigQuery credentials

In your editor

The same compiler runs as a language server inside VS Code, so you catch drift, type errors, and contract violations where you write the code, not just in CI.

Your .rocky models compile to SQL live as you type, with type-aware hover, inline column types, and go-to-definition across the graph.

A Rocky DSL model on the left and its compiled SQL on the right, updating live as you type

The Inspector turns any model into a trust dashboard: schema, column-level lineage, tests, per-model cost, and a governance card that flags classified columns and unmasked PII.

The Rocky Inspector's Overview as a model trust dashboard, its Governance card flagging two classified columns with one left unmasked

Install the VS Code extension →

Where Rocky is today

The trust primitives (compiler, branches, replay, lineage, contracts, cost attribution) are production-grade on Databricks. The rest is in progress:

If those gaps are blockers for your team, open a discussion. The roadmap is shaped by where production pipelines are actually getting hurt.

How it compares to dbt Core

Disaster | What dbt Core does | What Rocky does

Upstream changes a column type | Silent; surfaces as a downstream failure later | E013 at compile, blocks the PR Required column dropped from a contract | Caught at build time via contract: enforced | E010 at compile, blocks the PR Column rename with unknown blast radius | dbt docs is post-hoc and table-level; dbt Cloud Enterprise adds column lineage in the UI, also post-hoc and not PR-blocking | rocky lineage-diff at PR time, column-level, downstream consumers listed, blocks the merge SELECT * pulls a new column you didn't expect | Silent | P002 warning, downstream consumers named Snowflake-only function written for a Databricks project | No dialect-portability lint; runs in dev, fails in prod | P001 dialect-portability lint at compile Run cost doubles, no one knows which model | No per-model cost attribution; reconstruct it from warehouse query history | RunOutput.cost_summary per model, every run Auditor asks: who changed fct_revenue.amount, when, and why? | Run history in dbt Cloud, but no content-addressed record of code and output | rocky replay <run_id>: a content-addressed record of the exact code and the output it produced Sev-2 at 3 AM, half the pipeline already ran | dbt retry resumes from the failed model; no within-run checkpoint or circuit breaker | rocky run --resume-latest: checkpoint, three-state circuit breaker, skip what succeeded

Each row is a real failure mode and a Rocky command that turns it into a non-event. The same primitives back every row: typed compiler, content-addressed state, column-level lineage, per-model cost.

dbt Core defined this category, and rocky import-dbt converts a vanilla dbt project to Rocky in one command. In June 2026 dbt Labs open-sourced the Fusion runtime as dbt Core v2.0 (Rust, Apache 2.0, alpha); the recommended Fusion distribution adds SQL type-checking and column-level lineage on top, though it still templates with Jinja and its build-failing strict checks are opt-in. Neither dbt Core v2.0 nor Fusion ships named branches, a content-addressed run record, per-model cost as a first-class column, a cross-warehouse dialect lint, or declarative RBAC and masking; dbt's governance and cost features live in its paid platform, while Rocky's are open source under Apache 2.0.

Subprojects

Path | Artifact | Language | Description

engine/ | rocky CLI binary | Rust | Core SQL transformation engine, 23-crate Cargo workspace integrations/dagster/ | dagster-rocky PyPI wheel | Python | Dagster resource and component wrapping the Rocky CLI editors/vscode/ | Rocky VSIX | TypeScript | VS Code extension; LSP client + commands for AI features examples/playground/ | (config only) | TOML / SQL | Self-contained DuckDB sample pipeline used for smoke tests and benchmarks

Each subproject has its own README with detailed usage. The engine/README.md is the canonical product reference for the Rocky CLI.

Adapters

Role | Adapter | Status | Notes

Warehouse | Databricks | Production | SQL Statement API · Unity Catalog · schema-prefix branches (SHALLOW CLONE is a follow-up) Warehouse | Snowflake | Beta | REST connector · GRANT/REVOKE reconciliation · schema-prefix branches (zero-copy CLONE is a follow-up) Warehouse | BigQuery | Beta | REST connector · schema-prefix branches Warehouse | DuckDB | Local / Testing | Embedded · powers rocky playground (no credentials needed) Warehouse | Trino | Beta | REST /v1/statement polling client · Basic + JWT auth · Docker conformance harness behind trino-conformance feature Source | Fivetran | Production | REST connector + table discovery Source | Airbyte | Beta | Catalog discovery Source | Iceberg | Beta | REST catalog discovery of namespaces and tables Source | Manual | Production | Schema/table lists inline in rocky.toml

Building a warehouse Rocky doesn't ship in-tree (ClickHouse, Redshift, …)? See the Adapter SDK guide and the Rust-native skeleton POC.

Building from source

git clone https://github.com/rocky-data/rocky.git cd rocky just build # builds engine + dagster wheel + vscode extension just test # runs all test suites just lint # cargo clippy/fmt + ruff + eslint

just is optional; you can also build each subproject directly. See CONTRIBUTING.md for per-subproject build commands.

Releases

Each artifact is released independently using a tag-namespaced scheme:

See CONTRIBUTING.md for the full release flow.

Documentation

Full documentation lives at rocky-data.dev : concepts, guides, CLI reference, Dagster integration, and the adapter SDK.

Contributing

See CONTRIBUTING.md. Before opening a PR, please read the cross-project change guidance: schema and DSL changes must update consumers atomically.

Sponsoring

Rocky is free and open source. If it saves your team time, consider sponsoring the project so development can continue.

License

Apache 2.0