add explicit types

msgpack4z · Dec 24, 2023 · f1a6d2b · f1a6d2b
1 parent 651aae7
commit f1a6d2b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/main/scala/msgpack4z/MsgpackUnion.scala b/src/main/scala/msgpack4z/MsgpackUnion.scala
@@ -253,7 +253,7 @@ object MsgpackUnion {
   val string: Extractor[String] = new Extractor[String] {
     override def unapply(value: MsgpackUnion) =
       value.string
-    override def apply(value: String) =
+    override def apply(value: String): MsgpackUnion =
       new MsgpackString(value)
   }
   val binary: Extractor[Array[Byte]] = new Extractor[Array[Byte]] {
@@ -289,13 +289,13 @@ object MsgpackUnion {
   val map: Extractor[Map[MsgpackUnion, MsgpackUnion]] = new Extractor[Map[MsgpackUnion, MsgpackUnion]] {
     override def unapply(value: MsgpackUnion) =
       value.map
-    override def apply(value: Map[MsgpackUnion, MsgpackUnion]) =
+    override def apply(value: Map[MsgpackUnion, MsgpackUnion]): MsgpackUnion =
       new MsgpackMap(value)
   }
   val imap: Extractor[IMap[MsgpackUnion, MsgpackUnion]] = new Extractor[IMap[MsgpackUnion, MsgpackUnion]] {
     override def unapply(value: MsgpackUnion) =
       value.imap
-    override def apply(value: IMap[MsgpackUnion, MsgpackUnion]) = {
+    override def apply(value: IMap[MsgpackUnion, MsgpackUnion]): MsgpackUnion = {
       val builder = Map.newBuilder[MsgpackUnion, MsgpackUnion]
       value.foldlWithKey(())((_, k, v) => builder += ((k, v)))
       new MsgpackMap(builder.result())
@@ -306,7 +306,7 @@ object MsgpackUnion {
   val ext: Extractor[(Byte, Array[Byte])] = new Extractor[(Byte, Array[Byte])] {
     override def unapply(value: MsgpackUnion) =
       value.ext
-    override def apply(t: (Byte, Array[Byte])) =
+    override def apply(t: (Byte, Array[Byte])): MsgpackUnion =
       new MsgpackExt(t._1, t._2)
   }
   val bool: Boolean => MsgpackUnion = { value =>

diff --git a/src/test/scala/msgpack4z/CaseClassExample.scala b/src/test/scala/msgpack4z/CaseClassExample.scala
@@ -13,8 +13,8 @@ object CaseClassExample extends Scalaprops {
 
   val `case class map example` = forAll {
     val factory = new PackerUnpackerFactory {
-      def packer = MsgOutBuffer.create()
-      def unpacker(bytes: Array[Byte]) = MsgInBuffer(bytes)
+      def packer: MsgPacker = MsgOutBuffer.create()
+      def unpacker(bytes: Array[Byte]): MsgUnpacker = MsgInBuffer(bytes)
     }
 
     val mapCodec = CaseMapCodec.string(factory)