Scala ArrayBuffer/ListBuffer vs Java ArrayList

在我们编写Scala代码的时候，由于ListBuffer方便的特性，所以经常会使用ListBuffer。过去也一直没有什么问题。

但是呢，最近在我们的项目中，出现了一个非常严重的性能问题。通过JProfiler分析，发现是调用ListBuffer的获取数据的方法时，太慢导致的。

差距有多大呢？以前需要运行12个小时，改成了Java的List以后，直接缩短了四个小时。

这篇文章中，我不会具体分析ArrayBuffer/ScalaBuffer等的代码，只会给出测试的结果。

Scala ArrayBuffer

package com.hyper

import scala.collection.JavaConverters._
import scala.collection.mutable.ArrayBuffer

object TestScalaList {

    def main(args: Array[String]): Unit = {

        val scalaListBuffer: ArrayBuffer[String] = ArrayBuffer[String]()
        val itemNumber = 100000

        for (i <- 0 until itemNumber) {
            scalaListBuffer += i.toString
        }

        IterateList.iterate(scalaListBuffer.toList.asJava)
    }

}

IterateList的代码如下:

package com.hyper;

import com.google.common.base.Stopwatch;

import java.util.List;
import java.util.concurrent.TimeUnit;

public class IterateList {

    public static void iterate(List<String> list) {
        int listLength = list.size();

        System.out.println("List length: " + listLength);

        Stopwatch stopwatch = new Stopwatch();
        stopwatch.start();

        System.out.println("List class: " + list.getClass());
        for (int i = 0; i < listLength; i++) {
            list.get(i);
        }

        long elapse = stopwatch.elapsed(TimeUnit.SECONDS);
        System.out.println("elapse: " + elapse);
    }

}

运行上面的代码，我们可以看到如下结果:

运行了15s。时间不短。它的底层还是数组实现的呢。

Java ArrayList

那我们再来看下Java ArrayList的性能。

package com.hyper;

import java.util.ArrayList;
import java.util.List;

public class TestJavaList {

    public static void main(String[] args) {
        List<String> list = new ArrayList<>();

        int itemNumber = 100000;

        for (int i = 0; i < itemNumber; i++) {
            list.add(String.valueOf(i));
        }

        IterateList.iterate(list);
    }

}

结果如下:

仅仅用了0s，1s都不到。

都是用数组实现的，差距咋这么大呢？

我没有仔细探究，猜测一个是因为SeqWrapper这东西，增加了调用栈的深度，进一步增加了运行时间，但是这个应该不会导致这么大的差距。所以更可能的原因是，Scala实现的这个ArrayBuffer就有问题。

有时间仔细研究一下源码和字节码，补充上来。

Scala ListBuffer

这个是我们Scala中最常用的数据结构，Spark代码中也有意无意的总用到这个数据结构。

我们都知道，Java中ArrayList和LinkedList这两种数据结构，各有各的优缺点，各有各的应用场景。在Scala中，ArrayBuffer就是ArrayList的另一个版本，而ListBuffer就是LinkedList的另一个版本。所以，拿ListBuffer那两个数据结构对比，有点不公平。毕竟ListBuffer不适合随机读取。

但是既然做了，这里还是贴上来。

package com.hyper

import scala.collection.JavaConverters._
import scala.collection.mutable.ListBuffer

object TestScalaList {

    def main(args: Array[String]): Unit = {

        val scalaListBuffer: ListBuffer[String] = ListBuffer[String]()
        val itemNumber = 100000

        for (i <- 0 until itemNumber) {
            scalaListBuffer += i.toString
        }

        IterateList.iterate(scalaListBuffer.toList.asJava)
    }

}

结果如下:

Oh My God，竟然运行了45s。

总结

从测试结果中，我们可以看到，Scala代码，性能跟Java代码还是有一定差距的。

所以能用Java还是尽量用Java为好。