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

Generator clean up.

Browse source
This commit is contained in:
go-jet 2019-08-14 12:50:31 +02:00
parent 4ab9d73a8b
commit 91dc633b45
8 changed files with 98 additions and 68 deletions
Download patch file Download diff file Expand all files Collapse all files

58
generator/internal/metadata/column_info.go → generator/internal/metadata/column_meta_data.go
View file

@ -7,17 +7,37 @@ import (
"strings"
)
// ColumnInfo metadata struct
type ColumnInfo struct {
// ColumnMetaData struct
type ColumnMetaData struct {
Name string
IsNullable bool
DataType string
IsUnsigned bool
EnumName string
IsUnsigned bool
SqlBuilderColumnType string
GoBaseType string
GoModelType string
}
// SqlBuilderColumnType returns type of jet sql builder column
func (c ColumnInfo) SqlBuilderColumnType() string {
func NewColumnMetaData(name string, isNullable bool, dataType string, enumName string, isUnsigned bool) ColumnMetaData {
columnMetaData := ColumnMetaData{
Name: name,
IsNullable: isNullable,
DataType: dataType,
EnumName: enumName,
IsUnsigned: isUnsigned,
}
columnMetaData.SqlBuilderColumnType = columnMetaData.getSqlBuilderColumnType()
columnMetaData.GoBaseType = columnMetaData.getGoBaseType()
columnMetaData.GoModelType = columnMetaData.getGoModelType()
return columnMetaData
}
// getSqlBuilderColumnType returns type of jet sql builder column
func (c ColumnMetaData) getSqlBuilderColumnType() string {
switch c.DataType {
case "boolean":
return "Bool"
@ -45,13 +65,13 @@ func (c ColumnInfo) SqlBuilderColumnType() string {
"double": // MySQL
return "Float"
default:
fmt.Println("Unsupported sql type: " + c.DataType + ", using string column instead for sql builder.")
fmt.Println("- [SQL Builder] Unsupported sql column '" + c.Name + " " + c.DataType + "', using StringColumn instead.")
return "String"
}
}
// GoBaseType returns model type for column info.
func (c ColumnInfo) GoBaseType() string {
// getGoBaseType returns model type for column info.
func (c ColumnMetaData) getGoBaseType() string {
switch c.DataType {
case "USER-DEFINED", "enum":
return utils.ToGoIdentifier(c.EnumName)
@ -85,15 +105,15 @@ func (c ColumnInfo) GoBaseType() string {
case "uuid":
return "uuid.UUID"
default:
fmt.Println("Unsupported sql type: " + c.DataType + ", using string instead for model type.")
fmt.Println("- [Model ] Unsupported sql column '" + c.Name + " " + c.DataType + "', using string instead.")
return "string"
}
}
// GoModelType returns model type for column info with optional pointer if
// column can be NULL.
func (c ColumnInfo) GoModelType() string {
typeStr := c.GoBaseType()
func (c ColumnMetaData) getGoModelType() string {
typeStr := c.GoBaseType
if strings.Contains(typeStr, "int") && c.IsUnsigned {
typeStr = "u" + typeStr
@ -107,7 +127,7 @@ func (c ColumnInfo) GoModelType() string {
}
// GoModelTag returns model field tag for column
func (c ColumnInfo) GoModelTag(isPrimaryKey bool) string {
func (c ColumnMetaData) GoModelTag(isPrimaryKey bool) string {
tags := []string{}
if isPrimaryKey {
@ -121,7 +141,7 @@ func (c ColumnInfo) GoModelTag(isPrimaryKey bool) string {
return ""
}
func getColumnInfos(db *sql.DB, querySet MetaDataQuerySet, schemaName, tableName string) ([]ColumnInfo, error) {
func getColumnsMetaData(db *sql.DB, querySet DialectQuerySet, schemaName, tableName string) ([]ColumnMetaData, error) {
rows, err := db.Query(querySet.ListOfColumnsQuery(), schemaName, tableName)
@ -130,20 +150,18 @@ func getColumnInfos(db *sql.DB, querySet MetaDataQuerySet, schemaName, tableName
}
defer rows.Close()
ret := []ColumnInfo{}
ret := []ColumnMetaData{}
for rows.Next() {
columnInfo := ColumnInfo{}
var isNullable string
err := rows.Scan(&columnInfo.Name, &isNullable, &columnInfo.DataType, &columnInfo.EnumName, &columnInfo.IsUnsigned)
columnInfo.IsNullable = isNullable == "YES"
var name, isNullable, dataType, enumName string
var isUnsigned bool
err := rows.Scan(&name, &isNullable, &dataType, &enumName, &isUnsigned)
if err != nil {
return nil, err
}
ret = append(ret, columnInfo)
ret = append(ret, NewColumnMetaData(name, isNullable == "YES", dataType, enumName, isUnsigned))
}
err = rows.Err()

4
generator/internal/metadata/query.go → generator/internal/metadata/dialect_query_set.go
View file

@ -5,7 +5,7 @@ import (
"strings"
)
type MetaDataQuerySet interface {
type DialectQuerySet interface {
ListOfTablesQuery() string
PrimaryKeysQuery() string
ListOfColumnsQuery() string
@ -123,7 +123,7 @@ func (m *MySqlQuerySet) GetEnumsMetaData(db *sql.DB, schemaName string) ([]MetaD
enumValues = strings.Replace(enumValues[1:len(enumValues)-1], "'", "", -1)
ret = append(ret, EnumInfo{
ret = append(ret, EnumMetaData{
name: enumName,
Values: strings.Split(enumValues, ","),
})

10
generator/internal/metadata/enum_info.go → generator/internal/metadata/enum_meta_data.go
View file

@ -4,18 +4,18 @@ import (
"database/sql"
)
// EnumInfo struct
type EnumInfo struct {
// EnumMetaData struct
type EnumMetaData struct {
name string
Values []string
}
// Name returns enum name
func (e EnumInfo) Name() string {
func (e EnumMetaData) Name() string {
return e.name
}
func getEnumInfos(db *sql.DB, querySet MetaDataQuerySet, schemaName string) ([]MetaData, error) {
func getEnumInfos(db *sql.DB, querySet DialectQuerySet, schemaName string) ([]MetaData, error) {
rows, err := db.Query(querySet.ListOfEnumsQuery(), schemaName)
@ -49,7 +49,7 @@ func getEnumInfos(db *sql.DB, querySet MetaDataQuerySet, schemaName string) ([]M
ret := []MetaData{}
for enumName, enumValues := range enumsInfosMap {
ret = append(ret, EnumInfo{
ret = append(ret, EnumMetaData{
enumName,
enumValues,
})

8
generator/internal/metadata/schema_info.go → generator/internal/metadata/schema_meta_data.go
View file

@ -5,14 +5,14 @@ import (
"fmt"
)
// SchemaInfo metadata struct
type SchemaInfo struct {
// SchemaMetaData struct
type SchemaMetaData struct {
TableInfos []MetaData
EnumInfos []MetaData
}
// GetSchemaInfo returns schema information from db connection.
func GetSchemaInfo(db *sql.DB, schemaName string, querySet MetaDataQuerySet) (schemaInfo SchemaInfo, err error) {
func GetSchemaInfo(db *sql.DB, schemaName string, querySet DialectQuerySet) (schemaInfo SchemaMetaData, err error) {
schemaInfo.TableInfos, err = getTableInfos(db, querySet, schemaName)
@ -31,7 +31,7 @@ func GetSchemaInfo(db *sql.DB, schemaName string, querySet MetaDataQuerySet) (sc
return
}
func getTableInfos(db *sql.DB, querySet MetaDataQuerySet, schemaName string) ([]MetaData, error) {
func getTableInfos(db *sql.DB, querySet DialectQuerySet, schemaName string) ([]MetaData, error) {
rows, err := db.Query(querySet.ListOfTablesQuery(), schemaName)

26
generator/internal/metadata/table_info.go → generator/internal/metadata/table_meta_data.go
View file

@ -5,27 +5,27 @@ import (
"github.com/go-jet/jet/internal/utils"
)
// TableInfo metadata struct
type TableInfo struct {
// TableMetaData metadata struct
type TableMetaData struct {
SchemaName string
name string
PrimaryKeys map[string]bool
Columns []ColumnInfo
Columns []ColumnMetaData
}
// Name returns table info name
func (t TableInfo) Name() string {
func (t TableMetaData) Name() string {
return t.name
}
// IsPrimaryKey returns if column is a part of primary key
func (t TableInfo) IsPrimaryKey(column string) bool {
func (t TableMetaData) IsPrimaryKey(column string) bool {
return t.PrimaryKeys[column]
}
// MutableColumns returns list of mutable columns for table
func (t TableInfo) MutableColumns() []ColumnInfo {
ret := []ColumnInfo{}
func (t TableMetaData) MutableColumns() []ColumnMetaData {
ret := []ColumnMetaData{}
for _, column := range t.Columns {
if t.IsPrimaryKey(column.Name) {
@ -39,11 +39,11 @@ func (t TableInfo) MutableColumns() []ColumnInfo {
}
// GetImports returns model imports for table.
func (t TableInfo) GetImports() []string {
func (t TableMetaData) GetImports() []string {
imports := map[string]string{}
for _, column := range t.Columns {
columnType := column.GoBaseType()
columnType := column.GoBaseType
switch columnType {
case "time.Time":
@ -63,12 +63,12 @@ func (t TableInfo) GetImports() []string {
}
// GoStructName returns go struct name for sql builder
func (t TableInfo) GoStructName() string {
func (t TableMetaData) GoStructName() string {
return utils.ToGoIdentifier(t.name) + "Table"
}
// GetTableInfo returns table info metadata
func GetTableInfo(db *sql.DB, querySet MetaDataQuerySet, schemaName, tableName string) (tableInfo TableInfo, err error) {
func GetTableInfo(db *sql.DB, querySet DialectQuerySet, schemaName, tableName string) (tableInfo TableMetaData, err error) {
tableInfo.SchemaName = schemaName
tableInfo.name = tableName
@ -78,7 +78,7 @@ func GetTableInfo(db *sql.DB, querySet MetaDataQuerySet, schemaName, tableName s
return
}
tableInfo.Columns, err = getColumnInfos(db, querySet, schemaName, tableName)
tableInfo.Columns, err = getColumnsMetaData(db, querySet, schemaName, tableName)
if err != nil {
return
@ -87,7 +87,7 @@ func GetTableInfo(db *sql.DB, querySet MetaDataQuerySet, schemaName, tableName s
return
}
func getPrimaryKeys(db *sql.DB, querySet MetaDataQuerySet, schemaName, tableName string) (map[string]bool, error) {
func getPrimaryKeys(db *sql.DB, querySet DialectQuerySet, schemaName, tableName string) (map[string]bool, error) {
rows, err := db.Query(querySet.PrimaryKeysQuery(), schemaName, tableName)

3
generator/mysql/mysql_generator.go
View file

@ -50,6 +50,9 @@ func Generate(destDir string, dbConn DBConnection) error {
func openConnection(dbConn DBConnection) (*sql.DB, error) {
var connectionString = fmt.Sprintf("%s:%s@tcp(%s:%d)/%s", dbConn.User, dbConn.Password, dbConn.Host, dbConn.Port, dbConn.DBName)
if dbConn.Params != "" {
connectionString += "?" + dbConn.Params
}
db, err := sql.Open("mysql", connectionString)
fmt.Println("Connecting to MySQL database: " + connectionString)