Question

我有一组模型类，这些模型的子集有2个属性，分别为createdBy和modifiedBy。
我只需要为具有那些属性的那些对象填充那些属性。目前，我正在以与某些样板代码匹配的模式进行操作。

  case class DataSourceInstanceRow(id: Int, value: String, createdBy: Option[String], modifiedBy: Option[String])
  case class FormDefinitionRow(id: Int, formData: String, createdBy: Option[String], modifiedBy: Option[String])
  case class DecisionTableDefinitionRow(id: Int, rows: Int, definitions: List[String], createdBy: Option[String], modifiedBy: Option[String])
  case class ReportDef(id: Int, reportType: Int, reportName: String)


  def populateLogs[T](t: T, user: String): T = {
    t match {
      case ds: DataSourceInstanceRow =>
        if(ds.id == -1) ds.copy(modifiedBy = Some(user), createdBy = Some(user)).asInstanceOf[T]
        else ds.copy(modifiedBy = Some(user)).asInstanceOf[T]
      case fd: FormDefinitionRow =>
        if(fd.id == -1) fd.copy(modifiedBy = Some(user), createdBy = Some(user)).asInstanceOf[T]
        else fd.copy(modifiedBy = Some(user)).asInstanceOf[T]
      case dtd: DecisionTableDefinitionRow =>
        if(dtd.id == -1) dtd.copy(modifiedBy = Some(user), createdBy = Some(user)).asInstanceOf[T]
        else dtd.copy(modifiedBy = Some(user)).asInstanceOf[T]
      case o => o
    }
  }

DataSourceInstanceRow，FormDefinitionRow，DecisiontableDefinitionRow具有modifiedBy和createdBy属性。但不是ReportDef

如何使用无形来创建抽象以从上述模式匹配中删除样板？

Answer 1

您可以使用Shapeless的Updater做这种事情：

import shapeless.{ LabelledGeneric, HList, Witness }
import shapeless.labelled.{FieldType, field}
import shapeless.ops.record.Updater

type CreatedBy = Witness.`'createdBy`.T
type ModifiedBy = Witness.`'modifiedBy`.T

def populateLogs[T, R <: HList](t: T, user: String)(implicit
  gen: LabelledGeneric.Aux[T, R],
  cb: Updater.Aux[R, FieldType[CreatedBy, Option[String]], R] = null,
  mb: Updater.Aux[R, FieldType[ModifiedBy, Option[String]], R] = null
): T = (
  for {
    createdBy <- Option(cb)
    modifiedBy <- Option(mb)
  } yield gen.from(
    createdBy(modifiedBy(gen.to(t), field(Some(user))), field(Some(user)))
  )
).getOrElse(t)

然后：

scala> populateLogs(DataSourceInstanceRow(1, "abc", None, None), "foo")
res0: DataSourceInstanceRow = DataSourceInstanceRow(1,abc,Some(foo),Some(foo))

scala> populateLogs(ReportDef(1, 2, "abc"), "foo")
res1: ReportDef = ReportDef(1,2,abc)

此实现基于您可以在隐式参数上放置null默认值的事实，因此，如果编译器找不到隐式参数，则编译器将使用该默认值。它很简单并且可以正常工作，但是有些人讨厌它。一种更原则的方法使用隐式优先级：

trait UpdateBoth[T] extends ((T, String) => T)

object UpdateBoth extends LowPriorityUpdateBothInstances {
  implicit def updateWithFields[T, R <: HList](implicit
    gen: LabelledGeneric.Aux[T, R],
    cb: Updater.Aux[R, FieldType[CreatedBy, Option[String]], R],
    mb: Updater.Aux[R, FieldType[ModifiedBy, Option[String]], R]
  ): UpdateBoth[T] = (t, user) =>
    gen.from(cb(mb(gen.to(t), field(Some(user))), field(Some(user))))
}

trait LowPriorityUpdateBothInstances {
  implicit def updateAny[T]: UpdateBoth[T] = (t, _) => t
}

def populateLogs[T](t: T, user: String)(implicit update: UpdateBoth[T]): T =
  update(t, user)

这将以完全相同的方式工作。

动态填充一组案例类的某些属性

1 个答案: