summaryrefslogtreecommitdiffhomepage
path: root/misc/examples/coroutines/scheduler.c
blob: be1810d22ff6bc81f25181472483c0ffb60d76a6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
// https://www.youtube.com/watch?v=8sEe-4tig_A
#include <stdio.h>
#include <stc/coroutine.h>

#define i_type cco_tasks
#define i_key cco_task*
#define i_keydrop(x) { puts("free task"); free(*x); }
#define i_no_clone
#include <stc/cqueue.h>

typedef struct {
    cco_tasks tasks;
} cco_scheduler;

void cco_scheduler_drop(cco_scheduler* sched) {
    cco_tasks_drop(&sched->tasks);
}

int cco_scheduler_run(cco_scheduler* sched) {
    while (!cco_tasks_empty(&sched->tasks)) {
        cco_task* task = cco_tasks_pull(&sched->tasks);
        if (cco_resume_task(task, NULL))
            cco_tasks_push(&sched->tasks, task);
        else
            cco_tasks_value_drop(&task);
    }
    return 0;
}

static int taskA(cco_task* task, cco_runtime* rt) {
    cco_routine(task) {
        puts("Hello, from task A");
        cco_yield();
        puts("A is back doing work");
        cco_yield();
        puts("A is back doing more work");
        cco_yield();
        puts("A is back doing even more work");
    }
    return 0;
}

static int taskB(cco_task* task, cco_runtime* rt) {
    cco_routine(task) {
        puts("Hello, from task B");
        cco_yield();
        puts("B is back doing work");
        cco_yield();
        puts("B is back doing more work");
    }
    return 0;
}

void Use(void) {
    cco_scheduler sched = {.tasks = c_init(cco_tasks, {
        c_new(cco_task, {.cco_func=taskA}), 
        c_new(cco_task, {.cco_func=taskB}),
    })};

    cco_scheduler_run(&sched);
    cco_scheduler_drop(&sched);
}

int main(void)
{
    Use();
}