是否可以引用Scala案例类构造函数参数的类型?

时间:2011-12-30 16:16:42

标签: scala

我的目标是创建一个案例类可以扩展的特征,它可以处理每个构造函数参数,然后将它们作为参数传递给案例类的方法。所有构造函数参数都是具有不同类型参数的相同类型,并且该方法将采用与每个构造函数参数的type参数匹配的参数。您可以将其视为构造函数参数的预处理器。例如,

case class Foo(input1:Input[Int], input2:Input[String]) extends MagicTrait {
  def run(input1:Int, input2:String) { ... }
}

这完全可行吗?它是否可行的方式不是非常丑陋(例如所有反射)?是否可以以案例类中的所有通用方式引用案例类的伴随对象(例如,获取Companion.unapply()的输出的函数)?

2 个答案:

答案 0 :(得分:7)

看作可接受的解决方案允许将预处理功能从实例移到关联对象,主要的剩余难点是您希望能够抽象出案例类的arity和类型(即形状)构造函数。使用shapelessHList实现和多态函数值可以实现这一点。

首先是一些预赛,

import shapeless.HList._
import shapeless.Functions._
import shapeless.Poly._
import shapeless.TypeOperators._

// Implementation of your Input wrapper
case class Input[T](value: T)

// Value extractor as a shapeless polymorphic function value
object value extends (Input ~> Id) {
  def default[T](i : Input[T]) = i.value
}

我们现在可以定义一个预处理器基类,它提供一个采用HListInput类型的apply方法,在其上映射value多态函数(即执行预处理)然后将生成的HListInput类型传递给提供的case类构造函数(以hlisted形式给出,见下文),

// Pre-processer base class
abstract class Preprocessor[In <: HList, Out <: HList, R](ctor : Out => R)
  (implicit mapper : MapperAux[value.type, In, Out]) {
    def apply(in : In) = ctor(in map value)
  }

现在我们使用后处理组件类型

定义案例类
case class Foo(input1 : Int, input2 : String)

并添加一行样板,

object FooBuilder extends Preprocessor((Foo.apply _).hlisted)

(此处Foo伴侣对象工厂方法以HListed形式提供为Preprocessor构造函数参数,如上所述。)

现在我们可以使用Foo构建FooBuilder个实例。

val foo = FooBuilder(Input(23) :: Input("foo") :: HNil)

不幸的是,(目前)无法将FooBuilder对象与Foo随播广告对象合并:如果您尝试让Foo随播广告Preprocessor会发现Foo工厂方法不能作为Preprocessor构造函数参数传递。

为了说明这个解决方案真的是抽象的类型和arity,这里是我们如何添加第二个不同形状的案例类,

case class Bar(input1 : Int, input2 : String, input3 : Boolean)

object BarBuilder extends Preprocessor((Bar.apply _).hlisted)

val bar = BarBuilder(Input(23) :: Input("foo") :: Input(true) :: HNil)

答案 1 :(得分:2)

case class Input[T](value: T)

trait MagicTrait[T,U] {
   val input1: Input[T]
   val input2: Input[U]
   def run: Unit
}

case class Foo(input1: Input[Int], input2: Input[String]) 
    extends MagicTrait[Int, String] {
  def run = println(input1.value * 2 + input2.value.toUpperCase)
}

scala> val m: MagicTrait[_,_] = Foo(Input(3), Input("hi"))
m: MagicTrait[_, _] = Foo(Input(3),Input(hi))

scala> m.run
6HI

编辑:

如果要查找类参数的类型,可以使用案例类扩展Product的事实:

scala> Foo(2, "hi").productIterator.map(_.asInstanceOf[AnyRef].getClass).toList
res13: List[java.lang.Class[_]] = 
         List(class java.lang.Integer, class java.lang.String)

但这会使用您想要避免的反射。这就是我们使用参数化的原因。

如果要返回其伴随对象,我不确定您是否可以在case类的上下文中以有用的,类型安全的方式执行此操作,因为伴随对象不会扩展指定其提取器方法的接口。您可能能够对结构类型做一些事情,但很可能有更好的方法来解决您尝试解决的任何问题。