[parser] model: add isUnsigned for allocator (#34)

pingcap · Oct 9, 2021 · 8c54ae8 · 8c54ae8
1 parent 027ca88
commit 8c54ae8
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/parser/model/model.go b/parser/model/model.go
@@ -238,6 +238,15 @@ func (t *TableInfo) GetPkColInfo() *ColumnInfo {
 	return nil
 }
 
+func (t *TableInfo) GetAutoIncrementColInfo() *ColumnInfo {
+	for _, colInfo := range t.Columns {
+		if mysql.HasAutoIncrementFlag(colInfo.Flag) {
+			return colInfo
+		}
+	}
+	return nil
+}
+
 // Cols returns the columns of the table in public state.
 func (t *TableInfo) Cols() []*ColumnInfo {
 	publicColumns := make([]*ColumnInfo, len(t.Columns))