From ee71fa49c1d958f1723fd3fba59b8e247cb4af76 Mon Sep 17 00:00:00 2001
From: Matei Zaharia <matei@eecs.berkeley.edu>
Date: Wed, 26 Sep 2012 14:14:17 -0700
Subject: [PATCH] Look for Kryo registrator using context class loader
---
core/src/main/scala/spark/KryoSerializer.scala | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/core/src/main/scala/spark/KryoSerializer.scala b/core/src/main/scala/spark/KryoSerializer.scala
index 8a3f565071..4b95e05bd3 100644
--- a/core/src/main/scala/spark/KryoSerializer.scala
+++ b/core/src/main/scala/spark/KryoSerializer.scala
@@ -256,7 +256,8 @@ class KryoSerializer extends Serializer with Logging {
val regCls = System.getProperty("spark.kryo.registrator")
if (regCls != null) {
logInfo("Running user registrator: " + regCls)
- val reg = Class.forName(regCls).newInstance().asInstanceOf[KryoRegistrator]
+ val classLoader = Thread.currentThread.getContextClassLoader
+ val reg = Class.forName(regCls, true, classLoader).newInstance().asInstanceOf[KryoRegistrator]
reg.registerClasses(kryo)
}
kryo
--
GitLab