add missing type checks

loopbackio · Jul 6, 2024 · 53b9e5e · 53b9e5e
1 parent c86728f
commit 53b9e5e
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/lib/migration.js b/lib/migration.js
@@ -629,7 +629,16 @@ function mixinMigration(MySQL, mysql) {
         const idName = this.idName(model);
         const idProp = this.getModelDefinition(model).properties[idName];
         const idColumnType = this.columnDataType(model, idName);
-        if (idProp.generated && idColumnType === 'INT') {
+        if (
+          idProp.generated && 
+          (
+            idColumnType === 'TINYINT' ||
+            idColumnType === 'SMALLINT' ||
+            idColumnType === 'MEDIUMINT' ||
+            idColumnType === 'INT' ||
+            idColumnType === 'BIGINT'
+          )
+          ) {
           sql.push(self.columnEscaped(model, idName) + ' ' +
             self.buildColumnDefinition(model, idName) + ' AUTO_INCREMENT PRIMARY KEY');
         } else {