Rework resolving auto increment type to handle own ColumnType impleme…

…ntation (#663)
JetBrains · Oct 26, 2019 · 934077a · 934077a
1 parent 6257f17
commit 934077a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/exposed-core/src/main/kotlin/org/jetbrains/exposed/sql/ColumnType.kt b/exposed-core/src/main/kotlin/org/jetbrains/exposed/sql/ColumnType.kt
@@ -64,7 +64,13 @@ class AutoIncColumnType(val delegate: ColumnType, private val _autoincSeq: Strin
         is EntityIDColumnType<*> -> resolveAutIncType(columnType.idColumn.columnType)
         is IntegerColumnType -> currentDialect.dataTypeProvider.integerAutoincType()
         is LongColumnType -> currentDialect.dataTypeProvider.longAutoincType()
-        else -> error("Unsupported type $delegate for auto-increment")
+        else -> guessAutIncTypeBy(columnType.sqlType())
+    } ?: error("Unsupported type $delegate for auto-increment")
+
+    private fun guessAutIncTypeBy(sqlType: String) : String? = when (sqlType) {
+        currentDialect.dataTypeProvider.longType() -> currentDialect.dataTypeProvider.longAutoincType()
+        currentDialect.dataTypeProvider.integerType() -> currentDialect.dataTypeProvider.integerAutoincType()
+        else -> null
     }
 
     override fun sqlType(): String = resolveAutIncType(delegate)