如何获取Scala函数的参数/返回类型?

时间:2015-09-05 15:17:38

标签: scala reflection anonymous-function scala-macros

我有一个函数,想获取它的参数类型和返回类型,以便在Scala宏中使用。

scala> val fn = (a: String, b: Double) => 123
fn: (String, Double) => Int = <function2>

scala> fn.getClass
res1: Class[_ <: (String, Double) => Int] = class $anonfun$1

在上面的示例中,参数类型和返回类型已经在两行打印,但我不知道如何访问它们。即使使用toString,我仍然会遇到<function2>符号右侧的class $anonfun$1=部分 - 否则可能会进行一些丑陋的字符串解析。

我发现MethodSymbolApi提供了一种为方法提取此信息的方法,但似乎这可能对此特定情况没有帮助。

我目前正在研究AST解析(作为scala.meta的一部分)来提取信息,但我认为这个问题看起来很基本,可以被标准反射库所覆盖,虽然我没能在那里找到我想要的东西。有什么想法吗?

根据@ johanandren的回答进行编辑:

我还没有找到一种更简洁的方法从TypeTag / Type中提取它们,但这确实有效。 :)

scala> val fn = (a: String, b: Double) => 123
scala> import scala.reflect.runtime.{universe => ru}
scala> def getTypeTag[T: ru.TypeTag](obj: T) = ru.typeTag[T]
scala> getTypeTag(fn).tpe.toString.split(" => ")
res179: Array[String] = Array((String, Double), Int)

2 个答案:

答案 0 :(得分:3)

getClass是Java反射API的一部分,它不太了解Scala类型,您应该查看Scala Reflection API。这应该让你开始,http://docs.scala-lang.org/overviews/reflection/overview.html

不确定,但我认为函数类型的TypeTag是你想要的。

答案 1 :(得分:3)

只是为了完整性,当你在Scala REPL中时,你可以访问类型:

scala> val fn = (a: String, b: Double) => 123
fn: (String, Double) => Int = <function2>

scala> :type fn
(String, Double) => Int

在运行时,Scala编译器不具备完整的类型信息。所以它形成了一个代码片段fn,(也有自己的符号表查找)。 https://github.com/scala/scala/blob/v2.10.5/src/compiler/scala/tools/nsc/interpreter/ILoop.scala#L449

然后将其传递给编译器,编译器然后将类型信息附加到implicit evidence类型。

(在此解释I want to get the type of a variable at runtime

scala> import scala.reflect.runtime.universe.{TypeTag, typeTag}
import scala.reflect.runtime.universe.{TypeTag, typeTag}

scala> def getTypeTag[T: TypeTag](obj: T) = typeTag[T]
getTypeTag: [T](obj: T)(implicit evidence$1: reflect.runtime.universe.TypeTag[T])reflect.runtime.universe.TypeTag[T]

现在我们有证据可以提供确切的类型信息:

scala> getTypeTag(fn)
res0: reflect.runtime.universe.TypeTag[(String, Double) => Int] = TypeTag[(String, Double) => Int]

scala> val targs = res0.tpe.typeArgs
targs: List[reflect.runtime.universe.Type] = List(String, Double, Int)

现在我们可以轻松访问这些类型:

scala> val (in, out) = (ta.init, ta.last)
in: List[reflect.runtime.universe.Type] = List(String, Double)
out: reflect.runtime.universe.Type = Int

scala> println(s"INPUTS: $in")
INPUTS: List(String, Double)

scala> println(s"OUTPUT: $out")
OUTPUT: Int