Add buildTableOptionCode method

takapi327 · Oct 10, 2023 · b39600d · b39600d
1 parent d3ad39d
commit b39600d
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/module/ldbc-codegen/src/main/scala/ldbc/codegen/model/Table.scala b/module/ldbc-codegen/src/main/scala/ldbc/codegen/model/Table.scala
@@ -16,3 +16,32 @@ object Table:
   )
 
   case class DropStatement(tableName: String)
+
+  def buildTableOptionCode(option: TableOption): String =
+    option match
+      case TableOption.AutoExtendSize(value)           => s"TableOption.AutoExtendSize($value)"
+      case TableOption.AutoIncrement(value)            => s"TableOption.AutoIncrement($value)"
+      case TableOption.AVGRowLength(value)             => s"TableOption.AVGRowLength($value)"
+      case TableOption.Character(value)                => s"TableOption.Character(\"$value\")"
+      case TableOption.CheckSum(value)                 => s"TableOption.CheckSum($value)"
+      case TableOption.Collate(value)                  => s"TableOption.Collate(\"$value\")"
+      case TableOption.Comment(value)                  => s"TableOption.Comment(\"$value\")"
+      case TableOption.Compression(value)              => s"TableOption.Compression(\"$value\")"
+      case TableOption.Connection(value)               => s"TableOption.Connection(\"$value\")"
+      case TableOption.Directory(str, value)           => s"TableOption.Directory(\"$str\", \"$value\")"
+      case TableOption.DelayKeyWrite(value)            => s"TableOption.DelayKeyWrite($value)"
+      case TableOption.Encryption(value)               => s"TableOption.Encryption(\"$value\")"
+      case TableOption.Engine(value)                   => s"TableOption.Engine(\"$value\")"
+      case TableOption.EngineAttribute(value)          => s"TableOption.EngineAttribute(\"$value\")"
+      case TableOption.InsertMethod(value)             => s"TableOption.InsertMethod(\"$value\")"
+      case TableOption.KeyBlockSize(value)             => s"TableOption.KeyBlockSize($value)"
+      case TableOption.MaxRows(value)                  => s"TableOption.MaxRows($value)"
+      case TableOption.MinRows(value)                  => s"TableOption.MinRows($value)"
+      case TableOption.PackKeys(value)                 => s"TableOption.PackKeys(\"$value\")"
+      case TableOption.RowFormat(value)                => s"TableOption.RowFormat(\"$value\")"
+      case TableOption.SecondaryEngineAttribute(value) => s"TableOption.SecondaryEngineAttribute(\"$value\")"
+      case TableOption.StatsAutoRecalc(value)          => s"TableOption.StatsAutoRecalc(\"$value\")"
+      case TableOption.StatsPersistent(value)          => s"TableOption.StatsPersistent(\"$value\")"
+      case TableOption.StatsSamplePages(value)         => s"TableOption.StatsSamplePages($value)"
+      case TableOption.Tablespace(name, value)         => s"TableOption.Tablespace(\"$name\", \"$value\")"
+      case TableOption.Union(value)                    => s"TableOption.Union(List(${ value.map(str => s"\"$str\"").mkString(",") }))"