用 Java 撸了一个 yield 的玩具

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

现在注册

已注册用户请登录

这是一个创建于 1905 天前的主题，其中的信息可能已经有所发展或是发生改变。

效果如下

public class GeneratorTest {  
    private int a = 0;  
    private int b = 1;  
    @Test
    public void test2() {  
        Generator<Integer> generator = new Generator<>();  
        generator.loop()  
                .and(() -> a += b)  
                .and(y -> y.yield(b))  
                .and(() -> b += a)  
                .until(y -> y.yieldIf(b > 1000).yieldBreak().yieldElse().yield(a));  
        for (Integer result : generator) {
            System.out.print("result = " + result);
            System.out.print("   a = " + a);
            System.out.println("   b = " + b);
         
        }
    }
}

运行结果

 result = 1   a = 1   b = 1
 result = 1   a = 1   b = 2
 result = 2   a = 3   b = 2
 result = 3   a = 3   b = 5
 result = 5   a = 8   b = 5
 result = 8   a = 8   b = 13
 result = 13   a = 21   b = 13
 result = 21   a = 21   b = 34
 result = 34   a = 55   b = 34
 result = 55   a = 55   b = 89
 result = 89   a = 144   b = 89
 result = 144   a = 144   b = 233
 result = 233   a = 377   b = 233
 result = 377   a = 377   b = 610

之前把另一个项目某模块从 bio 改成了 nio,尝到了协程的甜头,宽带可以跑满,基本不占 cpu.于是打算把剩余的模块改完.结果发现连接状态一多,handler 就会越来越长,如果还有条件和循环,那就几乎没法写了.于是我就想 java 如果有 yield 就好了.
目前实现 yield 的方法有三种,第一种是改字节码,但是比想象中麻烦,第二种是返回一个阻塞队列,yield()是阻塞方法,将结果用其他线程返回,但是涉及到多线程,而原来的目的就是像把项目改成协程的.所以就用了第三种,把需要 yield 的代码块当成 lambda,传入链表中依次调用,如果调用了参数的 yield 的方法就返回一次结果;
做完以后感觉自己走了弯路.java 有没有类似的包或者工具?用 stream 能不能达到类似效果?

16 条回复 • 2019-09-07 22:03:12 +08:00