diff options
author | Stewart Stewart <stewinsalot@gmail.com> | 2017-02-28 10:32:40 -0800 |
---|---|---|
committer | Stewart Stewart <stewinsalot@gmail.com> | 2017-02-28 10:32:40 -0800 |
commit | 2f9bedbefd8053a5148b235f66956ebf4a07b58b (patch) | |
tree | 68a8a72e15fa005bc2248d957cf517ee1dfa0e71 /src/main/scala/TypedIdTable.scala | |
parent | 6dc60108716bc2aaa1c3be6aeae06b0973524328 (diff) | |
download | slick-codegen-plugin-2f9bedbefd8053a5148b235f66956ebf4a07b58b.tar.gz slick-codegen-plugin-2f9bedbefd8053a5148b235f66956ebf4a07b58b.tar.bz2 slick-codegen-plugin-2f9bedbefd8053a5148b235f66956ebf4a07b58b.zip |
minor fixes
Diffstat (limited to 'src/main/scala/TypedIdTable.scala')
-rw-r--r-- | src/main/scala/TypedIdTable.scala | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/main/scala/TypedIdTable.scala b/src/main/scala/TypedIdTable.scala index 08b2b28..fdd8aa5 100644 --- a/src/main/scala/TypedIdTable.scala +++ b/src/main/scala/TypedIdTable.scala @@ -58,18 +58,18 @@ class TypedIdSourceCodeGenerator( class PrimaryKeyMapperDef extends TermDef { def primaryKeyColumn: Option[Column] = { table.model.columns - .filter(_.options.contains(slick.ast.ColumnOption.PrimaryKey)) + .flatMap(c => keyReferences.get(c).filter(_ == c)) .headOption.map(c => table.columnsByName(c.name)) } override def enabled = primaryKeyColumn.isDefined - override def doc = s"Implicit for mapping primary key of ${tableName} to a base column" + override def doc = s"Implicit for mapping primary key of ${tableName(table.model.name.table)} to a base column" - override def rawName = tableName+"KeyMapper" + override def rawName = tableName(table.model.name.table) + "KeyMapper" override def code = primaryKeyColumn.fold("") { column => - val tpe = s"BaseColumnType[${column.rawName}]" + val tpe = s"BaseColumnType[${column.rawType}]" s"""|implicit def $name: $tpe = |${modelTypeToColumnMaper(column.model.tpe)}[${pKeyTypeTag(column.model)}] |""".stripMargin.lines.mkString("").trim |