Итерация по коллекциям Java в Scala - PullRequest
110 голосов
/ 30 января 2009

Я пишу некоторый код Scala, который использует API-интерфейс Apache POI . Я хотел бы перебрать строки, содержащиеся в java.util.Iterator, которые я получаю из класса Sheet. Я хотел бы использовать итератор в цикле стиля 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 ]

251 голосов
/ 26 октября 2009

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

import scala.collection.JavaConversions._

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

26 голосов
/ 03 августа 2009

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

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

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

16 голосов
/ 31 мая 2017

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

import scala.collection.JavaConverters._

// ...

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

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

15 голосов
/ 30 января 2009

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

9 голосов
/ 21 февраля 2013

Для Scala 2.10:

// Feature warning if you don't enable implicit conversions...
import scala.language.implicitConversions
import scala.collection.convert.WrapAsScala.enumerationAsScalaIterator
5 голосов
/ 16 мая 2015

В 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
  }

}
4 голосов
/ 30 января 2009

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

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

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

val list = List.fromArray(array)
3 голосов
/ 01 января 2016

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

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

Имеет похожую концепцию, но менее многословно. ИМО:)

2 голосов
/ 07 августа 2015

Если вы хотите избежать последствий в 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 в Scala Iterator.

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

...