注册Kryo类不工作

Question

我有以下代码:

val conf = new SparkConf().setAppName("MyApp")
val sc = new SparkContext(conf)
conf.set("spark.serializer", "org.apache.spark.serializer.KryoSerializer")
new conf.registerKryoClasses(new Class<?>[]{
        Class.forName("org.apache.hadoop.io.LongWritable"),
        Class.forName("org.apache.hadoop.io.Text")
    });

但是我碰到了以下错误:

')' expected but '[' found.
[error]                 new conf.registerKryoClasses(new Class<?>[]{

如何解决这个问题?

Answer 1

您正在混合Scala和Java。在Scala中，你可以定义一个Array[Class[_]](而不是Class<?>[]):

val conf = new SparkConf()
            .setAppName("MyApp")
            .set("spark.serializer", "org.apache.spark.serializer.KryoSerializer")
            .registerKryoClasses(Array[Class[_]](
              Class.forName("org.apache.hadoop.io.LongWritable"),
              Class.forName("org.apache.hadoop.io.Text")
            ));
val sc = new SparkContext(conf)

我们甚至可以做得更好一点。为了不让我们的类在使用字符串字面值时出错，我们实际上可以利用这些类并使用classOf来获取它们的类类型:

import org.apache.hadoop.io.LongWritable
import org.apache.hadoop.io.Text
val conf = new SparkConf()
            .setAppName("MyApp")
            .set("spark.serializer", "org.apache.spark.serializer.KryoSerializer")
            .registerKryoClasses(Array[Class[_]](
              classOf[LongWritable],
              classOf[Test],
            ))
val sc = new SparkContext(conf)