如何在Scala編譯時檢查一些T是否是case類？

Question

我有以下宏：

package macros 

import scala.reflect.macros.blackbox.Context 

object CompileTimeAssertions { 
    def mustBeCaseClass[T]: Unit = 
    macro CompileTimeAssertionsImpl.mustBeCaseClass[T] 
} 

object CompileTimeAssertionsImpl { 
    def mustBeCaseClass[T: c.WeakTypeTag](c: Context): c.Expr[Unit] = { 
    import c.universe._ 
    val symbol = c.weakTypeTag[T].tpe.typeSymbol 
    if (!symbol.isClass || !symbol.asClass.isCaseClass) { 
     c.error(c.enclosingPosition, s"${symbol.fullName} must be a case class") 
    } 
    reify(Unit) 
    } 
} 

它工作時不參與仿製藥，但是當他們失敗：

import macros.CompileTimeAssertions._ 
import org.scalatest.{Matchers, WordSpec} 

case class ACaseClass(foo: String, bar: String) 

class NotACaseClass(baz: String) 

class MacroSpec extends WordSpec with Matchers { 
    "the mustBeCaseClass macro" should { 
    "compile when passed a case class" in { 
     mustBeCaseClass[ACaseClass] 
    } 

    "not compile when passed a vanilla class" in { 
//  mustBeCaseClass[NotACaseClass] // fails to compile as expected. 
    } 

    "compile when working with generics" in { 
//  class CaseClassContainer[T] { mustBeCaseClass[T] } // fails to compile. 
//  new CaseClassContainer[ACaseClass] 
    } 
    } 
} 

編譯器錯誤都是我的

MacroSpec.CaseClassContainer.T must be a case class 

我想知道當CaseClassContainer被實例化時T是什麼。這甚至有可能嗎？如果你能提供一個例子嗎？

在此先感謝。

Answer 1

感謝尤金和特拉維斯的建議，我能夠通過類型類來解決這個問題。這裏的解決方案：

package macros 

import scala.reflect.macros.blackbox.Context 

trait IsCaseClass[T] 

object IsCaseClass { 
    implicit def isCaseClass[T]: IsCaseClass[T] = 
    macro IsCaseClassImpl.isCaseClass[T] 
} 

object IsCaseClassImpl { 
    def isCaseClass[T] 
     (c: Context) 
     (implicit T: c.WeakTypeTag[T]): c.Expr[IsCaseClass[T]] = { 
    import c.universe._ 
    val symbol = c.weakTypeTag[T].tpe.typeSymbol 
    if (!symbol.isClass || !symbol.asClass.isCaseClass) { 
     c.abort(c.enclosingPosition, s"${symbol.fullName} must be a case class") 
    } else { 
     c.Expr[IsCaseClass[T]](q"_root_.macros.IsCaseClassImpl[$T]()") 
    } 
    } 
} 

case class IsCaseClassImpl[T]() extends IsCaseClass[T] 

這裏是用法：

import macros.IsCaseClass 
import org.scalatest.{Matchers, WordSpec} 

case class ACaseClass(foo: String, bar: String) 

class NotACaseClass(baz: String) 

class CaseClassContainer[T: IsCaseClass] 

class MacroSpec extends WordSpec with Matchers { 
    "the code" should { 
    "compile" in { 
     new CaseClassContainer[ACaseClass] 
    } 

    "not compile" in { 
//  new CaseClassContainer[NotACaseClass] 
    } 
    } 
} 

值得注意的使用abort代替error。中止返回Nothing，而錯誤返回Unit。當宏沒有返回任何東西時後者很好。

Answer 2

在scala 2.11及以上版本中，它現在更簡單了。我創建了一個小項目：https://github.com/samupra/CaseClassChecker