• Home
• io.getquill
• quill-core_2.11
• 0.6.0
• source code
• RenameAssignments.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.getquill.norm.RenameAssignments.scala Maven / Gradle / Ivy

package io.getquill.norm

import io.getquill.ast._

object RenameAssignments extends StatelessTransformer {

  override def apply(e: Action): Action =
    e match {
      case AssignedAction(insert @ Insert(table: Entity), assignments) =>
        AssignedAction(insert, renameAssignments(assignments, table))

      case AssignedAction(update @ Update(table: Entity), assignments) =>
        AssignedAction(update, renameAssignments(assignments, table))

      case AssignedAction(update @ Update(Filter(table: Entity, x, where)), assignments) =>
        AssignedAction(update, renameAssignments(assignments, table))

      case other =>
        super.apply(other)
    }

  private def renameAssignments(assignments: List[Assignment], table: Entity) = {
    val propertyAlias = table.properties.map(p => p.property -> p.alias).toMap
    assignments.map(a => a.copy(property = propertyAlias.getOrElse(a.property, a.property)))
  }
}