Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
44 commits
Select commit Hold shift + click to select a range
b6aa62d
Added support for the database engine plugin system for extending sql…
asmyasnikov Dec 28, 2025
5336821
Fix of endtoend tests
asmyasnikov Dec 28, 2025
2b88994
added install plugin-based-codegen's
asmyasnikov Dec 28, 2025
b1d156d
remove tmp file
asmyasnikov Dec 28, 2025
9f65d4f
removed go.{mod,sum}
asmyasnikov Dec 28, 2025
74b621f
SQLCDEBUG=processplugins=1
asmyasnikov Dec 28, 2025
cede5d3
Fix
asmyasnikov Dec 28, 2025
15b240d
Fix
asmyasnikov Dec 28, 2025
0b3b165
Apply suggestions from code review
asmyasnikov Dec 28, 2025
6c5b9a6
revert Combine
asmyasnikov Dec 28, 2025
7609ebc
.gitignore + README
asmyasnikov Jan 10, 2026
2c74313
simplified engine API
asmyasnikov Jan 27, 2026
88e6082
Apply suggestions from code review
asmyasnikov Jan 27, 2026
f39ae4a
Delete protos/engine/engine_grpc.pb.go
asmyasnikov Jan 27, 2026
18f5368
Delete protos/engine/engine.pb.go
asmyasnikov Jan 27, 2026
8eaef3c
Delete pkg/plugin/sdk.go
asmyasnikov Jan 27, 2026
fbaf6ba
Delete pkg/engine/engine.pb.go
asmyasnikov Jan 27, 2026
ce385ae
Delete pkg/plugin/codegen.pb.go
asmyasnikov Jan 27, 2026
a024d3e
Delete examples/plugin-based-codegen/README.md
asmyasnikov Jan 27, 2026
fbd5b43
Delete examples/plugin-based-codegen/gen/rust/queries.rs
asmyasnikov Jan 27, 2026
e6a730a
docs
asmyasnikov Jan 27, 2026
c8831c7
removed example
asmyasnikov Jan 27, 2026
6d5770f
fix
asmyasnikov Jan 27, 2026
d2417e8
Update .gitignore
asmyasnikov Jan 27, 2026
c50e9c7
pb.go
asmyasnikov Jan 27, 2026
e9cc264
fix comments
asmyasnikov Jan 27, 2026
ad7bf6c
simplified plugin engine code
asmyasnikov Jan 27, 2026
5d4c8dd
sourceFiles
asmyasnikov Jan 27, 2026
131d7bb
fix
asmyasnikov Jan 27, 2026
048a64d
Apply suggestions from code review
asmyasnikov Jan 27, 2026
79621b0
removed temp file
asmyasnikov Jan 27, 2026
d9df83b
Apply suggestions from code review
asmyasnikov Jan 27, 2026
55760fc
Apply suggestions from code review
asmyasnikov Jan 27, 2026
96dfabd
Apply suggestions from code review
asmyasnikov Jan 27, 2026
85475e2
removed engine interface
asmyasnikov Jan 27, 2026
0f81f5d
merge files
asmyasnikov Jan 27, 2026
7800a42
move md doc
asmyasnikov Jan 27, 2026
f6b34f0
Apply suggestions from code review
asmyasnikov Jan 27, 2026
830767e
revert changes
asmyasnikov Jan 27, 2026
e4667d2
revert
asmyasnikov Jan 27, 2026
9b9b3ed
docs
asmyasnikov Jan 27, 2026
a8fec25
fix
asmyasnikov Jan 27, 2026
778b45c
fixes and tests
asmyasnikov Jan 28, 2026
fb7e9a6
change ParseResponse - returns multiple statements from single call
asmyasnikov Jan 28, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,12 @@ mysqlsh:
proto:
buf generate

proto-engine-plugin:
protoc -I. \
--go_out=. --go_opt=module=github.com/sqlc-dev/sqlc --go_opt=Mprotos/engine/engine.proto=github.com/sqlc-dev/sqlc/pkg/engine \
--go-grpc_out=. --go-grpc_opt=module=github.com/sqlc-dev/sqlc --go-grpc_opt=Mprotos/engine/engine.proto=github.com/sqlc-dev/sqlc/pkg/engine \
protos/engine/engine.proto

remote-proto:
protoc \
--go_out=. --go_opt="Minternal/remote/gen.proto=github.com/sqlc-dev/sqlc/internal/remote" --go_opt=module=github.com/sqlc-dev/sqlc \
Expand Down
8 changes: 8 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,14 @@ Check out [an interactive example](https://play.sqlc.dev/) to see it in action,

Additional languages can be added via [plugins](https://docs.sqlc.dev/en/latest/reference/language-support.html#community-language-support).

## Supported database engines

- PostgreSQL
- MySQL
- SQLite

Additional database engines can be added via [engine plugins](https://docs.sqlc.dev/en/latest/guides/engine-plugins.html).

## Sponsors

Development is possible thanks to our sponsors. If you would like to support sqlc,
Expand Down
210 changes: 210 additions & 0 deletions docs/guides/engine-plugins.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,210 @@
# External Database Engines (Engine Plugins)

Engine plugins let you use sqlc with databases that are not built-in. You can add support for other SQL-compatible systems (e.g. CockroachDB, TiDB, or custom engines) by implementing a small external program that parses SQL and returns parameters and result columns.

## Why use an engine plugin?

- Use sqlc with a database that doesn't have native support.
- Reuse an existing SQL parser or dialect in a separate binary.
- Keep engine-specific logic outside the sqlc core.

Data returned by the engine plugin (SQL text, parameters, columns) is passed through to [codegen plugins](plugins.md) without an extra compiler/AST step. The plugin is the single place that defines how queries are interpreted for that engine.

**Limitation:** `sqlc vet` does not support plugin engines. Use vet only with built-in engines (postgresql, mysql, sqlite).

## Overview

An engine plugin is an external process that implements one RPC:

- **Parse** — accepts the **entire contents** of one query file (e.g. `query.sql`) and either schema SQL or connection parameters; returns **one Statement per query block** in that file (each with sql, parameters, columns, and name/cmd).

Process plugins (e.g. written in Go) talk to sqlc over **stdin/stdout** using **Protocol Buffers**. The protocol is defined in `protos/engine/engine.proto`.

## Compatibility

For Go plugins, compatibility is enforced at **compile time** by importing the engine package:

```go
import "github.com/sqlc-dev/sqlc/pkg/engine"
```

- If the plugin builds, it matches this version of the engine API.
- If the API changes in a breaking way, the plugin stops compiling until it's updated.

No version handshake is required; the proto schema defines the contract.

## Configuration

### sqlc.yaml

```yaml
version: "2"

engines:
- name: mydb
process:
cmd: sqlc-engine-mydb
env:
- MYDB_DSN

sql:
- engine: mydb
schema: "schema.sql"
queries: "queries.sql"
codegen:
- plugin: go
out: db
```

### Engine options

| Field | Description |
|-------|-------------|
| `name` | Engine name used in `sql[].engine` |
| `process.cmd` | Command to run (PATH or absolute path) |
| `env` | Environment variable names passed to the plugin |

Each engine must define either `process` (with `cmd`) or `wasm` (with `url` and `sha256`). See [Configuration reference](../reference/config.md) for the full `engines` schema.

### How sqlc finds the process plugin

For an engine with `process.cmd`, sqlc resolves and runs the plugin as follows:

1. **Command parsing** — `process.cmd` is split on whitespace. The first token is the executable; any further tokens are passed as arguments, and sqlc appends the RPC method name (`parse`) when invoking the plugin.

2. **Executable lookup** — The first token is resolved the same way as in the shell:
- If it contains a path separator (e.g. `/usr/bin/sqlc-engine-mydb` or `./bin/sqlc-engine-mydb`), it is treated as a path. Absolute paths are used as-is; relative paths are taken relative to the **current working directory of the process running sqlc**.
- If it has no path separator, the executable is looked up in the **PATH** of the process running sqlc. The plugin binary must be on PATH (e.g. after `go install` or adding its directory to PATH) or `process.cmd` must be an absolute path.

3. **Working directory** — The plugin process is started with its working directory set to the **directory containing the sqlc config file**. That directory is used for resolving relative paths inside the plugin, not for resolving `process.cmd` itself.

If the executable cannot be found or `process.cmd` is empty, sqlc reports an error and refers to this documentation.

## Implementing an engine plugin (Go)

### 1. Dependencies and entrypoint

```go
package main

import "github.com/sqlc-dev/sqlc/pkg/engine"

func main() {
engine.Run(engine.Handler{
PluginName: "mydb",
PluginVersion: "1.0.0",
Parse: handleParse,
})
}
```

The engine API exposes only **Parse**. There are no separate methods for catalog, keywords, comment syntax, or dialect.

### 2. Parse

sqlc calls Parse **once per query file** (e.g. once for `query.sql`). The plugin receives the full file contents and returns one **Statement** per query block in that file. sqlc then passes each statement to the codegen plugin as a separate query.

**Request**

- `sql` — The **entire contents** of one query file (all query blocks, with `-- name: X :one`-style comments).
- `schema_source` — One of:
- `schema_sql`: full schema as in schema.sql (for schema-based parsing).
- `connection_params`: DSN and options for database-only mode.

**Response**

Return `statements`: one `Statement` per query block. Each `Statement` has:

- `name` — Query name (from `-- name: GetUser` etc.).
- `cmd` — Command/type: use the `Cmd` enum (`engine.Cmd_CMD_ONE`, `engine.Cmd_CMD_MANY`, `engine.Cmd_CMD_EXEC`, etc.). See `protos/engine/engine.proto` for the full list.
- `sql` — Processed SQL for that block (as-is or with `*` expanded using schema).
- `parameters` — Parameters for this statement.
- `columns` — Result columns (names, types, nullability, etc.) for this statement.

The engine package provides helpers (optional) to split `query.sql` and parse `"-- name: X :cmd"` lines in the same way as the built-in engines:

- `engine.CommentSyntax` — Which comment styles to accept (`Dash`, `SlashStar`, `Hash`).
- `engine.ParseNameAndCmd(line, syntax)` — Parses a single line like `"-- name: ListAuthors :many"` → `(name, cmd, ok)`. `cmd` is `engine.Cmd`.
- `engine.QueryBlocks(content, syntax)` — Splits file content into `[]engine.QueryBlock` (each has `Name`, `Cmd`, `SQL`).
- `engine.StatementMeta(name, cmd, sql)` — Builds a `*engine.Statement` with name/cmd/sql set; you add parameters and columns.

Example handler using helpers:

```go
func handleParse(req *engine.ParseRequest) (*engine.ParseResponse, error) {
queryFileContent := req.GetSql()
syntax := engine.CommentSyntax{Dash: true, SlashStar: true, Hash: true}

var schema *SchemaInfo
if s := req.GetSchemaSql(); s != "" {
schema = parseSchema(s)
}
// Or use req.GetConnectionParams() for database-only mode.

blocks, _ := engine.QueryBlocks(queryFileContent, syntax)
var statements []*engine.Statement
for _, b := range blocks {
st := engine.StatementMeta(b.Name, b.Cmd, processSQL(b.SQL, schema))
st.Parameters = extractParameters(b.SQL)
st.Columns = extractColumns(b.SQL, schema)
statements = append(statements, st)
}
return &engine.ParseResponse{Statements: statements}, nil
}
```

Parameter and column types use the `Parameter` and `Column` messages in `engine.proto` (name, position, data_type, nullable, is_array, array_dims; for columns, table_name and schema_name are optional).

Support for sqlc placeholders (`sqlc.arg()`, `sqlc.narg()`, `sqlc.slice()`, `sqlc.embed()`) is up to the plugin: it can parse and map them into `parameters` (and schema usage) as needed.

### 3. Build and run

```bash
go build -o sqlc-engine-mydb .
# Ensure sqlc-engine-mydb is on PATH or use an absolute path in process.cmd
```

## Protocol

Process plugins use Protocol Buffers on stdin/stdout:

```
sqlc → stdin (protobuf) → plugin → stdout (protobuf) → sqlc
```

Invocation:

```bash
sqlc-engine-mydb parse # stdin: ParseRequest, stdout: ParseResponse
```

The definition lives in `protos/engine/engine.proto` (generated Go in `pkg/engine`). After editing the proto, run `make proto-engine-plugin` to regenerate the Go code.

## Example

The protocol and Go SDK are in this repository: `protos/engine/engine.proto` and `pkg/engine/` (including `sdk.go` with `engine.Run` and `engine.Handler`). Use them to build a binary that implements the Parse RPC; register it under `engines` in sqlc.yaml as shown above.

## Architecture

For each `sql[]` block, `sqlc generate` branches on the configured engine: built-in (postgresql, mysql, sqlite) use the compiler and catalog; any engine listed under `engines:` in sqlc.yaml uses the plugin path (no compiler). For the plugin path, sqlc calls Parse **once per query file**, sending the full file contents and schema (or connection params). The plugin returns **N statements** (one per query block); sqlc passes each statement to codegen as a separate query.

```
┌─────────────────────────────────────────────────────────────────┐
│ sqlc generate (plugin engine) │
│ 1. Per query file: one Parse(schema_sql|connection_params, │
│ full query file content) │
│ 2. ParseResponse.statements = one Statement per query block │
│ 3. Each statement → one codegen query (N helpers) │
└─────────────────────────────────────────────────────────────────┘

sqlc sqlc-engine-mydb
│──── spawn, args: ["parse"] ──────────────────────────────► │
│──── stdin: ParseRequest{sql=full query.sql, schema_sql|…} ► │
│◄─── stdout: ParseResponse{statements: [stmt1, stmt2, …]} ── │
```

## See also

- [Codegen plugins](plugins.md) — Custom code generators that consume engine output.
- [Configuration reference](../reference/config.md)
- Proto schema: `protos/engine/engine.proto`
1 change: 1 addition & 0 deletions docs/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -96,5 +96,6 @@ code ever again.
howto/ci-cd.md
guides/using-go-and-pgx.rst
guides/plugins.md
guides/engine-plugins.md
guides/development.md
guides/privacy.md
Loading
Loading