Fix #1322

sqlc-dev · kyleconroy · Oct 16, 2023 · Oct 13, 2023 · Oct 13, 2023 · Oct 13, 2023
commit 3c14abadca42b2ead16e5cd334d40e381a1a42ce
diff --git a/internal/endtoend/testdata/func_return_table/issue.md b/internal/endtoend/testdata/func_return_table/issue.md
@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/1322
diff --git a/internal/endtoend/testdata/func_return_table/postgresql/pgx/exec.json b/internal/endtoend/testdata/func_return_table/postgresql/pgx/exec.json
@@ -0,0 +1,3 @@
+{
+  "contexts": ["managed-db"]
+}
diff --git a/internal/endtoend/testdata/func_return_table/postgresql/pgx/go/db.go b/internal/endtoend/testdata/func_return_table/postgresql/pgx/go/db.go
diff --git a/internal/endtoend/testdata/func_return_table/postgresql/pgx/go/models.go b/internal/endtoend/testdata/func_return_table/postgresql/pgx/go/models.go
diff --git a/internal/endtoend/testdata/func_return_table/postgresql/pgx/go/query.sql.go b/internal/endtoend/testdata/func_return_table/postgresql/pgx/go/query.sql.go
diff --git a/internal/endtoend/testdata/func_return_table/postgresql/pgx/query.sql b/internal/endtoend/testdata/func_return_table/postgresql/pgx/query.sql
@@ -0,0 +1,2 @@
+-- name: Foo :one
+SELECT * FROM register_account('a', 'b');
diff --git a/internal/endtoend/testdata/func_return_table/postgresql/pgx/schema.sql b/internal/endtoend/testdata/func_return_table/postgresql/pgx/schema.sql
@@ -0,0 +1,27 @@
+CREATE TABLE accounts (
+  id         INTEGER GENERATED ALWAYS AS IDENTITY PRIMARY KEY,
+  username   TEXT NOT NULL UNIQUE,
+  password   TEXT NOT NULL
+);
+
+-- this is a useless and horrifying function cause we don't hash
+-- the password, this is just to repro the bug in sqlc
+CREATE OR REPLACE FUNCTION register_account(
+    _username TEXT,
+    _password VARCHAR(70)
+)
+RETURNS TABLE (
+    account_id   INTEGER
+)
+AS $$
+BEGIN
+  INSERT INTO accounts (username, password)
+       VALUES (
+         _username,
+         _password
+       )
+    RETURNING id INTO account_id;
+
+  RETURN NEXT;
+END;
+$$ LANGUAGE plpgsql;
diff --git a/internal/endtoend/testdata/func_return_table/postgresql/pgx/sqlc.yaml b/internal/endtoend/testdata/func_return_table/postgresql/pgx/sqlc.yaml
@@ -0,0 +1,10 @@
+version: "2"
+sql:
+  - engine: "postgresql"
+    schema: "schema.sql"
+    queries: "query.sql"
+    gen:
+      go:
+        package: "querytest"
+        out: "go"
+        sql_package: "pgx/v5"
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		-- name: Foo :one
		SELECT * FROM register_account('a', 'b');