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

xhebox · Oct 8, 2021 · 5281092 · 5281092
1 parent 0e8df89
commit 5281092
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))