#16 Enhance codec with non-streaming methods (#27)

zio · Mar 8, 2021 · be72f2e · be72f2e
1 parent 3b8c4d8
commit be72f2e
Show file tree

Hide file tree

Showing 3 changed files with 231 additions and 138 deletions.
diff --git a/core/src/main/scala/zio/schema/codec/Codec.scala b/core/src/main/scala/zio/schema/codec/Codec.scala
@@ -1,9 +1,13 @@
 package zio.schema.codec
 
+import zio.Chunk
 import zio.schema._
 import zio.stream.ZTransducer
 
 trait Codec {
   def encoder[A](schema: Schema[A]): ZTransducer[Any, Nothing, A, Byte]
   def decoder[A](schema: Schema[A]): ZTransducer[Any, String, Byte, A]
+
+  def encode[A](schema: Schema[A]): A => Chunk[Byte]
+  def decode[A](schema: Schema[A]): Chunk[Byte] => Either[String, A]
 }
diff --git a/core/src/main/scala/zio/schema/codec/ProtobufCodec.scala b/core/src/main/scala/zio/schema/codec/ProtobufCodec.scala
@@ -15,12 +15,21 @@ object ProtobufCodec extends Codec {
         ZIO.succeed(opt.map(values => values.flatMap(Encoder.encode(None, schema, _))).getOrElse(Chunk.empty))
     )
 
+  override def encode[A](schema: Schema[A]): A => Chunk[Byte] = a => Encoder.encode(None, schema, a)
+
   override def decoder[A](schema: Schema[A]): ZTransducer[Any, String, Byte, A] =
     ZTransducer.fromPush(
       (opt: Option[Chunk[Byte]]) =>
         ZIO.fromEither(opt.map(chunk => Decoder.decode(schema, chunk).map(Chunk(_))).getOrElse(Right(Chunk.empty)))
     )
 
+  override def decode[A](schema: Schema[A]): Chunk[Byte] => Either[String, A] =
+    ch =>
+      if (ch.isEmpty)
+        Left("No bytes to decode")
+      else
+        Decoder.decode(schema, ch)
+
   object Protobuf {
 
     sealed trait WireType {}