diff options
| -rw-r--r-- | src/main/scala/SchemaParser.scala | 23 |
1 files changed, 15 insertions, 8 deletions
diff --git a/src/main/scala/SchemaParser.scala b/src/main/scala/SchemaParser.scala index a5f0205..5ea61bf 100644 --- a/src/main/scala/SchemaParser.scala +++ b/src/main/scala/SchemaParser.scala @@ -9,15 +9,22 @@ import slick.{model => m} object ModelTransformation { + def citextColumnNoLength(column: m.Column): m.Column = + if (column.options contains SqlType("citext")) { + column.copy(options = column.options.filter { + case _: Length => false + case _ => true + }) + } else column + def citextNoLength(dbModel: m.Model): m.Model = - dbModel.copy(tables = dbModel.tables.map(table => - table.copy(columns = table.columns.map(column => - if (column.options contains SqlType("citext")) { - column.copy(options = column.options.filter { - case _: Length => false - case _ => true - }) - } else column)))) + dbModel.copy( + tables = dbModel.tables.map( + table => + table.copy( + columns = table.columns.map(citextColumnNoLength), + indices = table.indices.map(index => index.copy(columns = index.columns.map(citextColumnNoLength))) + ))) def references( dbModel: m.Model, |