scalikejdbc.orm.associations.JoinsFeature.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scalikejdbc-orm_3 Show documentation
scalikejdbc-orm
The newest version!
package scalikejdbc.orm.associations

import scalikejdbc.orm.Alias
import scalikejdbc.orm.basic.{ IdFeature, SQLSyntaxSupportBase }
import scalikejdbc.orm.eagerloading.IncludesQueryRepository
import scalikejdbc.orm.finder.FinderFeatureWithId
import scalikejdbc.orm.querying.QueryingFeatureWithId
import scalikejdbc.{
  HasExtractor,
  NoExtractor,
  ResultName,
  SQL,
  WrappedResultSet
}

/**
 * Provides #joins APIs.
 *
 * NOTE: CRUDFeature has copy implementation from this trait.
 */
trait JoinsFeature[Entity]
  extends SQLSyntaxSupportBase[Entity]
  with AssociationsFeature[Entity] {
  self: IdFeature[?] =>

  /**
   * Appends join definition on runtime.
   *
   * @param associations associations
   * @return self
   */
  def joins[Id](
    associations: Association[?]*
  ): JoinsFeature[Entity]
    with IdFeature[Id]
    with FinderFeatureWithId[Id, Entity]
    with QueryingFeatureWithId[
      Id,
      Entity
    ] = {
    val _self = this
    val _associations = associations

    new JoinsFeature[Entity]
      with IdFeature[Id]
      with FinderFeatureWithId[Id, Entity]
      with QueryingFeatureWithId[Id, Entity] {
      override protected val underlying
        : scalikejdbc.orm.basic.SQLSyntaxSupportBase[Entity] = _self

      override def defaultAlias = _self.defaultAlias

      override def tableName = _self.tableName

      override def columnNames = _self.columnNames

      override def primaryKeyField = _self.primaryKeyField

      override def primaryKeyFieldName = _self.primaryKeyFieldName

      override def rawValueToId(value: Any) =
        _self.rawValueToId(value).asInstanceOf[Id]

      override def idToRawValue(id: Id): Any = id

      override val associations = _self.associations ++ _associations

      override val defaultJoinDefinitions = _self.defaultJoinDefinitions
      override val defaultBelongsToExtractors = _self.defaultBelongsToExtractors
      override val defaultHasOneExtractors = _self.defaultHasOneExtractors
      override val defaultOneToManyExtractors = _self.defaultOneToManyExtractors

      override def autoSession = underlying.autoSession

      override def connectionPoolName: Any = underlying.connectionPoolName

      override def connectionPool = underlying.connectionPool

      override def defaultScope(alias: Alias[Entity]) =
        _self.defaultScope(alias)
      // override def singleSelectQuery = _self.singleSelectQuery

      def extract(rs: WrappedResultSet, n: ResultName[Entity]) =
        underlying.extract(rs, n)
    }
  }

  override def extract(sql: SQL[Entity, NoExtractor])(implicit
    includesRepository: IncludesQueryRepository[Entity]
  ): SQL[Entity, HasExtractor] = {
    extractWithAssociations(
      sql,
      belongsToAssociations,
      hasOneAssociations,
      hasManyAssociations
    )
  }

}