soc*_*soc 9 sql macros scala compilation
我正在尝试构建一些类似SQL的抽象,我遇到了一个问题.
这是一个简化的"数据库表":
trait Coffee {
def id: Long
def name: String
def brand: String
}
Run Code Online (Sandbox Code Playgroud)
这是我的查询抽象:
import language.experimental.macros
object Query {
def from[T] =
macro QueryMacros.fromMacro[T]
}
class From[T] {
def select[S](s: T => S): Select[T] =
macro QueryMacros.selectMacro[T, S]
}
class Select[T] {
def where(pred: T => Boolean): Where =
macro QueryMacros.whereMacro[T]
}
class Where(val result: String)
Run Code Online (Sandbox Code Playgroud)
这是我的宏实现:
import scala.reflect.macros.Context
object QueryMacros {
val result = new StringBuilder
def fromMacro[T : c.WeakTypeTag](c: Context): c.Expr[From[T]] = {
result ++= ("FROM " + c.weakTypeOf[T])
c.universe.reify(new From[T])
}
def selectMacro[T : c.WeakTypeTag, S : c.WeakTypeTag](c: Context)(s: c.Expr[T => S]): c.Expr[Select[T]] = {
result ++= ("SELECT " + s.tree)
c.universe.reify(new Select[T])
}
def whereMacro[S](c: Context)(pred: c.Expr[S]): c.Expr[Where] = {
result ++= ("WHERE " + pred.tree)
c.universe.reify(new Where(result.toString))
}
}
Run Code Online (Sandbox Code Playgroud)
这是我的示例代码:
object Main extends App {
println("Query start")
val query =
Query.from[Coffee]
.select(_.id)
.where(_.brand == "FairTrade")
println(query.result)
println("Query end")
}
Run Code Online (Sandbox Code Playgroud)
它编译并运行正常,但输出是:
Query start
Query end
Run Code Online (Sandbox Code Playgroud)
基本上,result
似乎是空的.我预计它会保留树木的累积字符串.
如何将数据从宏编译阶段传递到下一阶段,以便在运行时显示?我当然可以明确地将当前字符串传递给下一个方法,但我想避免这种情况.
基本上,您需要有一个Queryable
抽象:1) 提供集合 API(from
、select
等),2) 通过具体化调用并将它们累积到内部来记住调用它的方法。
这个概念在我们的 ScalaDays 幻灯片 [1] 中有所解释,并在 Slick(开源)[2] 中实现。Queryable
顺便说一下,在 LINQ 中,它们与具体化调用并将它们提供给实现 的对象的方法大致相同IQueryable
,例如,如 [3] 中所述。
链接:
归档时间: |
|
查看次数: |
310 次 |
最近记录: |