Gleipnir/jet
4
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add support for database enum types.

Browse source
This commit is contained in:
zer0sub 2019-04-03 19:21:46 +02:00
parent 273bf1ed4c
commit 2c7a9f5058
9 changed files with 202 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

7
generator/metadata/column_info.go
View file

@ -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"