Тип функции метода метода из экземпляра MethodMirror в Scala

Question

Предположим, у меня есть экземпляр MethodMirror, созданный для определенного метода объекта. По полям зеркала я могу легко получить доступ к типу возвращаемого значения и параметрам метода. Но мне действительно нужно получить тип, который этот метод имел бы как функцию.

Вот пример кода игрушки, который поможет мне объяснить, чего я хочу достичь. Я использую Scala 2.11.6.

import scala.reflect.runtime.universe._ 

object ForStackOverflow { 
    object Obj { 
    def method(x:String, y:String):Int = 0 
    def expectedRetType():((String, String) => Int) = ??? 
    } 

    def main(args: Array[String]) { 
    val mirror:Mirror = runtimeMirror(getClass.getClassLoader) 
    val instanceMirror = mirror.reflect(Obj) 

    val methodSymbol:MethodSymbol = instanceMirror.symbol.toType.decl(TermName("method")).asMethod 
    val methodMirror = instanceMirror.reflectMethod(methodSymbol) 

    println(methodMirror.symbol.returnType) 
    println(methodMirror.symbol.paramLists(0).map { x => x.info.resultType }.mkString(", ")) 

    val expectedSymbol:MethodSymbol = instanceMirror.symbol.toType.decl(TermName("expectedRetType")).asMethod 
    println("I would like to produce from a 'methodMirror' this: "+expectedSymbol.returnType) 
    } 
} 

Я хочу, чтобы произвести Type экземпляр из methodMirror, который будет представлять собой функцию. Для этого примера это должно быть (String, String) => Int. Я бы предпочел решение, которое не слишком сильно зависит от конкретных классов Scala FunctionX.

Answer 1

Метод getEtaExpandedMethodType ниже выполняет то, что вы просили, и даже обрабатывает методы с несколькими списками параметров.

С другой стороны, он не обрабатывает общие методы. На примере def method[T](x: T) = 123, когда eta-extended, создает функцию типа Any => Int, но getEtaExpandedMethodType сообщит T => Int, который не только неверен, но и не имеет смысла вообще (T не имеет смысла в этом контексте).

def getEtaExpandedMethodType(methodSymbol: MethodSymbol): Type = { 
    val typ = methodSymbol.typeSignature 
    def paramType(paramSymbol: Symbol): Type = { 
    // TODO: handle the case where paramSymbol denotes a type parameter 
    paramSymbol.typeSignatureIn(typ) 
    } 

    def rec(paramLists: List[List[Symbol]]): Type = { 
    paramLists match { 
     case Nil => methodSymbol.returnType 
     case params :: otherParams => 
     val functionClassSymbol = definitions.FunctionClass(params.length) 
     appliedType(functionClassSymbol, params.map(paramType) :+ rec(otherParams)) 
    } 
    } 
    if (methodSymbol.paramLists.isEmpty) { // No arg method 
    appliedType(definitions.FunctionClass(0), List(methodSymbol.returnType)) 
    } else { 
    rec(methodSymbol.paramLists) 
    } 
} 
def getEtaExpandedMethodType(methodMirror: MethodMirror): Type = getEtaExpandedMethodType(methodMirror.symbol) 

тест РЕПЛ:

scala> val mirror: Mirror = runtimeMirror(getClass.getClassLoader) 
mirror: reflect.runtime.universe.Mirror = ... 

scala> val instanceMirror = mirror.reflect(Obj) 
instanceMirror: reflect.runtime.universe.InstanceMirror = instance mirror for Obj$@21b6e507 

scala> val tpe = instanceMirror.symbol.toType 
tpe: reflect.runtime.universe.Type = Obj.type 

scala> getEtaExpandedMethodType(tpe.decl(TermName("method1")).asMethod) 
res28: reflect.runtime.universe.Type = (String, String) => scala.Int 

scala> getEtaExpandedMethodType(tpe.decl(TermName("method2")).asMethod) 
res29: reflect.runtime.universe.Type =() => String 

scala> getEtaExpandedMethodType(tpe.decl(TermName("method3")).asMethod) 
res30: reflect.runtime.universe.Type =() => scala.Long 

scala> getEtaExpandedMethodType(tpe.decl(TermName("method4")).asMethod) 
res31: reflect.runtime.universe.Type = String => (scala.Float => scala.Double) 

scala> getEtaExpandedMethodType(tpe.decl(TermName("method5")).asMethod) 
res32: reflect.runtime.universe.Type = T => scala.Int 

scala> getEtaExpandedMethodType(tpe.decl(TermName("method6")).asMethod) 
res33: reflect.runtime.universe.Type = T => scala.Int 

Answer 2

Здесь, вероятно, наиболее простое решение с использованием universe.appliedType. Он не работает в случае нескольких списков параметров. Я публикую это, чтобы показать альтернативный способ решения этой проблемы.

def getEtaExpandedMethodType2(methodSymbol: MethodSymbol): Type = { 
    val typesList = methodSymbol.info.paramLists(0).map(x => x.typeSignature) :+ methodSymbol.returnType 
    val arity = methodSymbol.paramLists(0).size 
    universe.appliedType(definitions.FunctionClass(arity), typesList) 
}