void TestMain() {
    co[]() {

        defer[]() {
            info() << "defer1";
        };
        defer[]() {
            info() << "defer2";
        };
        defer[]() {
            info() << "defer3";
            auto e = recovery();
            if (e.has_value()) {
                info() << "recovery this error:" << e.value().what();
            }
        };

        info() << "throw panic";

        panic("panic");

        info() << "safe return";
    };
}


int main() {
    lane::IOManager iom(2, "hello", false);
    iom.addTask(TestMain);
}

void TestWaitGroup() {
    info() << "start runing";
    auto wait = new lane::WaitGroup;

    for (int i = 0; i < 5; ++i) {
        wait->add(1);
        co[=]() {
            defer[&]() {
                wait->done();
            };
            sleep(1);
            info() << " fast finish";
        };
    }
    wait->add(1);
    co[=]() {
        defer[&]() {
            wait->done();
        };
        sleep(2);
        info() << " slow finish";
    };

    wait->wait();
    info() << "end";
}

void TestMain() {
    lane::IOManager iom(2, "waitgroup", false);
    iom.addTask(TestWaitGroup);
}

void TestPrint() {
    auto wg = new lane::WaitGroup;

    // 创建两个 无缓冲channel，分别用于控制字母和数字的打印
    auto letters = lane::Channel<bool>(0);
    auto numbers = lane::Channel<bool>(0);

    // 字母打印协程
    wg->add(1);
    co[&]() {
        defer[&]() {
            wg->done();
        };
        for (char ch = 'A'; ch <= 'C'; ch++) {
            info() << ch;  // 打印字母
            letters.input(true);  // 发送信号，通知数字打印协程可以打印
            numbers.output();  // 等待数字打印协程的信号
        }
        letters.close();  // 关闭字母 channel
    };


    // 数字打印协程
    wg->add(1);
    co[&]() {
        defer[&]() {
            wg->done();
        };
        for (int num = 1; num <= 3; num++) {
            letters.output();  // 等待字母打印协程的信号
            info() << num;     // 打印数字
            numbers.input(true);
        }
        numbers.close();  // 关闭数字 channel
    };

    wg->wait();  // 等待所有协程完成
}


int main() {
    lane::IOManager iom(2, "channel", false);
    iom.addTask(TestPrint);
    return 0;
}