scala - 移动范围的函数

Question

是否有一个现有的函数可以移动 Range 产生另一个 Range，比如

val r = 1 to 5
val s = r.map(_ + 2)  // produces Vector(3, 4, 5, 6, 7)

我想得到 3 到 7。

Answer 1

下面是我将如何实现它:

implicit class RangeHasShift(val r: Range) extends AnyVal {
  def shift(n: Int): Range = {
    val start1 = r.start + n
    val end1   = r.end   + n

    // overflow check
    if ((n > 0 && (start1 < r.start || end1 < r.end)) ||
        (n < 0 && (start1 > r.start || end1 > r.end)))
      throw new IllegalArgumentException(s"$r.shift($n) causes number overflow")

    if (r.isInclusive)
      new Range.Inclusive(start1, end1, r.step)
    else
      new Range          (start1, end1, r.step)
  }
}

def check(r: Range) = assert(r == r.shift(123).shift(-123))

check(1 to 10)
check(1 to -1)
check(1 to -1 by -1)
check(1 to 10 by 3)
check(1 until 10)
check(1 until -1)
check(1 until -1 by -1)
check(1 until 10 by 3)

我想知道这是否存在于 API 的某处？