La risposta basata su Machinist è probabilmente la migliore. Qui è una variante più hackish che rileva casi come inferire Any
o AnyRef
o la tipica combinazione di due classi non correlate casi (Product with Serializable
):
import scala.collection.breakOut
import scala.language.experimental.macros
import scala.reflect.macros.blackbox
object Implicits {
implicit class TripleEquals[A](a: A) {
def === [B >: A](b: B): Boolean = macro Macros.equalsImpl[A, B]
}
}
object Macros {
private val positiveList = Set("scala.Boolean", "scala.Int", "scala.Long",
"scala.Float", "scala.Double", "scala.Option)
private val negativeList = Set("java.lang.Object", "java.io.Serializable",
"<refinement>")
def equalsImpl[A: c.WeakTypeTag, B: c.WeakTypeTag](c: blackbox.Context)
(b: c.Expr[A]): c.Tree = {
import c.universe._
val bTpe = weakTypeOf[B]
val base = bTpe.baseClasses
val names: Set[String] = base.collect {
case sym if sym.isClass => sym.fullName
} (breakOut)
// if a primitive is inferred, we're good. otherwise:
if (names.intersect(positiveList).isEmpty) {
// exclude all such as scala.Product, scala.Equals
val withoutTopLevel = names.filterNot { n =>
val i = n.lastIndexOf('.')
i == 5 && n.startsWith("scala")
}
// exclude refinements and known Java types
val excl = withoutTopLevel.diff(negativeList)
if (excl.isEmpty) {
c.abort(c.enclosingPosition, s"Inferred type is too generic: `$bTpe`")
}
}
// now simply rewrite as `a == b`
val q"$_($a)" = c.prefix.tree
q"$a == $b"
}
}
Questo non funziona con tipi superiori-kinded, ancora, in modo tuple stanno deliberatamente fallendo, mentre sfortunatamente compila Some(1) === Some("hello")
.
Edit: A built a small library che migliora su questo per supportare i tipi di alto-kinded.
fonte
2016-04-23 22:15:55
Che cosa potrebbe acquistare un'implementazione di macro su una classe di valore implicita? –
@Le classi di valore diTravisBrown salvano il costo di allocazione, ma non salvano il costo di riferimento indiretto, se non sbaglio. Questo potrebbe essere rilevante: http://typelevel.org/blog/2015/08/06/machinist.html –