Итерация по коллекциям 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.