运行时值的精炼和存在类型

时间:2017-03-27 15:55:12

标签: scala shapeless singleton-type refinement-type refined

假设我想在一些字符串和整数标识符之间进行映射,并且我希望我的类型使得无法获得运行时故障,因为有人试图查找超出范围的id。这是一个简单的API:

trait Vocab {
  def getId(value: String): Option[Int]
  def getValue(id: Int): Option[String] 
}

但是,如果用户通常会从getId获取ID并因此知道它们是有效的,那么这很烦人。以下是这种意义上的改进:

trait Vocab[Id] {
  def getId(value: String): Option[Id]
  def getValue(id: Id): String
}

现在我们可以这样:

class TagId private(val value: Int) extends AnyVal

object TagId {
  val tagCount: Int = 100

  def fromInt(id: Int): Option[TagId] =
    if (id >= 0 && id < tagCount) Some(new TagId(id)) else None
}

然后我们的用户可以使用Vocab[TagId]而不必担心在典型情况下检查getValue查找是否失败,但如果需要,他们仍然可以查找任意整数。但是,它仍然很尴尬,因为我们必须为每种我们想要词汇的东西编写一个单独的类型。

我们也可以使用refined执行类似的操作:

import eu.timepit.refined.api.Refined
import eu.timepit.refined.numeric.Interval.ClosedOpen
import shapeless.Witness

class Vocab(values: Vector[String]) {
  type S <: Int
  type P = ClosedOpen[Witness.`0`.T, S]

  def size: S = values.size.asInstanceOf[S]

  def getId(value: String): Option[Refined[Int, P]] = values.indexOf(value) match {
    case -1 => None
    case i  => Some(Refined.unsafeApply[Int, P](i))
  }

  def getValue(id: Refined[Int, P]): String = values(id.value)
}

现在即使在编译时不知道S,编译器仍然能够跟踪它给出的id在0和S之间的事实,所以我们不要当我们回到值时,我们不得不担心失败的可能性(当然,如果我们使用相同的vocab实例)。

我想要的是能够写下这个:

val x = 2
val vocab = new Vocab(Vector("foo", "bar", "qux"))

eu.timepit.refined.refineV[vocab.P](x).map(vocab.getValue)

这样用户可以在真正需要时轻松查找任意整数。但是这不能编译:

scala> eu.timepit.refined.refineV[vocab.P](x).map(vocab.getValue)
<console>:17: error: could not find implicit value for parameter v: eu.timepit.refined.api.Validate[Int,vocab.P]
       eu.timepit.refined.refineV[vocab.P](x).map(vocab.getValue)
                                          ^

我可以通过为Witness提供S实例来进行编译:

scala> implicit val witVocabS: Witness.Aux[vocab.S] = Witness.mkWitness(vocab.size)
witVocabS: shapeless.Witness.Aux[vocab.S] = shapeless.Witness$$anon$1@485aac3c

scala> eu.timepit.refined.refineV[vocab.P](x).map(vocab.getValue)
res1: scala.util.Either[String,String] = Right(qux)

当然,当值超出范围时,它会失败(在运行时但安全):

scala> val y = 3
y: Int = 3

scala> println(eu.timepit.refined.refineV[vocab.P](y).map(vocab.getValue))
Left(Right predicate of (!(3 < 0) && (3 < 3)) failed: Predicate failed: (3 < 3).)

我还可以将见证定义放在我的Vocab类中,然后导入vocab._以便在需要时使其可用,但我真正想要的是能够提供refineV支持,无需额外的导入或定义。

我尝试过各种各样的东西:

object Vocab {
  implicit def witVocabS[V <: Vocab](implicit
    witV: Witness.Aux[V]
  ): Witness.Aux[V#S] = Witness.mkWitness(witV.value.size)
}

但是这仍然需要为每个vocab实例明确定义:

scala> implicit val witVocabS: Witness.Aux[vocab.S] = Vocab.witVocabS
witVocabS: shapeless.Witness.Aux[vocab.S] = shapeless.Witness$$anon$1@1bde5374

scala> eu.timepit.refined.refineV[vocab.P](x).map(vocab.getValue)
res4: scala.util.Either[String,String] = Right(qux)

我知道我可以使用宏来实现witVocabS,但我觉得应该有更好的方法来做这种事情,因为它看起来像一个非常合理的用例(我不是很熟悉精致,所以我完全有可能错过一些明显的东西。

2 个答案:

答案 0 :(得分:12)

事实证明,如果我们通过使用http://awiacja.imgw.pl/rss/metar00.php?airport=EPWAS分配单一类型values.size来使类型参数shapeless.Witness具体化,那么这可以正常工作:

import eu.timepit.refined.api.Refined
import eu.timepit.refined.numeric.Interval.ClosedOpen
import shapeless.Witness

class Vocab(values: Vector[String]) {
  val sizeStable: Int = values.size
  val sizeWitness = Witness(sizeStable)

  type S = sizeWitness.T
  type P = ClosedOpen[Witness.`0`.T, S]

  def size: S = sizeWitness.value

  def getId(value: String): Option[Refined[Int, P]] = values.indexOf(value) match {
    case -1 => None
    case i => Some(Refined.unsafeApply[Int, P](i))
  }

  def getValue(id: Refined[Int, P]): String = values(id.value)
}

如果Scala允许单一类型的AnyVal,我们可以删除sizeWitness并定义type S = sizeStable.typeSIP-23 implementation中解除了此限制。

使用refineV现在只需使用路径相关类型vocab.P

scala> val vocab = new Vocab(Vector("foo", "bar", "baz"))
vocab: Vocab = Vocab@5fae6bb9

scala> refineV[vocab.P](2)
res0: Either[String,eu.timepit.refined.api.Refined[Int,vocab.P]] = Right(2)

scala> refineV[vocab.P](4)
res1: Either[String,eu.timepit.refined.api.Refined[Int,vocab.P]] = Left(Right predicate of (!(4 < 0) && (4 < 3)) failed: Predicate failed: (4 < 3).)

scala> refineV[vocab.P](2).map(vocab.getValue)
res2: scala.util.Either[String,String] = Right(baz)

这是有效的,因为编译器现在可以在Witness.Aux[vocab.S]个实例的范围之外找到隐式Vocab

scala> val s = implicitly[shapeless.Witness.Aux[vocab.S]]
s: shapeless.Witness.Aux[vocab.S] = shapeless.Witness$$anon$1@16cd7aa2

scala> s.value
res2: s.T = 3

refine现在使用此隐式实例构建Validate[Int, vocab.P]实例,refineV用于确定Int是否为vocab的有效索引。

答案 1 :(得分:4)

由于您用于优化Int的谓词依赖于Vocab,因此一种解决方案是为此Witness.Aux[S]添加隐式refineV和别名类:

import eu.timepit.refined._
import eu.timepit.refined.api.Refined
import eu.timepit.refined.numeric.Interval.ClosedOpen
import shapeless.Witness

class Vocab(values: Vector[String]) {
  type S <: Int
  type P = ClosedOpen[Witness.`0`.T, S]

  def size: S = values.size.asInstanceOf[S]

  def getId(value: String): Option[Refined[Int, P]] = values.indexOf(value) match {
    case -1 => None
    case i  => Some(Refined.unsafeApply[Int, P](i))
  }

  def getValue(id: Refined[Int, P]): String = values(id.value)

  implicit val witnessS: Witness.Aux[S] = Witness.mkWitness(size)

  def refine(i: Int): Either[String, Refined[Int, P]] =
    refineV[P](i)
}

现在使用Vocab.refine不需要任何其他导入:

scala> val vocab = new Vocab(Vector("foo", "bar", "baz"))
vocab: Vocab = Vocab@490b83b3

scala> vocab.refine(1)
res4: Either[String,eu.timepit.refined.api.Refined[Int,vocab.P]] = Right(1)

scala> vocab.refine(3)
res5: Either[String,eu.timepit.refined.api.Refined[Int,vocab.P]] = Left(Right predicate of (!(3 < 0) && (3 < 3)) failed: Predicate failed: (3 < 3).)