Fix PG procedure output type for separate schema

Author: Arthur-Sk

internal/compiler/output_columns.go

@@ -71,6 +71,8 @@ func (c *Compiler) outputColumns(qc *QueryCatalog, node ast.Node) ([]*Column, er
 				typeName := arg.Type.Name
 				if arg.Type.Names != nil && len(arg.Type.Names.Items) > 0 {
 					typeName = astutils.Join(arg.Type.Names, ".")
+				} else if arg.Type.Schema != "" {
+					typeName = arg.Type.Schema + "." + arg.Type.Name
 				}
 				cols = append(cols, &Column{Name: name, DataType: typeName, NotNull: false})
 			}

internal/endtoend/testdata/ddl_create_procedure_with_out_args/postgresql/pgx/v5/go/query.sql.go

@@ -1,11 +1,25 @@
 -- name: CallInsertData :one
-CALL insert_data($1, $2, null);
+CALL insert_data($1, $2, null, null, null, null, null, null, null, null, null, null, null);
 
--- name: CallInsertDataNoArgs :exec
-CALL insert_data(1, 2, null);
+-- name: CallInsertDataNoArgs :one
+CALL insert_data(1, 2, null, null, null, null, null, null, null, null, null, null, null);
 
 -- name: CallInsertDataNamed :one
-CALL insert_data(b => $1, a => $2, c => null);
+CALL insert_data(
+        b => $1,
+        a => $2,
+        c => null,
+        i => null,
+        j => null,
+        k => null,
+        d => null,
+        h => null,
+        e => null,
+        m => null,
+        f => null,
+        g => null,
+        l => null
+     );
 
--- name: CallInsertDataSqlcArgs :exec
-CALL insert_data(sqlc.arg('foo'), sqlc.arg('bar'), sqlc.arg('с'));
+-- name: CallInsertDataSqlcArgs :one
+CALL insert_data(sqlc.arg('foo'), sqlc.arg('bar'), null, null, null, null, null, null, null, null, null, null, null);

internal/endtoend/testdata/ddl_create_procedure_with_out_args/postgresql/pgx/v5/query.sql

@@ -1,15 +1,55 @@
-CREATE TABLE tbl (
-        value integer
+CREATE TABLE tbl
+(
+    value integer
 );
 
 -- https://www.postgresql.org/docs/current/sql-createprocedure.html
-CREATE PROCEDURE insert_data(IN a integer, IN b integer, OUT c integer)
+CREATE PROCEDURE insert_data(
+    IN a integer,
+    IN b integer,
+    -- Numbers
+    OUT c integer,
+    OUT i float,
+    OUT j numeric,
+    OUT k real,
+    -- Text
+    OUT d varchar,
+    OUT h text,
+    -- Time
+    OUT e timestamp,
+    OUT m interval,
+    -- Other
+    OUT f jsonb,
+    OUT g bytea,
+    OUT l boolean
+)
     LANGUAGE plpgsql
-AS $$
+AS
+$$
 BEGIN
     INSERT INTO tbl VALUES (a);
     INSERT INTO tbl VALUES (b);
 
     c := 777;
+
+    -- Numbers assignments
+    i := random() * 100;
+    j := (random() * 500)::numeric(10, 2);
+    k := (random() * 50)::real;
+
+    -- Text assignments
+    d := 'Varchar val ' || floor(random() * 100);
+    h := 'Text val ' || md5(random()::text);
+
+    -- Time assignments
+    e := now();
+    m := make_interval(hours => floor(random() * 24)::int);
+
+    -- Other assignments
+    f := '{
+      "key": "random"
+    }'::jsonb;
+    g := '\xDEADBEEF'::bytea;
+    l := (random() > 0.5);
 END;
 $$;