Skip to content
Snippets Groups Projects
Commit ee71fa49 authored by Matei Zaharia's avatar Matei Zaharia
Browse files

Look for Kryo registrator using context class loader

parent 58eb44ac
No related branches found
No related tags found
No related merge requests found
...@@ -256,7 +256,8 @@ class KryoSerializer extends Serializer with Logging { ...@@ -256,7 +256,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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment