165 lines
5.5 KiB
HTML
165 lines
5.5 KiB
HTML
|
|
<html>
|
|
<head>
|
|
<link href='./../_highlighter/SyntaxHighlighter.css' rel='stylesheet' type='text/css'/>
|
|
<script language='javascript' src='./../_highlighter/shAll.js'></script>
|
|
</head>
|
|
<body>
|
|
<pre name="code" class="scala" style="width:100%">
|
|
/**
|
|
* Copyright (C) 2009 Scalable Solutions.
|
|
*/
|
|
|
|
package se.scalablesolutions.akka.serialization
|
|
|
|
import com.google.protobuf.Message
|
|
import java.io.{ObjectOutputStream, ByteArrayOutputStream, ObjectInputStream, ByteArrayInputStream}
|
|
import reflect.{BeanProperty, Manifest}
|
|
import sbinary.DefaultProtocol
|
|
import org.codehaus.jackson.map.ObjectMapper
|
|
import sjson.json.{Serializer =>SJSONSerializer}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
trait Serializer {
|
|
def deepClone(obj: AnyRef): AnyRef
|
|
def out(obj: AnyRef): Array[Byte]
|
|
def in(bytes: Array[Byte], clazz: Option[Class[_]]): AnyRef
|
|
}
|
|
|
|
// For Java API
|
|
class SerializerFactory {
|
|
import Serializer._
|
|
def getJava: Java.type = Java
|
|
def getJavaJSON: JavaJSON.type = JavaJSON
|
|
def getScalaJSON: ScalaJSON.type = ScalaJSON
|
|
def getSBinary: SBinary.type = SBinary
|
|
def getProtobuf: Protobuf.type = Protobuf
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
object Serializer {
|
|
val EMPTY_CLASS_ARRAY = Array[Class[_]]()
|
|
val EMPTY_ANY_REF_ARRAY = Array[AnyRef]()
|
|
|
|
object NOOP extends Serializer {
|
|
def deepClone(obj: AnyRef): AnyRef = obj
|
|
def out(obj: AnyRef): Array[Byte] = obj.asInstanceOf[Array[Byte]]
|
|
def in(bytes: Array[Byte], clazz: Option[Class[_]]): AnyRef = bytes
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
object Java extends Serializer {
|
|
def deepClone(obj: AnyRef): AnyRef = in(out(obj), None)
|
|
|
|
def out(obj: AnyRef): Array[Byte] = {
|
|
val bos = new ByteArrayOutputStream
|
|
val out = new ObjectOutputStream(bos)
|
|
out.writeObject(obj)
|
|
out.close
|
|
bos.toByteArray
|
|
}
|
|
|
|
def in(bytes: Array[Byte], clazz: Option[Class[_]]): AnyRef = {
|
|
val in = new ObjectInputStream(new ByteArrayInputStream(bytes))
|
|
val obj = in.readObject
|
|
in.close
|
|
obj
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
object Protobuf extends Serializer {
|
|
def deepClone(obj: AnyRef): AnyRef = in(out(obj), Some(obj.getClass))
|
|
|
|
def out(obj: AnyRef): Array[Byte] = {
|
|
if (!obj.isInstanceOf[Message]) throw new IllegalArgumentException("Can't serialize a non-protobuf message using protobuf [" + obj + "]")
|
|
obj.asInstanceOf[Message].toByteArray
|
|
}
|
|
|
|
def in(bytes: Array[Byte], clazz: Option[Class[_]]): AnyRef = {
|
|
if (!clazz.isDefined) throw new IllegalArgumentException("Need a protobuf message class to be able to serialize bytes using protobuf")
|
|
// TODO: should we cache this method lookup?
|
|
val message = clazz.get.getDeclaredMethod("getDefaultInstance", EMPTY_CLASS_ARRAY: _*).invoke(null, EMPTY_ANY_REF_ARRAY: _*).asInstanceOf[Message]
|
|
message.toBuilder().mergeFrom(bytes).build
|
|
}
|
|
|
|
def in(bytes: Array[Byte], clazz: Class[_]): AnyRef = {
|
|
if (clazz == null) throw new IllegalArgumentException("Protobuf message can't be null")
|
|
in(bytes, Some(clazz))
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
object JavaJSON extends Serializer {
|
|
private val mapper = new ObjectMapper
|
|
|
|
def deepClone(obj: AnyRef): AnyRef = in(out(obj), Some(obj.getClass))
|
|
|
|
def out(obj: AnyRef): Array[Byte] = {
|
|
val bos = new ByteArrayOutputStream
|
|
val out = new ObjectOutputStream(bos)
|
|
mapper.writeValue(out, obj)
|
|
out.close
|
|
bos.toByteArray
|
|
}
|
|
|
|
def in(bytes: Array[Byte], clazz: Option[Class[_]]): AnyRef = {
|
|
if (!clazz.isDefined) throw new IllegalArgumentException("Can't deserialize JSON to instance if no class is provided")
|
|
val in = new ObjectInputStream(new ByteArrayInputStream(bytes))
|
|
val obj = mapper.readValue(in, clazz.get).asInstanceOf[AnyRef]
|
|
in.close
|
|
obj
|
|
}
|
|
|
|
def in(json: String, clazz: Class[_]): AnyRef = {
|
|
if (clazz == null) throw new IllegalArgumentException("Can't deserialize JSON to instance if no class is provided")
|
|
mapper.readValue(json, clazz).asInstanceOf[AnyRef]
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
object ScalaJSON extends Serializer {
|
|
def deepClone(obj: AnyRef): AnyRef = in(out(obj), None)
|
|
|
|
def out(obj: AnyRef): Array[Byte] = SJSONSerializer.SJSON.out(obj)
|
|
|
|
def in(bytes: Array[Byte], clazz: Option[Class[_]]): AnyRef = SJSONSerializer.SJSON.in(bytes)
|
|
|
|
def in(json: String): AnyRef = SJSONSerializer.SJSON.in(json)
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
object SBinary {
|
|
import sbinary.DefaultProtocol._
|
|
|
|
def deepClone[T <: AnyRef](obj: T)(implicit w : Writes[T], r : Reads[T]): T = in[T](out[T](obj), None)
|
|
|
|
def out[T](t : T)(implicit bin : Writes[T]): Array[Byte] = toByteArray[T](t)
|
|
|
|
def in[T](array : Array[Byte], clazz: Option[Class[T]])(implicit bin : Reads[T]): T = fromByteArray[T](array)
|
|
|
|
def in[T](array : Array[Byte])(implicit bin : Reads[T]): T = fromByteArray[T](array)
|
|
}
|
|
}
|
|
|
|
</pre>
|
|
<script language='javascript'>
|
|
dp.SyntaxHighlighter.ClipboardSwf = './../_highlighter/clipboard.swf';
|
|
dp.SyntaxHighlighter.HighlightAll('code');
|
|
</script>
|
|
</body>
|
|
</html>
|