使用shapeless从案例类中提取数据,修改它,然后重新创建案例类

时间:2016-07-31 14:29:27

标签: scala shapeless

我有一个包含一些案例类的程序。其中一个可能是一个Person,它依赖于其他一些类,包括一个名为StemmedString的类,这在这里很重要:

case class Person(name: String, skills: List[Skill], hobbbies: List[StemmedString]
case class Skill(score: Int, title: StemmedString)
case class StemmedString(original: String, stemmed: String)

现在我希望能够将此人翻译成另一种语言。要做到这一点,我想从Person - >列表[字符串]。这是一个需要翻译的单个字符串列表。然后我想从List [String] - >返回人。

我希望能够定义所有应该翻译的东西的类型,而不需要事先知道Person的格式,这样就可以推广到由同一个TYPES组成的多个case类。被翻译。

让我们说一下我们翻译的是StemmedString的所有内容。

我可以使用LabelledGeneric和flatMap来创建两个HList,一个要翻译的值,另一个不会被翻译的值:

trait LowPriorityUnTranslatable extends Poly1 {
  implicit def default[T] = at[T](_ :: HNil)
}

object unTranslatable extends LowPriorityUnTranslatable {
  implicit def caseStemmedString[K, T] = at[FieldType[K, StemmedString]](x => HNil)
  implicit def caseSkill[K, T] = at[FieldType[K, Skill](x => HNil)
}


trait LowPriorityTranslatable extends Poly1 {
  implicit def default[T] = at[T](HNil)
}

object Translatable extends LowPriorityTranslatable {
  implicit def caseStemmedString[K, T] = at[FieldType[K, StemmedString]](_ :: HNil)
  implicit def caseSkill[K, T] = at[FieldType[K, Skill](_ :: HNil)
}

这感觉有点冗长,但效果很好。我现在有两个HList,它们可以很容易地连接起来并使用align返回原来的case类:

val person = Person("...")
val gen = LabelledGeneric[Person] 
val personList = gen.to(person)
val toTranslate = personList flatMap isTranslatable
val notTranslated = personList flatMap unTranslatable    
gen.from((toTranslate ++ notTranslated) align personList)

这很酷,现在我需要做的就是在中间添加一个翻译步骤。它很容易从isTranslatable - >列出[String]当然,但是我无法弄清楚如何以我可以再次返回的方式执行此操作。我开始尝试学习无形,因为它似乎是适合这种情况的正确工具,但我不太了解如何完全使用它。在我的脑海中,如果我can solve this question,那么我会很好,但可能有一种更简单的方法来使用无形来解决这个问题。任何见解都会非常感激!

1 个答案:

答案 0 :(得分:1)

如果您不介意使用可变数据结构(在本例中为Iterator),则可以使用everything / everywhere进行简单的解决方案。

首先,我们使用everything查询提取要翻译的字符串;在这个例子中,我们提取StemmedString个对象的原始值。使用Append函数连接单例列表。

trait TranslatableLP extends Poly1 {
  implicit def default[T] = at[T](_ => Nil: List[String])
}

object Translatable extends TranslatableLP {
  implicit def caseStemmedString = at[StemmedString](s => List(s.original))
}

object Append extends Poly2 {
  implicit val caseString = at[List[String], List[String]](_ ++ _)
}

val strings = everything(Translatable)(Append)(person)

现在我们翻译字符串:

def translate(s: String): String = ???

val translatedStrings = strings.map(translate)

最后,我们可以使用StemmedString使用翻译的字符串映射everywhere个对象:

object Update extends Poly1 {
  def iterator = translatedStrings.iterator
  implicit def caseStemmedString = at[StemmedString](_.copy(original = iterator.next))
}

val translated = everywhere(Update)(person)

当我找到一些时间时,我会尝试仅使用不可变数据结构来提出更清晰的解决方案。