9

我正在尝试构建一些类似 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似乎是空的。我希望它能够容纳堆积的树绳。

如何将我的数据从宏编译阶段传递到下一阶段，以便在运行时显示？我当然可以将当前字符串显式传递给下一个方法，但我想避免这种情况。

4

3 回答 3

3

基本上，您需要有一个Queryable抽象：1) 提供集合 API ( from,select等)，2) 通过具体化调用并在内部累积它们来记住在其上调用的方法。

这个概念在我们的 ScalaDays 幻灯片 [1] 中有所解释，并在 Slick（开源）[2] 中实现。顺便说一句，在 LINQ 中，它们对Queryable具体化调用并将它们提供给实现的对象的方法大致相同IQueryable，例如，如 [3] 中所述。

链接：

  1. http://scalamacros.org/talks/2012-04-18-ScalaDays2012.pdf
  2. https://github.com/slick/slick/tree/master/src/main/scala/scala/slick/queryable
  3. http://community.bartdesmet.net/blogs/bart/archive/2007/04/06/the-iqueryable-tales-linq-to-ldap-part-1-key-concepts.aspx
于 2012-08-02T15:06:03.327 回答
2

问题是没有将信息从一个宏调用传递到下一个宏调用。所有这些都发生在编译时，所以应该可以。问题在于最后调用的宏。由于它返回c.universe.reify(new Where(result.toString)),new Where(result.toString)在运行时被调用。然后result将是空的。您可以做的是 return c.Expr(tree)，其中treeapplyWhere的构造函数应用于String包含result.toString.

此外，您应该注意，您的代码取决于宏调用的编译顺序。如果您在多个代码文件中多次调用这些宏，则result可能包含来自先前调用的信息。最好重新考虑您的整个方法。

于 2012-08-02T13:01:48.170 回答
0

正如@Kim 指出的那样，聚合信息不是问题，而是宏扩展将生成result.toString在运行时评估的代码，当它确实为空时。我遇到了和你类似的问题，最后做了等效的result.toString替换resultExpr(c).splice

private def resultExpr(c :Context) = { 
  import c.universe._
   c.Expr[String](Literal(Constant(result.toString)))
}

（正如@Kim 还指出的那样，这会将所有宏调用的累积结果反馈给运行时，所以要小心！）

于 2012-08-23T10:21:45.103 回答