119 lines
3.1 KiB
Scala
119 lines
3.1 KiB
Scala
/**
|
|
* Copyright (C) 2009-2010 Scalable Solutions AB <http://scalablesolutions.se>
|
|
*/
|
|
|
|
package se.scalablesolutions.akka.serialization
|
|
|
|
import org.codehaus.jackson.map.ObjectMapper
|
|
|
|
import com.google.protobuf.Message
|
|
|
|
import scala.reflect.Manifest
|
|
|
|
import sbinary.DefaultProtocol
|
|
|
|
import java.io.{StringWriter, ByteArrayOutputStream, ObjectOutputStream}
|
|
|
|
import sjson.json.{Serializer=>SJSONSerializer}
|
|
|
|
object SerializationProtocol {
|
|
val SBINARY = 1
|
|
val SCALA_JSON = 2
|
|
val JAVA_JSON = 3
|
|
val PROTOBUF = 4
|
|
val JAVA = 5
|
|
val AVRO = 6
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
trait Serializable {
|
|
def toBytes: Array[Byte]
|
|
}
|
|
|
|
/**
|
|
* Serialization protocols.
|
|
*
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
object Serializable {
|
|
|
|
/**
|
|
* Example on how to use the SBinary serialization protocol:
|
|
* <pre>
|
|
* case class User(val usernamePassword: Tuple2[String, String],
|
|
* val email: String,
|
|
* val age: Int)
|
|
* extends Serializable.SBinary[User] {
|
|
* def this() = this(null, null, 0)
|
|
* import sbinary.DefaultProtocol._
|
|
* implicit object UserFormat extends Format[User] {
|
|
* def reads(in : Input) = User(
|
|
* read[Tuple2[String, String]](in),
|
|
* read[String](in),
|
|
* read[Int](in))
|
|
* def writes(out: Output, value: User) = {
|
|
* write[Tuple2[String, String]](out, value.usernamePassword)
|
|
* write[String](out, value.email)
|
|
* write[Int](out, value.age)
|
|
* }
|
|
* }
|
|
* def fromBytes(bytes: Array[Byte]) = fromByteArray[User](bytes)
|
|
* def toBytes: Array[Byte] = toByteArray(this)
|
|
* }
|
|
* </pre>
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
trait SBinary[T <: AnyRef] extends Serializable {
|
|
def fromBytes(bytes: Array[Byte]): T
|
|
def toBytes: Array[Byte]
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
trait JSON extends Serializable {
|
|
def toJSON: String
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
abstract class JavaJSON extends JSON {
|
|
|
|
def toJSON: String = {
|
|
val out = new StringWriter
|
|
val mapper = new ObjectMapper
|
|
mapper.writeValue(out, this)
|
|
out.close
|
|
out.toString
|
|
}
|
|
|
|
def toBytes: Array[Byte] = {
|
|
val bos = new ByteArrayOutputStream
|
|
val out = new ObjectOutputStream(bos)
|
|
val mapper = new ObjectMapper
|
|
mapper.writeValue(out, this)
|
|
out.close
|
|
bos.toByteArray
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
trait ScalaJSON extends JSON {
|
|
def toJSON: String = new String(toBytes, "UTF-8")
|
|
def toBytes: Array[Byte] = SJSONSerializer.SJSON.out(this)
|
|
}
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
trait Protobuf[T] extends Serializable {
|
|
def fromBytes(bytes: Array[Byte]): T = getMessage.toBuilder.mergeFrom(bytes).asInstanceOf[T]
|
|
def toBytes: Array[Byte] = getMessage.toByteArray
|
|
def getMessage: Message
|
|
}
|
|
}
|