如何将函数类型或Lambda用作Kotlin上下文接收器的类型？

你遇到的编译错误根源是对Kotlin上下文接收器的核心逻辑理解偏差：上下文接收器要求当前作用域中必须存在对应类型的实例，但你用context(Foo)指定的Foo是() -> Unit函数类型，在bar的Lambda执行时，这个类型的实例并没有被引入到作用域中，所以编译器找不到所需的上下文。

要实现「仅能在特定Lambda上下文中调用的函数」，正确的做法是定义专用的上下文标记类型，而非直接用函数类型作为上下文。以下是两种可行的实现方案：

方案一：专用标记上下文类 #

定义一个专门的类/接口作为上下文标识，通过带接收器的Lambda将上下文注入到目标作用域：

package yamin

// 定义专用的上下文标记类，用于限定函数调用范围
class FooContext

fun main() {
    bar {
        baz() // 现在可以正常调用，因为当前处于FooContext上下文中
    }
}

// bar函数接收带FooContext接收器的Lambda，确保Lambda执行时上下文存在
fun bar(block: FooContext.() -> Unit) {
    FooContext().block() // 创建上下文实例，让Lambda在该上下文中执行
}

// 限定baz只能在FooContext上下文中调用
context(FooContext)
fun baz() {
    // 业务逻辑
}

方案二：带行为的上下文类 #

如果需要上下文携带特定行为，可以在上下文类中添加属性或方法，同时保留调用范围的限制：

package yamin

// 上下文类携带需要的行为
class FooContext(val onExecute: () -> Unit)

fun main() {
    bar {
        baz()
        onExecute() // 可以直接调用上下文提供的行为
    }
}

fun bar(block: FooContext.() -> Unit) {
    // 初始化上下文并传入行为逻辑
    FooContext({ println("执行上下文行为") }).block()
}

context(FooContext)
fun baz() {
    // 仅能在FooContext上下文中调用
}

核心逻辑总结 #

上下文接收器的本质是「作用域中存在对应类型的实例」，而非用函数类型直接标记上下文。你之前的写法错误地将函数类型当作上下文标识，但该类型的实例从未在Lambda作用域中被提供，导致编译器无法找到所需上下文。

内容的提问来源于stack exchange，提问作者Yamin Siahmargooei