Fix #1322

Author: kyleconroy

internal/endtoend/testdata/func_return_table/issue.md

@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/1322

internal/endtoend/testdata/func_return_table/postgresql/pgx/exec.json

@@ -0,0 +1,3 @@
+{
+  "contexts": ["managed-db"]
+}

internal/endtoend/testdata/func_return_table/postgresql/pgx/go/db.go

@@ -0,0 +1,2 @@
+-- name: Foo :one
+SELECT * FROM register_account('a', 'b');

internal/endtoend/testdata/func_return_table/postgresql/pgx/go/models.go

@@ -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;

internal/endtoend/testdata/func_return_table/postgresql/pgx/go/query.sql.go

@@ -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"