Generator cleanup.

generator/internal/metadata/postgres-metadata/enum_info.go

@@ -0,0 +1,66 @@
+package postgres_metadata
+
+import (
+	"database/sql"
+	"github.com/go-jet/jet/generator/internal/metadata"
+)
+
+type EnumInfo struct {
+	name   string
+	Values []string
+}
+
+func (e EnumInfo) Name() string {
+	return e.name
+}
+
+func getEnumInfos(db *sql.DB, schemaName string) ([]metadata.MetaData, 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;`
+
+	rows, err := db.Query(query, 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 := []metadata.MetaData{}
+
+	for enumName, enumValues := range enumsInfosMap {
+		ret = append(ret, EnumInfo{
+			enumName,
+			enumValues,
+		})
+	}
+
+	return ret, nil
+}