Итерация по коллекциям Java в Scala

Я пишу некоторый код Scala, который использует Apache POI API. Я хотел бы перебрать строки, содержащиеся в java.util.Iterator что я получаю из класса листа. Я хотел бы использовать итератор в for each цикл стилей, поэтому я пытался преобразовать его в собственную коллекцию Scala, но безуспешно.

Я посмотрел на классы / черты оболочки Scala, но не могу понять, как правильно их использовать. Как перебрать коллекцию Java в Scala, не используя многословный while(hasNext()) getNext() стиль петли?

Вот код, который я написал на основе правильного ответа:

class IteratorWrapper[A](iter:java.util.Iterator[A])
{
    def foreach(f: A => Unit): Unit = {
        while(iter.hasNext){
          f(iter.next)
        }
    }
}

object SpreadsheetParser extends Application
{
    implicit def iteratorToWrapper[T](iter:java.util.Iterator[T]):IteratorWrapper[T] = new IteratorWrapper[T](iter)

    override def main(args:Array[String]):Unit =
    {
        val ios = new FileInputStream("assets/data.xls")
        val workbook = new HSSFWorkbook(ios)
        var sheet = workbook.getSheetAt(0)
        var rows = sheet.rowIterator()

        for (val row <- rows){
            println(row)
        }
    }
}

9 ответов

Решение

Есть класс-обёртка (scala.collection.jcl.MutableIterator.Wrapper). Так что, если вы определите

implicit def javaIteratorToScalaIterator[A](it : java.util.Iterator[A]) = new Wrapper(it)

тогда он будет действовать как подкласс итератора Scala, так что вы можете сделать foreach,

Начиная с Scala 2.8 все, что вам нужно сделать, это импортировать объект JavaConversions, который уже объявляет соответствующие преобразования.

import scala.collection.JavaConversions._

Это не будет работать в предыдущих версиях.

Scala 2.12.0 устарела scala.collection.JavaConversions, так что с 2.12.0 один из способов сделать это будет что-то вроде:

import scala.collection.JavaConverters._

// ...

for(k <- javaCollection.asScala) {
    // ...
}

(обратите внимание на импорт, новым является JavaConverters, не рекомендуется JavaConversions)

Правильный ответ здесь - определить неявное преобразование из Java Iterator в какой-то пользовательский тип. Этот тип должен реализовывать foreach метод, который делегирует основному Iterator, Это позволит вам использовать Scala forс любой Java Iterator,

Для Scala 2.10:

// Feature warning if you don't enable implicit conversions...
import scala.language.implicitConversions
import scala.collection.convert.WrapAsScala.enumerationAsScalaIterator

Если вы выполняете итерацию по большому набору данных, то вы, вероятно, не хотите загружать всю коллекцию в память с помощью .asScala неявное преобразование. В этом случае удобный способ заключается в реализации scala.collection.Iterator черта характера

import java.util.{Iterator => JIterator}

def scalaIterator[T](it: JIterator[T]) = new Iterator[T] {
  override def hasNext = it.hasNext
  override def next() = it.next()
} 

val jIterator: Iterator[String] = ... // iterating over a large dataset
scalaIterator(jIterator).take(2).map(_.length).foreach(println)  // only first 2 elements are loaded to memory

У него похожая концепция, но менее многословная IMO:)

В Scala 2.10.4+ (и, возможно, ранее) возможно неявное преобразование java.util.Iterator[A] в scala.collection.Iterator[A] путем импорта scala.collection.JavaConversions.asScalaIterator. Вот пример:

object SpreadSheetParser2 extends App {

  import org.apache.poi.hssf.usermodel.HSSFWorkbook
  import java.io.FileInputStream
  import scala.collection.JavaConversions.asScalaIterator

  val ios = new FileInputStream("data.xls")
  val workbook = new HSSFWorkbook(ios)
  var sheet = workbook.getSheetAt(0)
  val rows = sheet.rowIterator()

  for (row <- rows) {
    val cells = row.cellIterator()
    for (cell <- cells) {
      print(cell + ",")
    }
    println
  }

}

Вы можете преобразовать коллекцию Java в массив и использовать это:

val array = java.util.Arrays.asList("one","two","three").toArray
array.foreach(println)

Или продолжайте и конвертируйте массив в список Scala:

val list = List.fromArray(array)

Если вы хотите избежать последствий для scala.collection.JavaConversions, вы можете использовать scala.collection.JavaConverters для явного преобразования.

scala> val l = new java.util.LinkedList[Int]()
l: java.util.LinkedList[Int] = []

scala> (1 to 10).foreach(l.add(_))

scala> val i = l.iterator
i: java.util.Iterator[Int] = java.util.LinkedList$ListItr@11eadcba

scala> import scala.collection.JavaConverters._
import scala.collection.JavaConverters._

scala> i.asScala.mkString
res10: String = 12345678910

Обратите внимание на использование asScala способ конвертировать Java Iterator Скала Iterator,

JavaConverters были доступны с Scala 2.8.1.

Другие вопросы по тегам