Merge pull request #3 from go-gorm/feat_tableType

feat: support TableType

migrator.go

@@ -1,6 +1,7 @@
 package rawsql
 
 import (
+	"database/sql"
 	"strings"
 
 	"gorm.io/gorm"
@@ -12,6 +13,23 @@ type Migrator struct {
 	Dialector
 }
 
+func (m Migrator) TableType(value interface{}) (tableType gorm.TableType, err error) {
+	err = m.RunWithValue(value, func(stmt *gorm.Statement) error {
+		var (
+			schema, tableName = m.CurrentSchema(stmt, stmt.Table)
+		)
+		table, ok := m.tables[tableName]
+		if ok && table != nil {
+			tableType = &migrator.TableType{
+				SchemaValue:  schema,
+				NameValue:    tableName,
+				CommentValue: sql.NullString{String: table.Comment, Valid: true},
+			}
+		}
+		return nil
+	})
+	return tableType, err
+}
 func (m Migrator) ColumnTypes(value interface{}) ([]gorm.ColumnType, error) {
 	columnTypes := make([]gorm.ColumnType, 0)
 	err := m.RunWithValue(value, func(stmt *gorm.Statement) error {

table.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"reflect"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/pingcap/tidb/parser"
@@ -20,6 +21,7 @@ type Table struct {
 	ColumnTypes []gorm.ColumnType
 	Indexes     []gorm.Index
 	Name        string
+	Comment     string
 }
 
 type Parser interface {
@@ -59,6 +61,7 @@ func (d *defaultParser) ParseSQL(sql string) error {
 
 			d.tables[tableName] = &Table{
 				Name:        tableName,
+				Comment:     create.Table.TableInfo.Comment,
 				ColumnTypes: d.getColumnTypes(create),
 				Indexes:     d.getIndexes(create),
 			}
@@ -177,9 +180,9 @@ func (*defaultParser) getColumnType(col *ast.ColumnDef) gorm.ColumnType {
 		NameValue: sql.NullString{Valid: true, String: col.Name.OrigColName()},
 		DataTypeValue: sql.NullString{
 			Valid:  true,
-			String: types.TypeToStr(col.Tp.GetType(), col.Tp.GetCharset()),
+			String: strings.ToLower(types.TypeToStr(col.Tp.GetType(), col.Tp.GetCharset())),
 		},
-		ColumnTypeValue: sql.NullString{Valid: true, String: col.Tp.String()},
+		ColumnTypeValue: sql.NullString{Valid: true, String: strings.ToLower(col.Tp.String())},
 		PrimaryKeyValue: sql.NullBool{
 			Bool:  mysql.HasPriKeyFlag(col.Tp.GetFlag()),
 			Valid: mysql.HasPriKeyFlag(col.Tp.GetFlag()),