如何在Julia中编写类似Python含yield关键字的简单函数

你提供的Python演示代码如下：

def simpleFun():
    yield 1 
    yield 2 
    yield 3 
x=simpleFun() 
print(x.next()) 
print(x.next()) 
print(x.next()) 

对应的输出为：

1
2
3

由于Julia中的yield关键字语义和Python完全不同——Julia的yield用于任务调度，作用是让当前运行的任务主动暂停，让出CPU给其他可运行任务，而非生成迭代值。要实现和Python生成器等价的功能，有两种常用实现方式：

方式一：使用Channel（最贴近Python生成器的行为） #

Channel是Julia中实现生产者-消费者模式的核心工具，可以完美模拟Python生成器逐个返回值的逻辑：

function simpleFun()
    return Channel() do ch
        put!(ch, 1)
        put!(ch, 2)
        put!(ch, 3)
    end
end

x = simpleFun()
println(take!(x))
println(take!(x))
println(take!(x))

运行这段代码会输出：

1
2
3

方式二：自定义迭代器类型 #

如果需要更底层的迭代控制，可以手动实现迭代器接口：

struct SimpleGenerator
    state::Int
end

# 初始化迭代器实例
SimpleGenerator() = SimpleGenerator(0)

# 实现迭代起始逻辑
function Base.iterate(gen::SimpleGenerator)
    gen.state >= 3 && return nothing
    next_val = gen.state + 1
    return (next_val, SimpleGenerator(next_val))
end

# 实现迭代状态递进逻辑
function Base.iterate(gen::SimpleGenerator, state::SimpleGenerator)
    state.state >= 3 && return nothing
    next_val = state.state + 1
    return (next_val, SimpleGenerator(next_val))
end

# 使用迭代器
x = SimpleGenerator()
for val in x
    println(val)
end

运行后同样会输出1、2、3。

再明确下Julia的yield用法 #

举个简单例子直观展示Julia中yield的任务调度作用：

function task_func()
    for i in 1:3
        println("任务输出：$i")
        yield()  # 主动让出CPU
    end
end

# 创建并调度任务
t = @task task_func()
schedule(t)
yield()  # 主任务让出CPU给t
yield()
yield()

这段代码会依次输出任务中的内容，每次yield都会触发任务间的执行权切换。

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