Обогащение коллекций Scala методом

Question

Как мне добавить foreachWithIndex метод на коллекциях Scala?

Это то, что я мог придумать до сих пор:

implicit def iforeach[A, CC <: TraversableLike[A, CC]](coll: CC) = new {
  def foreachWithIndex[B](f: (A, Int) => B): Unit = {
    var i = 0
    for (c <- coll) {
      f(c, i)
      i += 1
    }
  }
}

Это не работает:

Vector(9, 11, 34).foreachWithIndex { (el, i) =>
  println(el, i)
}

Вызывает следующую ошибку:

error: value foreachWithIndex is not a member of scala.collection.immutable.Vector[Int]
Vector(9, 11, 34).foreachWithIndex { (el, i) =>

Однако код работает, когда я явно применяю метод преобразования:

iforeach[Int, Vector[Int]](Vector(9, 11, 34)).foreachWithIndex { (el, i) =>
  println(el, i)
}

Выход:

(9,0)
(11,1)
(34,2)

Как мне заставить его работать без явного применения метода конвертации? Благодарю.

Answer 1

Вам нужно расширить Iterable:

class RichIter[A, C](coll: C)(implicit i2ri: C => Iterable[A]) {
    def foreachWithIndex[B](f: (A, Int) => B): Unit = {
    var i = 0
    for (c <- coll) {
      f(c, i)
      i += 1
    }
  }
}

implicit def iter2RichIter[A, C[A]](ca: C[A])(
    implicit i2ri: C[A] => Iterable[A]
): RichIter[A, C[A]] = new RichIter[A, C[A]](ca)(i2ri)

Vector(9, 11, 34) foreachWithIndex {
  (el, i) => println(el, i)
}

выход:

(9,0)
(11,1)
(34,2)

Смотрите этот пост Рекс Керр для получения дополнительной информации.

Answer 2

Короткий ответ: вы должны параметризовать CC если вы сделаете это таким образом, или выводчик типа не может понять, что A является. Другой короткий ответ - делай так, как я описываю в ответе на этот вопрос.

Чтобы расширить немного больше, нет никакой причины, что вам нужно CC <: TraversableLike- Просто возьми Traversable и начать с iforeach[A](coll: Traversable[A])! Вам не нужно использовать причудливые границы типов, чтобы использовать суперкласс / супертрейт. Если вы хотите сделать что-то более сложное, когда вы возвращаете другую коллекцию с сохраненным типом коллекции, тогда вам нужно использовать компоновщики и тому подобное, что я опишу в другом вопросе.

Answer 3

Если то, что вас интересует, это итерация только с индексом, вы можете просто пропустить всю часть сутенерства и сделать что-то вроде

coll.zipWithIndex.foreach { case (elem, index) =>
  /* ... */
}