我正在嘗試構建一些類似SQL的抽象,並且遇到了問題。將宏計算結果傳遞給運行時的建議方法是什麼?
這是一個簡化的「數據庫表」:
trait Coffee {
def id: Long
def name: String
def brand: String
}
這是我的查詢抽象:
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)
這是我的宏實現:
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))
}
}
這是我的示例代碼:
object Main extends App {
println("Query start")
val query =
Query.from[Coffee]
.select(_.id)
.where(_.brand == "FairTrade")
println(query.result)
println("Query end")
}
它編譯並運行正常,但輸出是:
Query start
Query end
基本上,result
似乎是空的。我預計它會保存積累的樹木串。
如何將數據從宏編譯階段傳遞到下一階段,以便在運行時顯示? 我當然可以將當前字符串顯式傳遞給下一個方法,但我想避免這種情況。
嗨尤金,謝謝,我會看看。看起來有一個真正的解決方案。 :-) – soc 2012-08-02 15:43:15