创建一个返回每个内部对象的 lambda（如 forEach）

Question

如何创建自己的类似于 forEach (kotlin-stdlib) 的 lambda 表达式？

场景：

对于列表的每个对象，我将其包装在另一个对象中并调用如下包装器方法：

class Action {
  fun runA() {
    objects.forEach { obj ->
      val wrapper = Wrapper(obj)
      wrapper.runA()
    }
  }
}

问题：

我想在不重复代码的情况下创建runB。 如何在 Kotlin 中优雅地做到这一点？

我无法传递像下面这样的参数方法，因为我要调用的方法属于我尚未创建的实例。

class Action {

  fun runMethod(method: () -> Unit) {
    objects.forEach { obj ->
      val wrapper = Wrapper(obj)
      method() // I want: wrapper.method(), but wrapper is created in the forEach lambda
    }
  }

  fun runA() { ... }
  fun runB() { ... }
}

理想情况下，我想要这样的东西：

class Action {
// ...
fun runA() {
  runMethod { wrapper ->
    wrapper->runA()
  }
}

fun runB() {
  runMethod { wrapper ->
    wrapper->runB()
  }
}
}

Answer 1

如果我正确理解您的问题，您可以将接收器添加到传递给runMethod() 函数的 lambda。在传递的 lambda 中，Wrapper 变成 this：

fun main() {
    runMethod { runA() }
    runMethod { runB() }
    
    // or:
    
    runMethod(Wrapper::runA)
    runMethod(Wrapper::runB)
}

fun runMethod(method: Wrapper.() -> Unit) {
    objects.forEach { obj ->
        val wrapper = Wrapper(obj)
        wrapper.method()
    }
}