Spark、Kryo 序列化问题与 ProtoBuf 字段

Question

在转换 RDD 时，我在运行与 protobuf 字段的序列化相关的 spark 作业时看到错误。

com.esotericsoftware.kryo.KryoException：java.lang.UnsupportedOperationException 序列化跟踪：otherAuthors_ (com.thomsonreuters.kraken.medusa.dbor.proto.Book$DBBooks)

错误似乎是在这一点上创建的：

val booksPerTier: Iterable[(TimeTier, RDD[DBBooks])] = allTiers.map {
      tier => (tier, books.filter(b => isInTier(endOfInterval, tier, b) &&     !isBookPublished(o)).mapPartitions( it =>
      it.map{ord =>
        (ord.getAuthor, ord.getPublisherName, getGenre(ord.getSourceCountry))}))
}

val averagesPerAuthor = booksPerTier.flatMap { case (tier, opt) =>
  opt.map(o => (tier, o._1, PublisherCompanyComparison, o._3)).countByValue()
}

val averagesPerPublisher = booksPerTier.flatMap { case (tier, opt) =>
  opt.map(o => (tier, o._1, PublisherComparison(o._2), o._3)).countByValue()
}

该字段是 protobuf 中指定的列表，如下所示：

otherAuthors_ = java.util.Collections.emptyList()

如您所见，该代码实际上并未利用 Book Protobuf 中的该字段，尽管它仍在通过网络传输。

有没有人对此有任何建议？

Answer 1

好的，老问题，但这是给后代的答案。 默认 kryo 序列化程序不适用于某些集合。 有一个第三方库可以帮助它： kryo-serializers

在您的情况下，您可能需要在创建 spark 配置时提供自定义 kryo 注册器：

val conf = new SparkConf()
conf.set("spark.kryo.registrator", "MyKryoRegistrator")

在您的注册器中使用所需的自定义注册：

class MyKryoRegistrator extends KryoRegistrator {
    override def registerClasses(kryo: Kryo) {
        kryo.register( Collections.EMPTY_LIST.getClass(), new CollectionsEmptyListSerializer() );
        // Probably should use proto serializer for your proto classes
        kryo.register( Book.class, new ProtobufSerializer() );
    } 
}