Использование отражения Scala для проверки метода объекта или для определения типа ключа на карте

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

Ядром этого является метод lookUp(key: String, obj: AnyRef), который получает один ключ для поиска и один объект для поиска ключа. Если объект является картой, а ключи карты являются символом или строкой, то он будет искать на карте запрошенный ключ. В противном случае он проверит, есть ли у объекта метод с тем же именем ключа, и если это так, он вызовет этот метод.

В существующем коде Ruby это легко сделать:

def look_up(key, obj)
  if obj.respond_to?(:has_key?) && obj.has_key?(key)
    obj[key]
  elsif obj.respond_to?(key)
    obj.send(key)
  elsif obj.instance_variable_defined?(ivar = "@#{key}")
    obj.instance_variable_get(ivar)
  end
end

Поскольку это легко сделать, код Ruby также ищет переменную экземпляра с тем же именем. Мне не обязательно нужна эта функциональность в моей версии Scala.

Одна из проблем, с которыми я сталкиваюсь, заключается в том, что примеры, которые я нашел, требуют, чтобы я знал класс объекта заранее, чтобы я мог вызвать ru.typeOf[MyClass].declaration(ru.TermName("key")) (где ru является scala.reflect.runtime.universe).

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

ОБНОВЛЕНИЕ: я думал о чем-то вроде этого, но это перебор? Или необходимо правильно захватить типы на карте? Кроме того, это не компилируется. Map, Symbol а также String не правильные типы для их контекста.

def lookUp[T](obj: T, key: String)(implicit tag: ru.TypeTag[T]): Option[String] = tag.tpe match {
  case ru.TypeRef(a, Map, List(Symbol, _)) => if (obj.contains(Symbol(key))) Some(obj(Symbol(key)).toString) else None
  case ru.TypeRef(a, Map, List(String, _)) => if (obj.contains(key)) Some(obj(key).toString) else None
  case _ =>
    if (/* obj.key() exists */)
      // Some(obj.key().toString)
    else
      None
}

ОБНОВЛЕНИЕ 2: мне никогда не приходило в голову, что я мог бы использовать asInstanceOf с чем-то вроде Map[String, _], Я использовал второй пример кода @johny, чтобы найти свое решение. Я кеширую имена методов по классам в mutable.HashMap[Class[_], Set[String]],

def lookUp(obj: AnyRef, key: String): Option[String] = obj match {
  case m: Map[_, _] =>
    if (m.asInstanceOf[Map[String, _]].contains(key))
      extractValue(m.asInstanceOf[Map[String, _]](key))
    else if (m.asInstanceOf[Map[Symbol, _]].contains(Symbol(key)))
      extractValue(m.asInstanceOf[Map[Symbol, _]](Symbol(key)))
    else
      None
  case _ =>
    val klass = obj.getClass
    if (!methodsCache.contains(klass))
      methodsCache(klass) = klass.getMethods.toList.filter(_.getParameterCount == 0).map(_.getName).toSet
    val methodNames = methodsCache(klass)
    if (methodsCache(klass).contains(key))
      extractValue(klass.getDeclaredMethod(key).invoke(obj))
    else
      None
}

def extractValue(obj: Any): Option[String] = obj match {
  case null | None => None
  case Some(x) => Some(x.toString)
  case x => Some(x.toString)
}

1 ответ

Решение
 def lookUp(key: String, obj: AnyRef) {
    obj match {
      case x: Map[String, _] => x(key)
      case _                 => obj.getClass.getDeclaredMethod(key).invoke(obj)
    }
  }

Изменить: чтобы убедиться, что ключ карты либо String или scala.Symbol

def lookUp(key: String, obj: AnyRef)= {
    obj match {
      case x: Map[_, _]  => 
        if(x.asInstanceOf[Map[String,_]].contains(key))
          x.asInstanceOf[Map[String,_]].get(key)
        else if(x.asInstanceOf[Map[scala.Symbol,_]].contains(scala.Symbol(key)))
          x.asInstanceOf[Map[scala.Symbol,_]].get(scala.Symbol(key))
        else
          None

      case _ => Some(obj.getClass.getDeclaredMethod(key).invoke(obj))
    }
  }

Вышеприведенное также не гарантирует, что возвращаемый результат получен из Map,

Редактировать: Следуя примеру @JimCain

def lookUp[T:ru.TypeTag](obj: T, key: String): Option[Any] = ru.typeTag[T].tpe match {
    case ru.TypeRef(a, m, l) if(m.name.toString=="Map"&&l.head =:= ru.typeOf[java.lang.String])=> obj.asInstanceOf[Map[String,_]].get(key)
    case ru.TypeRef(a, m,l) if(m.name.toString=="Map"&&l.head =:= ru.typeOf[Symbol])=> obj.asInstanceOf[Map[Symbol,_]].get(scala.Symbol(key))
    case _ => Try(obj.getClass.getDeclaredMethod(key).invoke(obj)) match {
      case Success(x) => Some(x)
      case Failure(_) => None
    }
  }
Другие вопросы по тегам