ODIn/spark-instrumented-optimizer
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Look for Kryo registrator using context class loader

Browse source
This commit is contained in:
Matei Zaharia 2012-09-26 14:14:17 -07:00
parent 2498f95199
commit a417cd4d9d
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
core/src/main/scala/spark/KryoSerializer.scala
View file

@ -212,7 +212,8 @@ class KryoSerializer extends Serializer with Logging {
val regCls = System.getProperty("spark.kryo.registrator") val regCls = System.getProperty("spark.kryo.registrator")
if (regCls != null) { if (regCls != null) {
logInfo("Running user registrator: " + regCls) 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) reg.registerClasses(kryo)
} }
kryo kryo