使用镜头更改数据类的多个属性

时间:2019-09-25 18:25:39

标签: kotlin functional-programming arrow-kt

我正在Kotlin中试验Lenses,我想知道是否存在一种优雅的方法来同时更改一个对象的多个属性。假设我的域如下所示:

@optics
data class Parameters(
    val duration: Int,
    val length: Int) {
  companion object
}

@optics
data class Calculation(
    val product: String
    val parameters: Parameters) {
  companion object
}

借助@optics批注,易于编辑单个字段:

val calculation = Calculation(product = "prod", Parameters(duration = 10, length = 15))

Calculation.product.modify(calculation) { selectedProduct }
Calculation.parameters.duration(calculation) { newDuration() }
Calculation.parameters.length(calculation) { 10 }

这些镜头在隔离状态下可以完美工作,但是当我想一次应用三个变换时,应该使用什么正确的图案?我可以使用var并每次都覆盖calculation,但这对我来说不是很习惯。

1 个答案:

答案 0 :(得分:3)

Arrow当前不提供此类功能,但您可以自己编写通用解决方案。

下面的代码段演示了如何实现此目标,您可以添加其他方法以从Lens<S, Tuple2<FA, FB>>Lens<S, Tuple3<FA, FB, FC>>等组成

@optics data class Char(val name: String, val health: Int) {
  companion object
}

infix fun <S, FA, FB> Lens<S, FA>.aside(other: Lens<S, FB>): Lens<S, Tuple2<FA, FB>> = object : Lens<S, Tuple2<FA, FB>> {
  override fun get(s: S): Tuple2<FA, FB> = Tuple2(this@aside.get(s), other.get(s))
  override fun set(s: S, b: Tuple2<FA, FB>): S = other.set(this@aside.set(s, b.a), b.b)
}

fun main() {
  val original = Char("", 0)
  val charName: Lens<Char, String> = Char.name
  val charHealth: Lens<Char, Int> = Char.health
  val charNameAndHealth: Lens<Char, Tuple2<String, Int>> = charName.aside(charHealth)

  charNameAndHealth.modify(original) { Tuple2("Test", 30) }
}