diff --git a/tool/sql2go/parser.go b/tool/sql2go/parser.go index ea12b53..6cbb3a0 100644 --- a/tool/sql2go/parser.go +++ b/tool/sql2go/parser.go @@ -20,23 +20,44 @@ const ( CreateSQLColumnTPL = "{FIELD} {TYPE} `json:\"{JSON_TAG}\" gorm:\"column:{COLUMN};default:{DEFAULT_VALUE};{NOT_NULL}\"` // {COMMENT}" ) +// BasicTableInfo ... +// +// Author : go_developer@163.com<白茶清欢> +// +// Date : 11:47 上午 2021/11/17 +type BasicTableInfo struct { + TableName string + ModelStruct string + PrimaryField string + PrimaryFieldType string +} + // ParseCreateTableSql 解析建表sql // // Author : go_developer@163.com<白茶清欢> // // Date : 4:49 下午 2021/10/25 -func ParseCreateTableSql(sql string) (string, error) { +func ParseCreateTableSql(sql string) (string, *BasicTableInfo, error) { var ( - stmt sqlparser.Statement - err error + stmt sqlparser.Statement + err error + basic *BasicTableInfo ) + basic = &BasicTableInfo{ + TableName: "", + ModelStruct: "", + PrimaryField: "ID", + PrimaryFieldType: "", + } sql = strings.ReplaceAll(strings.ReplaceAll(sql, "CURRENT_TIMESTAMP()", "CURRENT_TIMESTAMP"), "current_timestamp()", "CURRENT_TIMESTAMP") if stmt, err = sqlparser.ParseStrictDDL(sql); nil != err { - return "", err + return "", nil, err } r := stmt.(*sqlparser.DDL) - structResult := "type " + util.SnakeCaseToCamel(sqlparser.String(r.NewName)) + " struct { \n" + basic.TableName = sqlparser.String(r.NewName) + basic.ModelStruct = util.SnakeCaseToCamel(basic.TableName) + structResult := "type " + basic.ModelStruct + " struct { \n" for _, item := range r.TableSpec.Columns { data := map[string]string{ @@ -47,6 +68,9 @@ func ParseCreateTableSql(sql string) (string, error) { "{COMMENT}": item.Name.String() + " " + string(item.Type.Comment.Val), "{TYPE}": sqlTypeMap[item.Type.Type], } + if data["{FIELD}"] == "ID" { + basic.PrimaryFieldType = data["{TYPE}"] + } if item.Type.NotNull { data["{NOT_NULL}"] = "NOT NULL" } @@ -60,5 +84,5 @@ func ParseCreateTableSql(sql string) (string, error) { structResult += val + "\n" } structResult = structResult + "}" - return structResult, nil + return structResult, basic, nil }