How to make a typeclass works with an heterogenous List in scala

问题

Given the following typeclass and some instances for common types

trait Encoder[A] {
  def encode(a: A): String
}

object Encoder {
  implicit val stringEncoder = new Encoder[String] {
    override def encode(a: String): String = a
  }
  implicit val intEncoder = new Encoder[Int] {
    override def encode(a: Int): String = String.valueOf(a)
  }
  implicit def listEncoder[A: Encoder] =
    new Encoder[List[A]] {
      override def encode(a: List[A]): String = {
        val encoder = implicitly[Encoder[A]]
        a.map(encoder.encode).mkString(",")
      }
    }
}

Is there a way to make it work

Encoder.listEncoder.encode(List("a", 1))

回答1:

If you define an instance for Any

object Encoder {
  implicit val anyEncoder = new Encoder[Any] {
    override def encode(a: Any): String = a.toString
  }

  //instances for String, Int, List[A]
}

then

Encoder.listEncoder[Any].encode(List("a", 1))

will work.

You have to specify type parameter here explicitly (listEncoder[Any]) because that's how type inference work in scala. Indeed, after the type of argument of encode is inferred

Encoder.listEncoder[???].encode(List("a", 1))
//                              ^^^^^^^^^^^^
//                              List[Any]

it's too late to come back to infer the type parameter of listEncoder.

If you define a delegator function

def encode[A](a: A)(implicit encoder: Encoder[A]): String = encoder.encode(a)

or syntax

implicit class EncoderOps[A](a: A)(implicit encoder: Encoder[A]) {
  def encode: String = encoder.encode(a)
}

    // or
// implicit class EncoderOps[A](a: A) {
//   def encode(implicit encoder: Encoder[A]): String = encoder.encode(a)
// }

then you don't have to specify type parameter explicitly

encode("a")
encode(1)
encode(List("a", 1))

"a".encode
1.encode
List("a", 1).encode

By the way, List("a", 1) is not a heterogenous list, it's an ordinary homogenous list with element type Any. A heterogenous list would be

val l: String :: Int :: HNil = "a" :: 1 :: HNil

You can try a magnet instead of type class

trait Magnet {
  def encode(): String
}

object Magnet {
  implicit def fromInt(a: Int): Magnet = new Magnet {
    override def encode(): String = String.valueOf(a)
  }
  implicit def fromString(a: String): Magnet = new Magnet {
    override def encode(): String = a
  }
}

def encode(m: Magnet): String = m.encode()

encode("a")
encode(1)
List[Magnet]("a", 1).map(_.encode())

or HList instead of List[A]

sealed trait HList 
case class ::[+H, +T <: HList](head: H, tail: T) extends HList 
case object HNil extends HList 
type HNil = HNil.type

implicit class HListOps[L <: HList](l: L) {
  def ::[A](a: A): A :: L = new ::(a, l)
}

trait Encoder[A] {
  def encode(a: A): String
}

object Encoder {
  implicit val stringEncoder: Encoder[String] = new Encoder[String] {
    override def encode(a: String): String = a
  }
  implicit val intEncoder: Encoder[Int] = new Encoder[Int] {
    override def encode(a: Int): String = String.valueOf(a)
  }
  implicit val hnilEncoder: Encoder[HNil] = new Encoder[HNil] {
    override def encode(a: HNil): String = ""
  }
  implicit def hconsEncoder[H, T <: HList](implicit
                                           hEncoder: Encoder[H],
                                           tEncoder: Encoder[T]
                                          ): Encoder[H :: T] = new Encoder[H :: T] {
    override def encode(a: H :: T): String = 
      s"${hEncoder.encode(a.head)},${tEncoder.encode(a.tail)}"
  }
}

def encode[A](a: A)(implicit encoder: Encoder[A]): String = encoder.encode(a)

implicit class EncoderOps[A](a: A)(implicit encoder: Encoder[A]) {
  def encode: String = encoder.encode(a)
}

val l: String :: Int :: HNil = "a" :: 1 :: HNil
encode(l)
l.encode

来源：https://stackoverflow.com/questions/64011072/how-to-make-a-typeclass-works-with-an-heterogenous-list-in-scala

标签

scala

typeclass

implicit