Add support for database enum types.

generator/metadata/column_info.go

@@ -10,6 +10,7 @@ type ColumnInfo struct {
 	Name       string
 	IsNullable bool
 	DataType   string
+	EnumName   string
 	TableInfo  *TableInfo
 }
 
@@ -65,6 +66,8 @@ func (c ColumnInfo) GoBaseType() string {
 		return snaker.SnakeToCamel(forignKeyTable)
 	} else {
 		switch c.DataType {
+		case "USER-DEFINED":
+			return c.EnumName
 		case "boolean":
 			return "bool"
 		case "smallint":
@@ -105,7 +108,7 @@ func (c ColumnInfo) ToGoFieldName() string {
 func fetchColumnInfos(db *sql.DB, tableInfo *TableInfo) ([]ColumnInfo, error) {
 
 	query := `
-SELECT column_name, is_nullable, data_type 
+SELECT column_name, is_nullable, data_type, udt_name
 FROM information_schema.columns
 where table_schema = $1 and table_name = $2
 order by ordinal_position;`
@@ -124,7 +127,7 @@ order by ordinal_position;`
 	for rows.Next() {
 		columnInfo := ColumnInfo{}
 		var isNullable string
-		err := rows.Scan(&columnInfo.Name, &isNullable, &columnInfo.DataType)
+		err := rows.Scan(&columnInfo.Name, &isNullable, &columnInfo.DataType, &columnInfo.EnumName)
 
 		columnInfo.IsNullable = isNullable == "YES"
 

generator/metadata/database_info.go

@@ -8,6 +8,7 @@ type DatabaseInfo struct {
 	DatabaseName string
 	SchemaName   string
 	TableInfos   []TableInfo
+	EnumInfos    []EnumInfo
 }
 
 func GetDatabaseInfo(db *sql.DB, databaseName, schemaName string) (*DatabaseInfo, error) {
@@ -16,6 +17,7 @@ func GetDatabaseInfo(db *sql.DB, databaseName, schemaName string) (*DatabaseInfo
 		databaseName,
 		schemaName,
 		[]TableInfo{},
+		[]EnumInfo{},
 	}
 
 	var err error
@@ -25,5 +27,11 @@ func GetDatabaseInfo(db *sql.DB, databaseName, schemaName string) (*DatabaseInfo
 		return nil, err
 	}
 
+	databaseInfo.EnumInfos, err = fetchEnumInfos(db, databaseInfo)
+
+	if err != nil {
+		return nil, err
+	}
+
 	return databaseInfo, nil
 }

generator/metadata/enum_info.go

@@ -0,0 +1,70 @@
+package metadata
+
+import (
+	"database/sql"
+	"fmt"
+)
+
+type EnumInfo struct {
+	Name   string
+	Values []string
+}
+
+func (e *EnumInfo) goValueName(index int) {
+	return
+}
+
+func fetchEnumInfos(db *sql.DB, databaseInfo *DatabaseInfo) ([]EnumInfo, error) {
+	query := `
+SELECT t.typname,  
+       e.enumlabel
+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;`
+
+	//fmt.Println(query, schemaName)
+
+	rows, err := db.Query(query, &databaseInfo.SchemaName)
+
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	enumsInfosMap := map[string][]string{}
+	for rows.Next() {
+		var enumName string
+		var enumValue string
+		err = rows.Scan(&enumName, &enumValue)
+		if err != nil {
+			return nil, err
+		}
+
+		enumValues := enumsInfosMap[enumName]
+
+		enumValues = append(enumValues, enumValue)
+
+		enumsInfosMap[enumName] = enumValues
+	}
+
+	err = rows.Err()
+
+	if err != nil {
+		return nil, err
+	}
+
+	ret := []EnumInfo{}
+
+	for enumName, enumValues := range enumsInfosMap {
+		ret = append(ret, EnumInfo{
+			enumName,
+			enumValues,
+		})
+	}
+
+	fmt.Println("FOUND", len(ret), " enums")
+
+	return ret, nil
+}