gistfile1.scala

class KryoInjection2(val output: Output) extends Injection[AnyRef, Array[Byte]] {
  val input: Input = new Input()
  val kryo = KryoBijection.getKryo

  def register(clazz: Class[_]) {
    kryo.register(clazz)
  }

  def apply(obj: AnyRef): Array[Byte] = {
    output.clear()
    kryo.writeClassAndObject(output, obj)
    output.toBytes
  }

  def invert(b: Array[Byte]): Option[AnyRef] = {
    input.setBuffer(b)
    Option(allCatch.opt(kryo.readClassAndObject(input)).orNull)
  }
}