scalikejdbc.OneToManies6SQL.scala Maven / Gradle / Ivy
/*
* Copyright 2013 - 2015 scalikejdbc.org
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
* either express or implied. See the License for the specific language
* governing permissions and limitations under the License.
*/
package scalikejdbc
import scala.collection.mutable.LinkedHashMap
import scala.collection.generic.CanBuildFrom
import scala.language.higherKinds
private[scalikejdbc] trait OneToManies6Extractor[A, B1, B2, B3, B4, B5, B6, E <: WithExtractor, Z]
extends SQL[Z, E]
with RelationalSQLResultSetOperations[Z] {
private[scalikejdbc] def extractOne: WrappedResultSet => A
private[scalikejdbc] def extractTo1: WrappedResultSet => Option[B1]
private[scalikejdbc] def extractTo2: WrappedResultSet => Option[B2]
private[scalikejdbc] def extractTo3: WrappedResultSet => Option[B3]
private[scalikejdbc] def extractTo4: WrappedResultSet => Option[B4]
private[scalikejdbc] def extractTo5: WrappedResultSet => Option[B5]
private[scalikejdbc] def extractTo6: WrappedResultSet => Option[B6]
private[scalikejdbc] def transform: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z
private[scalikejdbc] def processResultSet(result: (LinkedHashMap[A, (Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6])]),
rs: WrappedResultSet): LinkedHashMap[A, (Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6])] = {
val o = extractOne(rs)
val (to1, to2, to3, to4, to5, to6) = (extractTo1(rs), extractTo2(rs), extractTo3(rs), extractTo4(rs), extractTo5(rs), extractTo6(rs))
result.keys.find(_ == o).map { _ =>
(to1 orElse to2 orElse to3 orElse to4 orElse to5 orElse to6).map { _ =>
val (ts1, ts2, ts3, ts4, ts5, ts6) = result.apply(o)
result += (o -> (
to1.map(t => if (ts1.contains(t)) ts1 else ts1 :+ t).getOrElse(ts1),
to2.map(t => if (ts2.contains(t)) ts2 else ts2 :+ t).getOrElse(ts2),
to3.map(t => if (ts3.contains(t)) ts3 else ts3 :+ t).getOrElse(ts3),
to4.map(t => if (ts4.contains(t)) ts4 else ts4 :+ t).getOrElse(ts4),
to5.map(t => if (ts5.contains(t)) ts5 else ts5 :+ t).getOrElse(ts5),
to6.map(t => if (ts6.contains(t)) ts6 else ts6 :+ t).getOrElse(ts6)
))
}.getOrElse(result)
}.getOrElse {
result += (
o -> (
to1.map(t => Vector(t)).getOrElse(Vector.empty),
to2.map(t => Vector(t)).getOrElse(Vector.empty),
to3.map(t => Vector(t)).getOrElse(Vector.empty),
to4.map(t => Vector(t)).getOrElse(Vector.empty),
to5.map(t => Vector(t)).getOrElse(Vector.empty),
to6.map(t => Vector(t)).getOrElse(Vector.empty)
)
)
}
}
private[scalikejdbc] def toTraversable(session: DBSession, sql: String, params: Seq[_], extractor: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z): Traversable[Z] = {
session.foldLeft(statement, parameters: _*)(LinkedHashMap[A, (Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6])]())(processResultSet).map {
case (one, (t1, t2, t3, t4, t5, t6)) => extractor(one, t1, t2, t3, t4, t5, t6)
}
}
}
class OneToManies6SQL[A, B1, B2, B3, B4, B5, B6, E <: WithExtractor, Z](
override val statement: String,
override private[scalikejdbc] val rawParameters: Seq[Any])(one: WrappedResultSet => A)(to1: WrappedResultSet => Option[B1], to2: WrappedResultSet => Option[B2], to3: WrappedResultSet => Option[B3], to4: WrappedResultSet => Option[B4], to5: WrappedResultSet => Option[B5], to6: WrappedResultSet => Option[B6])(extractor: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z)
extends SQL[Z, E](statement, rawParameters)(SQL.noExtractor[Z]("one-to-many extractor(one(RS => A).toManies(RS => Option[B1]...)) is specified, use #map((A,B) =>Z) instead."))
with AllOutputDecisionsUnsupported[Z, E] {
def map(extractor: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z): OneToManies6SQL[A, B1, B2, B3, B4, B5, B6, HasExtractor, Z] = {
new OneToManies6SQL(statement, rawParameters)(one)(to1, to2, to3, to4, to5, to6)(extractor)
}
override def toTraversable(): OneToManies6SQLToTraversable[A, B1, B2, B3, B4, B5, B6, E, Z] = {
new OneToManies6SQLToTraversable[A, B1, B2, B3, B4, B5, B6, E, Z](statement, rawParameters)(one)(to1, to2, to3, to4, to5, to6)(extractor)
}
override def toList(): OneToManies6SQLToList[A, B1, B2, B3, B4, B5, B6, E, Z] = {
new OneToManies6SQLToList[A, B1, B2, B3, B4, B5, B6, E, Z](statement, rawParameters)(one)(to1, to2, to3, to4, to5, to6)(extractor)
}
override def toOption(): OneToManies6SQLToOption[A, B1, B2, B3, B4, B5, B6, E, Z] = {
new OneToManies6SQLToOption[A, B1, B2, B3, B4, B5, B6, E, Z](statement, rawParameters)(one)(to1, to2, to3, to4, to5, to6)(extractor)(true)
}
override def headOption(): OneToManies6SQLToOption[A, B1, B2, B3, B4, B5, B6, E, Z] = {
new OneToManies6SQLToOption[A, B1, B2, B3, B4, B5, B6, E, Z](statement, rawParameters)(one)(to1, to2, to3, to4, to5, to6)(extractor)(false)
}
override def toCollection: OneToManies6SQLToCollection[A, B1, B2, B3, B4, B5, B6, E, Z] = {
new OneToManies6SQLToCollection[A, B1, B2, B3, B4, B5, B6, E, Z](statement, rawParameters)(one)(to1, to2, to3, to4, to5, to6)(extractor)
}
override def single(): OneToManies6SQLToOption[A, B1, B2, B3, B4, B5, B6, E, Z] = toOption()
override def first(): OneToManies6SQLToOption[A, B1, B2, B3, B4, B5, B6, E, Z] = headOption()
override def list(): OneToManies6SQLToList[A, B1, B2, B3, B4, B5, B6, E, Z] = toList()
override def traversable(): OneToManies6SQLToTraversable[A, B1, B2, B3, B4, B5, B6, E, Z] = toTraversable()
override def collection: OneToManies6SQLToCollection[A, B1, B2, B3, B4, B5, B6, E, Z] = toCollection
}
class OneToManies6SQLToList[A, B1, B2, B3, B4, B5, B6, E <: WithExtractor, Z](
override val statement: String,
override private[scalikejdbc] val rawParameters: Seq[Any])(one: WrappedResultSet => A)(to1: WrappedResultSet => Option[B1], to2: WrappedResultSet => Option[B2], to3: WrappedResultSet => Option[B3], to4: WrappedResultSet => Option[B4], to5: WrappedResultSet => Option[B5], to6: WrappedResultSet => Option[B6])(extractor: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z)
extends SQL[Z, E](statement, rawParameters)(SQL.noExtractor[Z]("one-to-many extractor(one(RS => A).toManies(RS => Option[B1])) is specified, use #map((A,B) =>Z) instead."))
with SQLToList[Z, E]
with AllOutputDecisionsUnsupported[Z, E]
with OneToManies6Extractor[A, B1, B2, B3, B4, B5, B6, E, Z] {
import GeneralizedTypeConstraintsForWithExtractor._
override def apply()(implicit session: DBSession, context: ConnectionPoolContext = NoConnectionPoolContext, hasExtractor: ThisSQL =:= SQLWithExtractor): List[Z] = {
executeQuery[List](session, (session: DBSession) => toTraversable(session, statement, rawParameters, extractor).toList)
}
private[scalikejdbc] def extractOne: WrappedResultSet => A = one
private[scalikejdbc] def extractTo1: WrappedResultSet => Option[B1] = to1
private[scalikejdbc] def extractTo2: WrappedResultSet => Option[B2] = to2
private[scalikejdbc] def extractTo3: WrappedResultSet => Option[B3] = to3
private[scalikejdbc] def extractTo4: WrappedResultSet => Option[B4] = to4
private[scalikejdbc] def extractTo5: WrappedResultSet => Option[B5] = to5
private[scalikejdbc] def extractTo6: WrappedResultSet => Option[B6] = to6
private[scalikejdbc] def transform: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z = extractor
}
final class OneToManies6SQLToCollection[A, B1, B2, B3, B4, B5, B6, E <: WithExtractor, Z] private[scalikejdbc](
override val statement: String,
override private[scalikejdbc] val rawParameters: Seq[Any])(one: WrappedResultSet => A)(to1: WrappedResultSet => Option[B1], to2: WrappedResultSet => Option[B2], to3: WrappedResultSet => Option[B3], to4: WrappedResultSet => Option[B4], to5: WrappedResultSet => Option[B5], to6: WrappedResultSet => Option[B6])(extractor: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z)
extends SQL[Z, E](statement, rawParameters)(SQL.noExtractor[Z]("one-to-many extractor(one(RS => A).toManies(RS => Option[B1])) is specified, use #map((A,B) =>Z) instead."))
with SQLToCollection[Z, E]
with AllOutputDecisionsUnsupported[Z, E]
with OneToManies6Extractor[A, B1, B2, B3, B4, B5, B6, E, Z] {
import GeneralizedTypeConstraintsForWithExtractor._
override def apply[C[_]]()(implicit session: DBSession, context: ConnectionPoolContext = NoConnectionPoolContext, hasExtractor: ThisSQL =:= SQLWithExtractor, cbf: CanBuildFrom[Nothing, Z, C[Z]]): C[Z] = {
executeQuery(session, (session: DBSession) => toTraversable(session, statement, rawParameters, extractor).to[C])
}
private[scalikejdbc] def extractOne: WrappedResultSet => A = one
private[scalikejdbc] def extractTo1: WrappedResultSet => Option[B1] = to1
private[scalikejdbc] def extractTo2: WrappedResultSet => Option[B2] = to2
private[scalikejdbc] def extractTo3: WrappedResultSet => Option[B3] = to3
private[scalikejdbc] def extractTo4: WrappedResultSet => Option[B4] = to4
private[scalikejdbc] def extractTo5: WrappedResultSet => Option[B5] = to5
private[scalikejdbc] def extractTo6: WrappedResultSet => Option[B6] = to6
private[scalikejdbc] def transform: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z = extractor
}
class OneToManies6SQLToTraversable[A, B1, B2, B3, B4, B5, B6, E <: WithExtractor, Z](
override val statement: String,
override private[scalikejdbc] val rawParameters: Seq[Any])(one: WrappedResultSet => A)(to1: WrappedResultSet => Option[B1], to2: WrappedResultSet => Option[B2], to3: WrappedResultSet => Option[B3], to4: WrappedResultSet => Option[B4], to5: WrappedResultSet => Option[B5], to6: WrappedResultSet => Option[B6])(extractor: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z)
extends SQL[Z, E](statement, rawParameters)(SQL.noExtractor[Z]("one-to-many extractor(one(RS => A).toMany(RS => Option[B1])) is specified, use #map((A,B) =>Z) instead."))
with SQLToTraversable[Z, E]
with AllOutputDecisionsUnsupported[Z, E]
with OneToManies6Extractor[A, B1, B2, B3, B4, B5, B6, E, Z] {
import GeneralizedTypeConstraintsForWithExtractor._
override def apply()(implicit session: DBSession, context: ConnectionPoolContext = NoConnectionPoolContext, hasExtractor: ThisSQL =:= SQLWithExtractor): Traversable[Z] = {
executeQuery[Traversable](session, (session: DBSession) => toTraversable(session, statement, rawParameters, extractor))
}
private[scalikejdbc] def extractOne: WrappedResultSet => A = one
private[scalikejdbc] def extractTo1: WrappedResultSet => Option[B1] = to1
private[scalikejdbc] def extractTo2: WrappedResultSet => Option[B2] = to2
private[scalikejdbc] def extractTo3: WrappedResultSet => Option[B3] = to3
private[scalikejdbc] def extractTo4: WrappedResultSet => Option[B4] = to4
private[scalikejdbc] def extractTo5: WrappedResultSet => Option[B5] = to5
private[scalikejdbc] def extractTo6: WrappedResultSet => Option[B6] = to6
private[scalikejdbc] def transform: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z = extractor
}
class OneToManies6SQLToOption[A, B1, B2, B3, B4, B5, B6, E <: WithExtractor, Z](
override val statement: String,
override private[scalikejdbc] val rawParameters: Seq[Any])(one: WrappedResultSet => A)(to1: WrappedResultSet => Option[B1], to2: WrappedResultSet => Option[B2], to3: WrappedResultSet => Option[B3], to4: WrappedResultSet => Option[B4], to5: WrappedResultSet => Option[B5], to6: WrappedResultSet => Option[B6])(extractor: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z)(protected val isSingle: Boolean = true)
extends SQL[Z, E](statement, rawParameters)(SQL.noExtractor[Z]("one-to-many extractor(one(RS => A).toMany(RS => Option[B1])) is specified, use #map((A,B) =>Z) instead."))
with SQLToOption[Z, E]
with AllOutputDecisionsUnsupported[Z, E]
with OneToManies6Extractor[A, B1, B2, B3, B4, B5, B6, E, Z] {
import GeneralizedTypeConstraintsForWithExtractor._
override def apply()(implicit session: DBSession, context: ConnectionPoolContext = NoConnectionPoolContext, hasExtractor: ThisSQL =:= SQLWithExtractor): Option[Z] = {
executeQuery[Option](session, (session: DBSession) => toSingle(toTraversable(session, statement, rawParameters, extractor)))
}
private[scalikejdbc] def extractOne: WrappedResultSet => A = one
private[scalikejdbc] def extractTo1: WrappedResultSet => Option[B1] = to1
private[scalikejdbc] def extractTo2: WrappedResultSet => Option[B2] = to2
private[scalikejdbc] def extractTo3: WrappedResultSet => Option[B3] = to3
private[scalikejdbc] def extractTo4: WrappedResultSet => Option[B4] = to4
private[scalikejdbc] def extractTo5: WrappedResultSet => Option[B5] = to5
private[scalikejdbc] def extractTo6: WrappedResultSet => Option[B6] = to6
private[scalikejdbc] def transform: (A, Seq[B1], Seq[B2], Seq[B3], Seq[B4], Seq[B5], Seq[B6]) => Z = extractor
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy