Add the ability to fully customize jet generated files.

generator/postgres/query_set.go

@@ -1,81 +1,83 @@
 package postgres
 
 import (
+	"context"
 	"database/sql"
-	"github.com/go-jet/jet/v2/generator/internal/metadata"
-	"github.com/go-jet/jet/v2/internal/utils"
+	"github.com/go-jet/jet/v2/generator/metadata"
+	"github.com/go-jet/jet/v2/internal/utils/throw"
+	"github.com/go-jet/jet/v2/qrm"
 )
 
 // postgresQuerySet is dialect query set for PostgreSQL
 type postgresQuerySet struct{}
 
-func (p *postgresQuerySet) ListOfTablesQuery() string {
-	return `
-SELECT table_name 
+func (p postgresQuerySet) GetTablesMetaData(db *sql.DB, schemaName string, tableType metadata.TableType) []metadata.Table {
+	query := `
+SELECT table_name as "table.name" 
 FROM information_schema.tables
-where table_schema = $1 and table_type = $2;
+WHERE table_schema = $1 and table_type = $2;
 `
+	var tables []metadata.Table
+
+	err := qrm.Query(context.Background(), db, query, []interface{}{schemaName, tableType}, &tables)
+	throw.OnError(err)
+
+	for i := range tables {
+		tables[i].Columns = p.GetTableColumnsMetaData(db, schemaName, tables[i].Name)
+	}
+
+	return tables
 }
 
-func (p *postgresQuerySet) PrimaryKeysQuery() string {
-	return `
-SELECT c.column_name
-FROM information_schema.key_column_usage AS c
-LEFT JOIN information_schema.table_constraints AS t
-ON t.constraint_name = c.constraint_name
-WHERE t.table_schema = $1 AND t.table_name = $2 AND t.constraint_type = 'PRIMARY KEY';
-`
-}
-
-func (p *postgresQuerySet) ListOfColumnsQuery() string {
-	return `
-SELECT column_name, is_nullable, data_type, udt_name, FALSE
-FROM information_schema.columns
+func (p postgresQuerySet) GetTableColumnsMetaData(db *sql.DB, schemaName string, tableName string) []metadata.Column {
+	query := `
+WITH primaryKeys AS (
+	SELECT column_name
+	FROM information_schema.key_column_usage AS c
+		LEFT JOIN information_schema.table_constraints AS t
+		ON t.constraint_name = c.constraint_name
+	WHERE t.table_schema = $1 AND t.table_name = $2 AND t.constraint_type = 'PRIMARY KEY'
+)
+SELECT column_name as "column.Name", 
+	   is_nullable = 'YES' as "column.isNullable",
+       (EXISTS(SELECT 1 from primaryKeys as pk where pk.column_name = columns.column_name)) as "column.IsPrimaryKey",
+	   dataType.kind as "dataType.Kind",	
+	   (case dataType.Kind when 'base' then data_type else LTRIM(udt_name, '_') end) as "dataType.Name", 
+	   FALSE as "dataType.isUnsigned"
+FROM information_schema.columns,
+	 LATERAL (select (case data_type
+				when 'ARRAY' then 'array'
+				when 'USER-DEFINED' then 
+					case (select typtype from pg_type where typname = columns.udt_name)
+						when 'e' then 'enum'
+						else 'user-defined'
+					end
+				else 'base'
+			end) as Kind) as dataType
 where table_schema = $1 and table_name = $2
-order by ordinal_position;`
+order by ordinal_position;
+`
+	var columns []metadata.Column
+	err := qrm.Query(context.Background(), db, query, []interface{}{schemaName, tableName}, &columns)
+	throw.OnError(err)
+
+	return columns
 }
 
-func (p *postgresQuerySet) ListOfEnumsQuery() string {
-	return `
-SELECT t.typname,  
-       e.enumlabel
+func (p postgresQuerySet) GetEnumsMetaData(db *sql.DB, schemaName string) []metadata.Enum {
+	query := `
+SELECT t.typname as "enum.name",  
+       e.enumlabel as "values"
 FROM pg_catalog.pg_type t 
    JOIN pg_catalog.pg_enum e on t.oid = e.enumtypid  
    JOIN pg_catalog.pg_namespace n ON n.oid = t.typnamespace
 WHERE n.nspname = $1
 ORDER BY n.nspname, t.typname, e.enumsortorder;`
-}
-
-func (p *postgresQuerySet) GetEnumsMetaData(db *sql.DB, schemaName string) []metadata.MetaData {
-	rows, err := db.Query(p.ListOfEnumsQuery(), schemaName)
-	utils.PanicOnError(err)
-	defer rows.Close()
-
-	enumsInfosMap := map[string][]string{}
-	for rows.Next() {
-		var enumName string
-		var enumValue string
-		err = rows.Scan(&enumName, &enumValue)
-		utils.PanicOnError(err)
-
-		enumValues := enumsInfosMap[enumName]
-
-		enumValues = append(enumValues, enumValue)
-
-		enumsInfosMap[enumName] = enumValues
-	}
-
-	err = rows.Err()
-	utils.PanicOnError(err)
-
-	ret := []metadata.MetaData{}
-
-	for enumName, enumValues := range enumsInfosMap {
-		ret = append(ret, metadata.EnumMetaData{
-			EnumName: enumName,
-			Values:   enumValues,
-		})
-	}
-
-	return ret
+
+	var result []metadata.Enum
+
+	err := qrm.Query(context.Background(), db, query, []interface{}{schemaName}, &result)
+	throw.OnError(err)
+
+	return result
 }