Question

我会确保方法只接受A或B或C的实例。我不想修改A，B和C的代码。

case class A
case class B
case class C
def method(aOrbOrc: Any) = ???

// method("toto") should not compile

Answer 1

你可以使用Miles Sabin的想法来实现联合类型（下面的代码取自Rex Kerr的variant）：

trait Contra[-A] {}
type Union[A,B,C] = {
  type Check[Z] = Contra[Contra[Z]] <:< Contra[Contra[A] with Contra[B] with Contra[C]]
}

然后做：

def method[T: Union[A,B,C]#Check](t: T) = ???

例如：

def method[T: Union[Int,String,Boolean]#Check](t:T) = ???

method(1)     // OK
method("foo") // OK
method(true)  // OK
method(1.5)   // does not compile

详细了解here。 here是指向Miles帖子的链接。

Answer 2

您可以使用Type Class。

case class A(s: String)
case class B(i: Int)
case class C(i1: Int, i2: Int)

// Type Class
trait ABC[T] { 
    def bar(t: T): String
}
// Instances
implicit object ABC_A extends ABC[A] {
    override def bar(t: A): String = "A : s = " + t.s
}
implicit object ABC_B extends ABC[B] {
    override def bar(t: B): String = "B : i = " + t.i
}
implicit object ABC_C extends ABC[C] {
    override def bar(t: C): String = "C : i1 = " + t.i1 + ", i2 = " + t.i2
}

def method[T](abc: T)(implicit instance: ABC[T]) = println(instance.bar(abc))

method(A("AAAAA")) // => A : s = AAAAA
method(B(123))     // => B : i = 123
method(C(9, 5))    // => C : i1 = 9, i2 = 5
method(1)          // compilation error

一种只应接受Scala中A或B或C实例的方法

2 个答案: