引入sql2go工具

2022-05-16 12:38:32 +08:00
parent ffb2a4608a
commit f9d322e453
9 changed files with 384 additions and 2 deletions
--- a/sql2go/parser.go
+++ b/sql2go/parser.go
@ -0,0 +1,92 @@
+// Package sql2go...
+//
+// Description : sql2go...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 2021-10-25 4:49 下午
+package sql2go
+
+import (
+	"errors"
+	"strings"
+
+	"git.zhangdeman.cn/zhangdeman/util"
+
+	"github.com/xwb1989/sqlparser"
+)
+
+const (
+	// CreateSQLColumnTPL 每个字段的模版
+	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, *BasicTableInfo, error) {
+	var (
+		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 "", nil, err
+	}
+
+	r, ok := stmt.(*sqlparser.DDL)
+	if !ok {
+		return "", nil, errors.New("input sql is not ddl")
+	}
+	basic.TableName = sqlparser.String(r.NewName)
+	basic.ModelStruct = util.String.SnakeCaseToCamel(basic.TableName)
+	structResult := "type " + basic.ModelStruct + " struct { \n"
+
+	for _, item := range r.TableSpec.Columns {
+		data := map[string]string{
+			"{FIELD}":         util.String.SnakeCaseToCamel(item.Name.String()),
+			"{COLUMN}":        item.Name.String(),
+			"{JSON_TAG}":      item.Name.String(),
+			"{DEFAULT_VALUE}": "",
+			"{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"
+		}
+		if nil != item.Type.Default {
+			data["{DEFAULT_VALUE}"] += string(item.Type.Default.Val)
+		}
+		val := CreateSQLColumnTPL
+		for k, v := range data {
+			val = strings.ReplaceAll(val, k, v)
+		}
+		structResult += val + "\n"
+	}
+	structResult = structResult + "}"
+	return structResult, basic, nil
+}