scala - 获取 Scala 枚举中值的上一个和下一个元素

Question

我想向 Scala 添加两个新操作 Enumeration获取给定特定值(如果存在)的上一个和下一个值。例如，我想写这样的东西:

object Nums extends MyNewEnumerationType {
  type Nums = Value
  val One,Two,Three = Value
}

Nums.nextOf(One) // Some(Two)
Nums.prevOf(One) // None

我的想法是创建一个新类并以这种方式添加方法:

class PrevNextEnum extends Enumeration {

  val prevOf = values.zip(None +: values.map{_.some}.toSeq).toMap
  val nextOf = {
    if (values.isEmpty) Map.empty
    else values.zip(values.tail.map{_.some}.toSeq :+ None).toMap
  }

}

问题是这不起作用，因为当 prevOf 和 nextOf 被初始化时，values 是空的。

第一个问题:为什么values是空的，什么时候填入值？

第二个问题:如何实现prevOf和nextOf？

第三个问题:是否可以将方法prevOf和nextOf添加到值类型而不是枚举？写 One.next 感觉比写 Num.nextOf(One)

更自然

Answer 1

尝试以下代码:

class PrevNextEnum extends Enumeration {

 lazy val prevOf = {
   val list = values.toList
   val map = list.tail.zip(list.map(Some(_))).toMap + (list.head -> None)
   map
 }
 lazy val nextOf = {
  val list = values.toList
  val map = (list.zip(list.tail.map(Some(_)) :+ None).toMap)
  map
 }
}

object Nums extends PrevNextEnum {
 type Nums = Value
 val One, Two, Three = Value
}

object App extends App {
  println(Nums.prevOf(Nums.Two))
  println(Nums.nextOf(Nums.One))
  println(Nums.nextOf(Nums.Three))
  println(Nums.prevOf(Nums.One))
}