如何使用shapeless在编译时将case类字段的名称作为字符串/符号获取?

时间:2016-03-15 09:07:49

标签: scala shapeless

我想以某种方式在编译时获取val中case类的字段名称(可能是单例类型的字符串或符号?)。

如下所示:

import shapeless._
case class MyClass(field1: String, field2: Int)
val field1Lens = lens[MyClass] >> 'field1 
// val name = field1Lens.name // it should be "field1", aka 'field1.name

我没有必要使用镜头,任何有效的技术都很好(LabelledGeneric的东西?)。我想有一些东西,我可以获得案例类字段的名称,而无需单独指定它。这样,如果我重构了类中field1成员的名称,name会相应地更改。

当然以下方法不起作用,因为宏在编译时不知道符号的名称:

val name = 'field1
val field1Lens = lens[MyClass] >> name // can't possibly work

我尝试了lens[MyClass] >> name.narrow,但它无效

这就是我目前正在做的事情,当然我不喜欢它:

// If I change the name of the field, compilation fails 
// and I'm forced to check this line of code so I can change the string in the line below
protected val fieldNameCheck = lens[X].someField
val someField = "someField"

编辑:好的,我查看了gabriele的问题,并使用Keys我能够获得包含记录(标记)键的HList。 我需要的是获得一个特定字段,而不是包含所有字段的列表。

我正在尝试使用select获取特定密钥,但到目前为止我还没有成功

import shapeless._
import shapeless.syntax.singleton._
import shapeless.ops.record._

case class Foo(bar: String, baz: Boolean)
val labl = LabelledGeneric[Foo]
val keys = Keys[labl.Repr].apply

// the following doesn't work
// val bar = 'bar.narrow
// keys.select[bar.type]
// keys.get('bar)

2 个答案:

答案 0 :(得分:6)

>>的参数是Witness,它将成员名称捕获为编译时符号。当您编写>> 'bar时,符号文字会隐式转换为Witness,这通常是您想要的,但您也可以自己提供:

scala> case class Foo(bar: String, baz: Boolean)
defined class Foo

scala> val barKey = shapeless.Witness('bar)
barKey: shapeless.Witness.Aux[shapeless.tag.@@[Symbol,String("bar")]] = ...

scala> shapeless.lens[Foo] >> barKey
res0: shapeless.Lens[Foo,String] = shapeless.Lens$$anon$7@344bfb60

正如我在上面的评论中提到的,您可能也对位置选择器感兴趣:

scala> shapeless.lens[Foo] >> shapeless.nat._1
res1: shapeless.Lens[Foo,Boolean] = shapeless.Lens$$anon$7@16e9d434

甚至只是:

scala> shapeless.lens[Foo] >> 1
res2: shapeless.Lens[Foo,Boolean] = shapeless.Lens$$anon$7@4be29007

这些不要求您在代码中的任何位置写入会员名称,但如果您重新安排会员,则会遇到麻烦。

答案 1 :(得分:3)

好的,多亏了特拉维斯的评论,我得到了它的工作:

import shapeless._
case class MyClass(field1: String, field2: Int)

def fieldName[A](fieldKey: Witness.Lt[_ <: Symbol])(implicit mkl: MkFieldLens[A, fieldKey.T]) = {
  lens[A] >> fieldKey
  fieldKey.value.name
}

println(fieldName[MyClass]('field1))