Blame docs/code/queue-cancel/main.c

Packit Service 7c31a4
#include <stdio.h>
Packit Service 7c31a4
#include <stdlib.h>
Packit Service 7c31a4
#include <unistd.h>
Packit Service 7c31a4
Packit Service 7c31a4
#include <uv.h>
Packit Service 7c31a4
Packit Service 7c31a4
#define FIB_UNTIL 25
Packit Service 7c31a4
uv_loop_t *loop;
Packit Service 7c31a4
uv_work_t fib_reqs[FIB_UNTIL];
Packit Service 7c31a4
Packit Service 7c31a4
long fib_(long t) {
Packit Service 7c31a4
    if (t == 0 || t == 1)
Packit Service 7c31a4
        return 1;
Packit Service 7c31a4
    else
Packit Service 7c31a4
        return fib_(t-1) + fib_(t-2);
Packit Service 7c31a4
}
Packit Service 7c31a4
Packit Service 7c31a4
void fib(uv_work_t *req) {
Packit Service 7c31a4
    int n = *(int *) req->data;
Packit Service 7c31a4
    if (random() % 2)
Packit Service 7c31a4
        sleep(1);
Packit Service 7c31a4
    else
Packit Service 7c31a4
        sleep(3);
Packit Service 7c31a4
    long fib = fib_(n);
Packit Service 7c31a4
    fprintf(stderr, "%dth fibonacci is %lu\n", n, fib);
Packit Service 7c31a4
}
Packit Service 7c31a4
Packit Service 7c31a4
void after_fib(uv_work_t *req, int status) {
Packit Service 7c31a4
    if (status == UV_ECANCELED)
Packit Service 7c31a4
        fprintf(stderr, "Calculation of %d cancelled.\n", *(int *) req->data);
Packit Service 7c31a4
}
Packit Service 7c31a4
Packit Service 7c31a4
void signal_handler(uv_signal_t *req, int signum)
Packit Service 7c31a4
{
Packit Service 7c31a4
    printf("Signal received!\n");
Packit Service 7c31a4
    int i;
Packit Service 7c31a4
    for (i = 0; i < FIB_UNTIL; i++) {
Packit Service 7c31a4
        uv_cancel((uv_req_t*) &fib_reqs[i]);
Packit Service 7c31a4
    }
Packit Service 7c31a4
    uv_signal_stop(req);
Packit Service 7c31a4
}
Packit Service 7c31a4
Packit Service 7c31a4
int main() {
Packit Service 7c31a4
    loop = uv_default_loop();
Packit Service 7c31a4
Packit Service 7c31a4
    int data[FIB_UNTIL];
Packit Service 7c31a4
    int i;
Packit Service 7c31a4
    for (i = 0; i < FIB_UNTIL; i++) {
Packit Service 7c31a4
        data[i] = i;
Packit Service 7c31a4
        fib_reqs[i].data = (void *) &data[i];
Packit Service 7c31a4
        uv_queue_work(loop, &fib_reqs[i], fib, after_fib);
Packit Service 7c31a4
    }
Packit Service 7c31a4
Packit Service 7c31a4
    uv_signal_t sig;
Packit Service 7c31a4
    uv_signal_init(loop, &sig);
Packit Service 7c31a4
    uv_signal_start(&sig, signal_handler, SIGINT);
Packit Service 7c31a4
Packit Service 7c31a4
    return uv_run(loop, UV_RUN_DEFAULT);
Packit Service 7c31a4
}