Added support for custom classes. (from 49ea48)

2011-04-27 12:30:16 -07:00 · 2011-04-27 12:30:16 -07:00 · 4e4c41026c
parent 65848da8df
commit 4e4c41026c
1 changed files with 4 additions and 1 deletions
--- a/core/src/main/scala/spark/broadcast/Broadcast.scala
+++ b/core/src/main/scala/spark/broadcast/Broadcast.scala
@ -207,7 +207,10 @@ extends Logging {
  }

  private def byteArrayToObject[OUT](bytes: Array[Byte]): OUT = {
-    val in = new ObjectInputStream(new ByteArrayInputStream(bytes))
+    val in = new ObjectInputStream (new ByteArrayInputStream (bytes)){
+      override def resolveClass(desc: ObjectStreamClass) =
+        Class.forName(desc.getName, false, currentThread.getContextClassLoader)
+    }    
    val retVal = in.readObject.asInstanceOf[OUT]
    in.close()
    return retVal