如何迭代斯卡拉

中定義的對象內的對象

我的問題是如何使用Scala的reflection package迭代在斯卡拉在定義的對象內的對象？如何迭代斯卡拉

object Units { 

    val values = CITIZEN :: WORKER :: Nil // I need something reflectional to list all of the case objects 

    case object CITIZEN extends Population 
    case object WORKER extends Population 
}

來源

2013-08-04 Pooya

有很多方法可以做到這一點，但我建議使用宏（編譯時反射）通過運行時反射，爲了兼顧性能和（更重要的）類型安全性。

這裏有一個快速的實現與宏：

import scala.language.experimental.macros 

object MacroUtils { 
    import scala.reflect.macros.Context 

    def values = macro MacroUtils.values_impl 
    def values_impl(c: Context) = { 
    import c.universe._ 

    val objs = c.enclosingClass.collect { 
     case ModuleDef(mods, name, _) if mods hasFlag Flag.CASE => Ident(name) 
    } 

    c.Expr[List[Any]](
     Apply(Select(reify(List).tree, newTermName("apply")), objs.toList) 
    ) 
    } 
} 

trait Population 

object Units { 
    val values = MacroUtils.values 
    case object CITIZEN extends Population 
    case object WORKER extends Population 
}

然後，例如：

scala> val populations: List[Population] = Units.values 
populations: List[Population] = List(CITIZEN, WORKER)

注意，編譯器知道這種情況下對象的列表可以靜態類型的列表人口。

來源

2013-08-04 14:36:58

這裏有一個問題，宏實現必須在'高清值靜態訪問對象=宏MacroUtils.values_impl' – Pooya

你移動'values'方法的定義是什麼？它對我來說完全適用於上面的粘貼。 –

是的，我曾在單位搞錯對象 – Pooya

我不知道這是否是做的最好的方法，但在這裏它是：

object Units { 

    val values = CITIZEN :: WORKER :: Nil 

    trait Population 

    case object CITIZEN extends Population 

    case object WORKER extends Population 

    val reflectionValues = { 
     import scala.reflect.runtime.{universe => ru} 
     val mirror = ru.runtimeMirror(getClass.getClassLoader) 
     val objects = ru.typeOf[Units.type].declarations.filter(_.isModule) 
     objects.map(o => mirror.reflectModule(o.asModule).instance.asInstanceOf[Population]).toList 
    } 

    def main(args: Array[String]) { 
     assert(values == reflectionValues) 
    } 
}

來源

2013-08-04 14:25:05

如何迭代斯卡拉

回答

相關問題