146 lines
2.8 KiB
Go
146 lines
2.8 KiB
Go
package metadata
|
|
|
|
import (
|
|
"database/sql"
|
|
"github.com/serenize/snaker"
|
|
)
|
|
|
|
type ColumnInfo struct {
|
|
Name string
|
|
IsNullable bool
|
|
DataType string
|
|
TableInfo *TableInfo
|
|
}
|
|
|
|
func (c ColumnInfo) IsUnique() bool {
|
|
for _, uniqueColumn := range c.TableInfo.PrimaryKeys {
|
|
if uniqueColumn == c.Name {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (c ColumnInfo) ToGoVarName() string {
|
|
return snaker.SnakeToCamel(c.Name) + "Column"
|
|
}
|
|
|
|
func (c ColumnInfo) ToSqlBuilderColumnType() string {
|
|
switch c.DataType {
|
|
case "boolean":
|
|
return "BoolColumn"
|
|
case "smallint":
|
|
return "IntegerColumn"
|
|
case "integer":
|
|
return "IntegerColumn"
|
|
case "bigint":
|
|
return "IntegerColumn"
|
|
case "date", "timestamp without time zone", "timestamp with time zone":
|
|
return "StringColumn"
|
|
case "bytea":
|
|
return "StringColumn"
|
|
case "text":
|
|
return "StringColumn"
|
|
case "real":
|
|
return "NumericColumn"
|
|
case "numeric", "double precision":
|
|
return "NumericColumn"
|
|
default:
|
|
return "StringColumn"
|
|
}
|
|
}
|
|
|
|
func (c ColumnInfo) ToGoType() string {
|
|
typeStr := c.GoBaseType()
|
|
if c.IsNullable || c.TableInfo.IsForeignKey(c.Name) {
|
|
return "*" + typeStr
|
|
}
|
|
|
|
return typeStr
|
|
}
|
|
|
|
func (c ColumnInfo) GoBaseType() string {
|
|
if forignKeyTable, ok := c.TableInfo.ForeignTableMap[c.Name]; ok {
|
|
return snaker.SnakeToCamel(forignKeyTable)
|
|
} else {
|
|
switch c.DataType {
|
|
case "boolean":
|
|
return "bool"
|
|
case "smallint":
|
|
return "int16"
|
|
case "integer":
|
|
return "int32"
|
|
case "bigint":
|
|
return "int64"
|
|
case "date", "timestamp without time zone", "timestamp with time zone":
|
|
return "time.Time"
|
|
case "bytea":
|
|
return "[]byte"
|
|
case "text":
|
|
return "string"
|
|
case "real":
|
|
return "float32"
|
|
case "numeric", "double precision":
|
|
return "float64"
|
|
default:
|
|
return "string"
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c ColumnInfo) ToGoDMFieldName() string {
|
|
if forignKeyTable, ok := c.TableInfo.ForeignTableMap[c.Name]; ok {
|
|
return snaker.SnakeToCamel(forignKeyTable)
|
|
} else {
|
|
return snaker.SnakeToCamel(c.Name)
|
|
}
|
|
}
|
|
|
|
func (c ColumnInfo) ToGoFieldName() string {
|
|
return snaker.SnakeToCamel(c.Name)
|
|
}
|
|
|
|
func fetchColumnInfos(db *sql.DB, tableInfo *TableInfo) ([]ColumnInfo, error) {
|
|
|
|
query := `
|
|
SELECT column_name, is_nullable, data_type
|
|
FROM information_schema.columns
|
|
where table_schema = $1 and table_name = $2
|
|
order by ordinal_position;`
|
|
|
|
//fmt.Println(query)
|
|
|
|
rows, err := db.Query(query, tableInfo.DatabaseInfo.SchemaName, &tableInfo.Name)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
ret := []ColumnInfo{}
|
|
|
|
for rows.Next() {
|
|
columnInfo := ColumnInfo{}
|
|
var isNullable string
|
|
err := rows.Scan(&columnInfo.Name, &isNullable, &columnInfo.DataType)
|
|
|
|
columnInfo.IsNullable = isNullable == "YES"
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
columnInfo.TableInfo = tableInfo
|
|
|
|
ret = append(ret, columnInfo)
|
|
}
|
|
|
|
err = rows.Err()
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ret, nil
|
|
}
|