繁体   English   中英

在Scala中迭代Java集合

[英]Iterating over Java collections in Scala

我正在编写一些使用Apache POI API的Scala代码。 我想迭代我从Sheet类中获得的java.util.Iterator中包含的行。 我想在for each样式循环中使用迭代器,所以我一直在尝试将其转换为原生Scala集合,但没有运气。

我查看了Scala包装器类/特性,但我看不出如何正确使用它们。 如何在不使用详细的while(hasNext()) getNext()样式的循环的情况下迭代Scala中的Java集合?

这是我根据正确答案编写的代码:

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

从Scala 2.8开始,您所要做的就是导入JavaConversions对象,该对象已经声明了适当的转换。

import scala.collection.JavaConversions._

但这在以前的版本中不起作用。

有一个包装类( scala.collection.jcl.MutableIterator.Wrapper )。 所以,如果你定义

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

然后它将充当Scala迭代器的子类,因此您可以执行foreach

编辑 :Scala 2.13.0弃用了scala.collection.JavaConverters ,因此从2.13.0开始,您需要使用scala.jdk.CollectionConverters

Scala 2.12.0弃用了scala.collection.JavaConversions ,因此从2.12.0开始,这样做的方法如下:

import scala.collection.JavaConverters._

// ...

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

(注意导入,new是JavaConverters,不推荐使用JavaConversions)

这里的正确答案是定义从Java的Iterator到某些自定义类型的隐式转换。 此类型应实现委托给底层Iteratorforeach方法。 这将允许您将Scala for -loop与任何Java Iterator

对于Scala 2.10:

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

使用Scala 2.10.4+(可能更早),可以通过导入scala.collection.JavaConversions.asScalaIterator将java.util.Iterator [A]隐式转换为scala.collection.Iterator [A]。 这是一个例子:

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)

如果您正在迭代大型数据集,那么您可能不希望使用.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.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

自Scala 2.8.1以来,JavaConverters一直可用。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM