skinny.orm.feature.includes.IncludesQueryRepository.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of skinny-orm_2.10 Show documentation
skinny-orm
The newest version!
package skinny.orm.feature.includes

import skinny.orm.feature.associations._
import scala.collection.mutable

/**
 * Entity repository for includes queries.
 *
 * @tparam Entity base entity type
 */
trait IncludesQueryRepository[Entity] {

  private[this] val belongsTo: mutable.Map[BelongsToExtractor[Entity], Seq[_]] = new scala.collection.concurrent.TrieMap()
  private[this] val hasOne: mutable.Map[HasOneExtractor[Entity], Seq[_]] = new scala.collection.concurrent.TrieMap()
  private[this] val hasMany: mutable.Map[HasManyExtractor[Entity], Seq[_]] = new scala.collection.concurrent.TrieMap()

  /**
   * Returns entities for belongsTo relation.
   *
   * @param extractor extractor
   * @return entities
   */
  def entitiesFor(extractor: BelongsToExtractor[Entity]): Seq[_] = belongsTo.getOrElse(extractor, Nil)

  /**
   * Returns entities for hasOne relation.
   *
   * @param extractor extractor
   * @return entities
   */
  def entitiesFor(extractor: HasOneExtractor[Entity]): Seq[_] = hasOne.getOrElse(extractor, Nil)

  /**
   * Returns entities for hasMany relation.
   *
   * @param extractor extractor
   * @return entities
   */
  def entitiesFor(extractor: HasManyExtractor[Entity]): Seq[_] = hasMany.getOrElse(extractor, Nil)

  /**
   * Put an entity to repository.
   *
   * @param extractor extractor
   * @param entity entity
   */
  def putAndReturn[A](extractor: BelongsToExtractor[Entity], entity: A): A = {
    belongsTo.update(extractor, belongsTo.getOrElse(extractor, Nil).+:(entity))
    entity
  }

  /**
   * Put an entity to repository.
   *
   * @param extractor extractor
   * @param entity entity
   */
  def putAndReturn[A](extractor: HasOneExtractor[Entity], entity: A): A = {
    hasOne.update(extractor, hasOne.getOrElse(extractor, Nil).+:(entity))
    entity
  }

  /**
   * Put an entity to repository.
   *
   * @param extractor extractor
   * @param entity entity
   */
  def putAndReturn[A](extractor: HasManyExtractor[Entity], entity: A): A = {
    hasMany.update(extractor, hasMany.getOrElse(extractor, Nil).+:(entity))
    entity
  }
}

/**
 * IncludesQueryRepository factory.
 */
object IncludesQueryRepository {

  def apply[Entity](): IncludesQueryRepository[Entity] = new DefaultIncludesQueryRepository[Entity]
}

/**
 * Default implementation.
 *
 * @tparam Entity base entity type
 */
class DefaultIncludesQueryRepository[Entity] extends IncludesQueryRepository[Entity]