fix(types): Updating type casting to bools (#74)

Updating type casting to bools
Jacobbrewer1 · Jan 31, 2025 · 21f8155 · 21f8155
1 parent 8371003
commit 21f8155
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pkg/entities/column.go b/pkg/entities/column.go
@@ -38,9 +38,17 @@ type Column struct {
 }
 
 func (c *Column) setTypeInfo(tp *types.FieldType) {
-	c.Type = tp.EvalType().String()
-	if tp.GetType() == mysql.TypeLonglong {
+	switch {
+	case tp.GetType() == mysql.TypeTiny:
+		c.Type = "tinyint"
+	case tp.GetType() == mysql.TypeShort:
+		c.Type = "smallint"
+	case tp.GetType() == mysql.TypeInt24:
+		c.Type = "mediumint"
+	case tp.GetType() == mysql.TypeLonglong:
 		c.Type = "bigint"
+	default:
+		c.Type = tp.EvalType().String()
 	}
 	c.TypeSize = tp.GetFlen()
 	c.TypePrecision = tp.GetDecimal()