Add tableOptions code

takapi327 · Oct 10, 2023 · 85a5c3e · 85a5c3e
1 parent b39600d
commit 85a5c3e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/module/ldbc-codegen/src/main/scala/ldbc/codegen/TableModelGenerator.scala b/module/ldbc-codegen/src/main/scala/ldbc/codegen/TableModelGenerator.scala
@@ -85,6 +85,10 @@ private[ldbc] object TableModelGenerator:
       s".keySet(table => ${ key.toCode("table", classNameFormatter, propertyNameFormatter) })"
     )
 
+    val tableOptions = statement.options.fold("")(
+      _.map(option => s".setOption(${ Table.buildTableOptionCode(option) })").mkString("\n  ")
+    )
+
     val builder = ColumnCodeBuilder(classNameFormatter)
 
     val columns =
@@ -111,6 +115,7 @@ private[ldbc] object TableModelGenerator:
          |    ${ columns.mkString(",\n    ") }
          |  )
          |  ${ keyDefinitions.mkString("\n  ") }
+         |  $tableOptions
          |""".stripMargin
 
     Files.write(outputFile.toPath, scalaSource.getBytes(summon[Codec].name))