Add Avro-related helpers

diff --git a/src/main/scala/gr/grnet/aquarium/message/avro/AvroHelpers.scala b/src/main/scala/gr/grnet/aquarium/message/avro/AvroHelpers.scala
new file mode 100644 (file)
index 0000000..9714773
--- /dev/null
+++ b/src/main/scala/gr/grnet/aquarium/message/avro/AvroHelpers.scala
@@ -0,0 +1,78 @@
+/*
+ * Copyright 2011-2012 GRNET S.A. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or
+ * without modification, are permitted provided that the following
+ * conditions are met:
+ *
+ *   1. Redistributions of source code must retain the above
+ *      copyright notice, this list of conditions and the following
+ *      disclaimer.
+ *
+ *   2. Redistributions in binary form must reproduce the above
+ *      copyright notice, this list of conditions and the following
+ *      disclaimer in the documentation and/or other materials
+ *      provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
+ * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
+ * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
+ * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
+ * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ *
+ * The views and conclusions contained in the software and
+ * documentation are those of the authors and should not be
+ * interpreted as representing official policies, either expressed
+ * or implied, of GRNET S.A.
+ */
+
+package gr.grnet.aquarium.message.avro
+
+import gr.grnet.aquarium.util.json.JsonHelpers
+import java.io.{OutputStream, ByteArrayOutputStream}
+import org.apache.avro.Schema
+import org.apache.avro.generic.{GenericRecord, GenericDatumWriter}
+import org.apache.avro.io.{JsonEncoder, EncoderFactory}
+import org.apache.avro.specific.{SpecificDatumWriter, SpecificRecord}
+
+
+/**
+ *
+ * @author Christos KK Loverdos <loverdos@gmail.com>
+ */
+object AvroHelpers {
+  def getJsonEncoder(schema: Schema, out: OutputStream): JsonEncoder = {
+    val encoder = EncoderFactory.get().jsonEncoder(schema, out)
+    val jsonGenerator = JsonHelpers.getJsonGenerator(out)
+    encoder.configure(jsonGenerator)
+  }
+
+  def jsonStringOfSpecificRecord[T <: SpecificRecord](t: T): String = {
+    val schema = t.getSchema()
+    val out = new ByteArrayOutputStream()
+    val encoder = getJsonEncoder(schema, out)
+    val writer = new SpecificDatumWriter[T](schema)
+
+    writer.write(t, encoder)
+    encoder.flush()
+    out.toString
+  }
+
+  def jsonStringOfGenericRecord[T <: GenericRecord](t: T): String = {
+    val schema = t.getSchema()
+    val out = new ByteArrayOutputStream()
+    val encoder = getJsonEncoder(schema, out)
+    val writer = new GenericDatumWriter[T](schema)
+
+    writer.write(t, encoder)
+    encoder.flush()
+    out.toString
+  }
+}