在无形Coproduct上进行模式匹配的清洁方式

时间:2017-04-16 18:49:32

标签: scala shapeless

我发现了两种可以在无形副产品上进行模式匹配的方法。我也搜索了这个主题并找到了this

import shapeless._

object ShapelessEx3 extends App {
   case class Red()
   case class Green()
   case class Blue()
   type Colors = Red :+: Green :+: Blue :+: CNil
   val blue = Coproduct[Colors](Blue())
   val green = Coproduct[Colors](Green())
   printColor1(blue)
   printColor2(green)

   def printColor1(c: Colors) : Unit = {
      (c.select[Red], c.select[Green], c.select[Blue]) match {
         case (Some(_), None, None) => println("Color is red")
         case (None, Some(_), None) => println("Color is green")
         case (None, None, Some(_)) => println("Color is blue")
         case _ => println("unknown color")
      }
   }
   def printColor2(c: Colors) : Unit = {
      c match {
         case Inl(Red()) => println("color is red")
         case Inr(Inl(Green())) => println("color is green")
         case Inr(Inr(Inl(Blue()))) => println("color is blue")
         case _ => println("unknown color")
      }
   }
}

但这两个函数都非常嘈杂,如case (None, Some(_), None)Inr(Inr(Inl(Blue())))。我怎么能像

这样简单的模式匹配
c match {
  case Red() =>
  case Green() => 
  case Blue() => 
}

2 个答案:

答案 0 :(得分:6)

您可以使用Poly1,例如

import shapeless._
case class Red()
case class Green()
case class Blue()
type Colors = Red :+: Green :+: Blue :+: CNil
val blue = Coproduct[Colors](Blue())
val green = Coproduct[Colors](Green())

object printColor extends Poly1 {
  implicit def caseRed = at[Red] {r=> println("red") ;r    }
  implicit def caseBlue = at[Blue]{b=> println("blue");b  }
  implicit def caseGreen= at[Green]{g =>println("green");g }
}                                                             
green map printColor
blue map printColor

答案 1 :(得分:0)

只是扩展了先前的答案。 Coproduct#fold 方法更适合匹配大小写功能。

import shapeless._
case class Red()
case class Green()
case class Blue()
type Colors = Red :+: Green :+: Blue :+: CNil

val blue = Coproduct[Colors](Blue())

object printColor extends Poly1 {
  implicit def caseRed   = at[Red](red => println("red"))
  implicit def caseBlue  = at[Blue](blue => println("blue"))
  implicit def caseGreen = at[Green](green => println("green"))
}

// This is equivalent to a match - case block
val printBlueResult: Unit = blue.fold(printColor)

object mapColor extends Poly1 {
  implicit def caseRed   = at[Red](red => 1)
  implicit def caseBlue  = at[Blue](blue => "a")
  implicit def caseGreen = at[Green](green => true)
}

// This maps the Coproduct to a new Coproduct
val mapBlueResult: Int :+: Boolean :+: String :+: CNil = blue.map(mapColor)