Fix PG functions

Author: Arthur-Sk

internal/compiler/output_columns.go

@@ -599,8 +599,7 @@ func (c *Compiler) sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, erro
 								DataType: arg.Type.Name,
 							})
 						}
-					}
-					if fn.ReturnType != nil {
+					} else if fn.ReturnType != nil {
 						table.Columns = []*Column{
 							{
 								Name:     colName,

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

@@ -1,2 +1,5 @@
 -- name: SelectAll :many
 SELECT * FROM public.get_test();
+
+-- name: SelectWithTime :many
+SELECT * FROM public.get_test($1::timestamp);

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

@@ -43,7 +43,7 @@ func (f *Function) OutArgs() []*Argument {
 	var args []*Argument
 	for _, a := range f.Args {
 		switch a.Mode {
-		case ast.FuncParamOut:
+		case ast.FuncParamOut, ast.FuncParamTable:
 			args = append(args, a)
 		}
 	}