使用 shapeless 更新 case class 字段

Question

我定义了以下类型：

sealed trait Refreshable {
  val expireAt: Instant
  val version: Int
}

case class A(id: Int, expireAt: Instant, version: Int) extends Refreshable

case class B(name: String, expireAt: Instant, version: Int) extends Refreshable

我想定义一个方法 refresh 来更新这些类型的两个字段，但我找不到我需要的隐含函数。看来我应该可以用 shapeless 做这样的事情？

def refresh[T <: RefreshableClass])(v: T)(implicit ev: LabelledGeneric[T]) = {
  val gen = ev.from(v)

  gen.updateWith('expire)(_.plus(...)).updateWith('version)(_+1)
}

Answer 1

~~封印你的特质~~

sealed trait Refreshable {
  val expireAt: Instant
  val version: Int
}

使用 Typelevel Scala 以下变体有效：

def refresh[T <: Refreshable, R <: HList](v: T)(implicit
  generic: LabelledGeneric.Aux[T, R],
  modifier: Modifier.Aux[R, Symbol @@ "version", Int, Int, R],
  modifier1: Modifier.Aux[R, Symbol @@ "expireAt", Instant, Instant, R]): R = {
  val rec = generic.to(v)
  val rec1 = modifier(rec, _ + 1)
  modifier1(rec1, _.plus(1L, ChronoUnit.MILLIS))
}

使用 Lightbend Scala（即标准 Scala）

def refresh[T <: Refreshable, R <: HList](v: T)(implicit
  generic: LabelledGeneric.Aux[T, R],
  modifier: Modifier.Aux[R, Witness.`'version`.T, Int, Int, R],
  modifier1: Modifier.Aux[R, Witness.`'expireAt`.T, Instant, Instant, R]): R = {
  val rec = generic.to(v)
  val rec1 = modifier(rec, _ + 1)
  modifier1(rec1, _.plus(1L, ChronoUnit.MILLIS))
}

如果您更喜欢使用 updateWith 这也是可能的：

def refresh[T <: Refreshable, R <: HList](v: T)(implicit
  generic: LabelledGeneric.Aux[T, R],
  modifier: Modifier.Aux[R, Witness.`'version`.T, Int, Int, R],
  modifier1: Modifier.Aux[R, Witness.`'expireAt`.T, Instant, Instant, R],
  selector: Selector.Aux[R, Witness.`'version`.T, Int],
  selector1: Selector.Aux[R, Witness.`'expireAt`.T, Instant]): R = {
  val rec = generic.to(v)
  rec.updateWith('expireAt)(_.plus(1L, ChronoUnit.MILLIS))
    .updateWith('version)(_ + 1)
}

通常你不需要指定类型参数T和R，它们应该被推断：

refresh(A(1, Instant.now, 1)) //1 :: 2017-09-20T18:53:34.039Z :: 2 :: HNil
refresh(B("a", Instant.now, 1)) // a :: 2017-09-20T18:53:34.074Z :: 2 :: HNil

但如果您确实需要，可以指定它们：

refresh[A, Record.`'id -> Int, 'expireAt -> Instant, 'version -> Int`.T](A(1, Instant.now, 1))
refresh[B, Record.`'name -> String, 'expireAt -> Instant, 'version -> Int`.T](B("a", Instant.now, 1))

Answer 2

可能最简单的方法是使用无形镜片。首先，一些导入：

import java.time._
import shapeless._

如果字段匹配，则不需要继承共同特征，因此出于本示例的目的，我将删除它：

case class A(id: Int, expireAt: Instant, version: Int)
case class B(name: String, expireAt: Instant, version: Int)

创建选择标记为 expireAt / version 的产品元素的路径

val (expirePath, versionPath) = (^.expireAt, ^.version)

编写接受给定路径的镜头工厂的方法：

def refresh[T](t: T)(implicit
                    expireL: expirePath.Lens[T, Instant],
                    versionL: versionPath.Lens[T, Int]): T = {
  val t1 = expireL().set(t)(Instant.now())
  val t2 = versionL().modify(t1)(_ + 1)

  t2
}

使用它：

println(refresh(A(1, null, 0)))
println(refresh(B("The B", null, 4)))

使用 shapeless 更新 case class 字段

update case class fields using shapeless

scala

shapeless