Add single query preparation

Fixes #2905
sqlc-dev · Jan 18, 2024 · 7526980 · 7526980
1 parent b4f03e9
commit 7526980
Show file tree

Hide file tree

Showing 7 changed files with 340 additions and 0 deletions.
diff --git a/examples/ondeck/mysql/db.go b/examples/ondeck/mysql/db.go
diff --git a/examples/ondeck/postgresql/db.go b/examples/ondeck/postgresql/db.go
diff --git a/examples/ondeck/sqlite/db.go b/examples/ondeck/sqlite/db.go
diff --git a/internal/codegen/golang/templates/stdlib/dbCode.tmpl b/internal/codegen/golang/templates/stdlib/dbCode.tmpl
@@ -88,6 +88,18 @@ type Queries struct {
 	{{- end}}
 	{{- end}}
 }
+{{if .EmitPreparedQueries -}}
+{{range .GoQueries}}
+
+func (q *Queries) Prepare{{.MethodName}}(ctx context.Context) error {
+        var err error
+	if q.{{.FieldName}}, err = q.db.PrepareContext(ctx, {{.ConstantName}}); err != nil {
+		return fmt.Errorf("error preparing query {{.MethodName}}: %w", err)
+	}
+        return nil
+}
+{{- end}}
+{{- end}}
 
 {{if not .EmitMethodsWithDBArgument}}
 func (q *Queries) WithTx(tx *sql.Tx) *Queries {

diff --git a/internal/endtoend/testdata/prepared_queries/mysql/go/db.go b/internal/endtoend/testdata/prepared_queries/mysql/go/db.go
diff --git a/internal/endtoend/testdata/prepared_queries/postgresql/stdlib/go/db.go b/internal/endtoend/testdata/prepared_queries/postgresql/stdlib/go/db.go
diff --git a/internal/endtoend/testdata/sqlc_slice_prepared/sqlite/go/db.go b/internal/endtoend/testdata/sqlc_slice_prepared/sqlite/go/db.go