[英]Type-safe equals macro?
对于Scala,是否存在类型安全的等于===
实现,其开销超过==
? 也就是说,与Scalaz和ScalaUtils中的===
不同,这是一个使用直接宏来执行检查的实现?
我想在很多地方使用===
,但这些都是热点,所以我不希望这会产生任何额外的运行时成本(比如构造类型等)。
我认为你可以通过机械师轻松实现。
GitHub上的README完全给出了===
示例:
import scala.{specialized => sp}
import machinist.DefaultOps
trait Eq[@sp A] {
def eqv(lhs: A, rhs: A): Boolean
}
object Eq {
implicit val intEq = new Eq[Int] {
def eqv(lhs: Int, rhs: Int): Boolean = lhs == rhs
}
implicit class EqOps[A](x: A)(implicit ev: Eq[A]) {
def ===(rhs: A): Boolean = macro DefaultOps.binop[A, Boolean]
}
}
那么你可以使用===
零开销(没有额外拨款,没有额外的间接)在==
如果您正在寻找一个开箱即用的实现, spire
(机械师起源)提供一个。
cats
提供一个。
它们都是基于宏的,因为它们使用machinist
来实现。
基于机械师的答案可能是最好的。 这是一个更为hackish的变体,可以检测诸如推断Any
或AnyRef
或两个不相关的案例类( 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"
}
}
这不适用于更高级别的类型,但是,元组故意失败,而不幸的是Some(1) === Some("hello")
编译。
编辑 :构建一个小型库 ,对此进行改进以支持更高级的类型。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.