model: add isUnsigned for allocator (pingcap#34)

tiancaiamao · Nov 12, 2018 · f9ff5ea · f9ff5ea
1 parent 8ac1198
commit f9ff5ea
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/model/model.go b/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))