com.avsystem.commons.collection.CollectionExtensions.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of commons-shared_2.12 Show documentation

AVSystem commons library for Scala

The newest version!

package com.avsystem.commons
package collection

import scala.collection.generic.CanBuildFrom
import scala.language.higherKinds

/**
  * Author: ghik
  * Created: 08/04/16.
  */
object CollectionExtensions {
  implicit class iteratorOps[A](private val it: Iterator[A]) extends AnyVal {
    def drainTo[C[_]](n: Int)(implicit cbf: CanBuildFrom[Nothing, A, C[A]]): C[A] = {
      val builder = cbf()
      var i = 0
      while (it.hasNext && i < n) {
        builder += it.next()
        i += 1
      }
      builder.result()
    }
  }
}