github/jooq
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[#4600] Add support for plain SQL APIs via Scala string interpolation

Browse Source
This commit is contained in:
lukaseder 2015-10-06 10:07:01 +02:00
parent cfe3013af9
commit abbbdba4d9
1 changed files with 35 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
jOOQ-scala/src/main/scala/org/jooq/scala/Conversions.scala
View File

@ -92,6 +92,41 @@ object Conversions {
// Enhanced jOOQ types
// -------------------------------------------------------------------------
implicit class SQLInterpolation(val sc : StringContext) extends AnyVal {
@PlainSQL
def sql(args: Any*) : SQL = DSL.sql(string(args), args.asInstanceOf[Seq[AnyRef]] : _*)
@PlainSQL
def condition(args : Any*) : Condition = DSL.condition(string(args), args.asInstanceOf[Seq[AnyRef]] : _*)
@PlainSQL
def table(args : Any*) : Table[Record] = DSL.table(string(args), args.asInstanceOf[Seq[AnyRef]] : _*)
@PlainSQL
def query(args : Any*) : Query = DSL.query(string(args), args.asInstanceOf[Seq[AnyRef]] : _*)
@PlainSQL
def resultQuery(args : Any*) : ResultQuery[Record] = DSL.resultQuery(string(args), args.asInstanceOf[Seq[AnyRef]] : _*)
private def string(args : Any*) = {
val pi = sc.parts.iterator
val sb = new StringBuilder(pi.next())
var i = 0;
while (pi.hasNext) {
sb += '{'
sb ++= (i toString)
sb += '}'
sb ++= pi.next()
i = i + 1;
}
sb.result
}
}
implicit class ScalaDSLContext (val ctx : DSLContext) {
def fetchAnyOption[R <: Record] (table : Table[R]) : Option[R] = Option(ctx.fetchAny(table))
def fetchAnyOption[R <: Record] (table : Table[R], condition : Condition) : Option[R] = Option(ctx.fetchAny(table, condition))