Fix for use with mysql:

Mysql require table for getExportedKeys. Otherwice it throws exception
Mysql return types in upper case

Author: MercurieVV

src/main/scala/com/geirsson/codegen/Codegen.scala

@@ -58,9 +58,9 @@ case class Codegen(options: CodegenOptions, namingStrategy: NamingStrategy) {
     }
   }
 
-  def getForeignKeys(db: Connection): Set[ForeignKey] = {
+  def getForeignKeys(db: Connection, tableName: String): Set[ForeignKey] = {
     val foreignKeys =
-      db.getMetaData.getExportedKeys(null, options.schema, null)
+      db.getMetaData.getExportedKeys(null, options.schema, tableName)
     results(foreignKeys).map { row =>
       ForeignKey(
         from = SimpleColumn(
@@ -79,12 +79,25 @@ case class Codegen(options: CodegenOptions, namingStrategy: NamingStrategy) {
     System.err.println(s"[${Console.YELLOW}warn${Console.RESET}] $msg")
   }
 
-  def getTables(db: Connection, foreignKeys: Set[ForeignKey]): Seq[Table] = {
+  def getTables(db: Connection): Seq[Table] = {
     val rs: ResultSet =
       db.getMetaData.getTables(null, options.schema, "%", Array("TABLE"))
     results(rs).flatMap { row =>
       val name = row.getString(TABLE_NAME)
       if (!excludedTables.contains(name)) {
+
+        val plainForeignKeys = getForeignKeys(db, name)
+
+        val foreignKeys = plainForeignKeys.map { fk =>
+
+          def resolve(col: SimpleColumn): SimpleColumn = {
+            plainForeignKeys.find(f => f.from == col).map(_.to).getOrElse(col)
+          }
+
+          ForeignKey(fk.from, resolve(fk.to))
+        }
+
+
         val columns = getColumns(db, name, foreignKeys)
         val mappedColumns = columns.filter(_.isRight).map(_.right.get)
         val unmappedColumns = columns.filter(_.isLeft).map(_.left.get)
@@ -110,7 +123,7 @@ case class Codegen(options: CodegenOptions, namingStrategy: NamingStrategy) {
       val simpleColumn = SimpleColumn(tableName, colName)
       val ref = foreignKeys.find(_.from == simpleColumn).map(_.to)
 
-      val typ = cols.getString(TYPE_NAME)
+      val typ = cols.getString(TYPE_NAME).toLowerCase
       columnType2scalaType.get(typ).map { scalaType =>
         Right(Column(
           tableName,
@@ -259,18 +272,7 @@ object Codegen extends AppOf[CodegenOptions] {
                                   codegenOptions.user,
                                   codegenOptions.password)
     val codegen = Codegen(codegenOptions, SnakeCaseReverse)
-    val plainForeignKeys = codegen.getForeignKeys(db)
-
-    val foreignKeys = plainForeignKeys.map { fk =>
-
-      def resolve(col: codegen.SimpleColumn): codegen.SimpleColumn = {
-        plainForeignKeys.find(f => f.from == col).map(_.to).getOrElse(col)
-      }
-
-      codegen.ForeignKey(fk.from, resolve(fk.to))
-    }
-
-    val tables = codegen.getTables(db, foreignKeys)
+    val tables = codegen.getTables(db)
     val generatedCode =
       codegen.tables2code(tables, SnakeCaseReverse, codegenOptions)
     val codeStyle = ScalafmtStyle.defaultWithAlign.copy(maxColumn = 120)