#include <stdio.h>
#include <unistd.h>
#include "tiny_pool.h"

void demo_fun(void *i) {
    int num = *(int*)i;
    printf("  task %d start\n", num);
    for (int t = 0; t < num; ++t) {
        sleep(1);
        printf("  task %d running...\n", num);
    }
    printf("  task %d complete\n", num);
}

int main() {

//    pool_t *pool = tiny_pool_create(2);
    pool_t *pool = tiny_pool_create(4);

    int dat[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};
    tiny_pool_submit(pool, demo_fun, (void*)&dat[0]);
    tiny_pool_submit(pool, demo_fun, (void*)&dat[1]);

    printf("+ main: pool booting\n");
    tiny_pool_boot(pool);
    printf("+ main: pool boot complete\n");

    printf("+ main: sleep 5s\n");
    sleep(5);
    printf("+ main: wake up\n");

    tiny_pool_submit(pool, demo_fun, (void*)&dat[2]);
    tiny_pool_submit(pool, demo_fun, (void*)&dat[3]);
    tiny_pool_submit(pool, demo_fun, (void*)&dat[4]);
    tiny_pool_submit(pool, demo_fun, (void*)&dat[5]);
    tiny_pool_submit(pool, demo_fun, (void*)&dat[6]);

    printf("+ main: sleep 8s\n");
    sleep(6);
    printf("+ main: wake up\n");

    tiny_pool_submit(pool, demo_fun, (void*)&dat[7]);
    tiny_pool_submit(pool, demo_fun, (void*)&dat[8]);

    printf("+ main: pool joining\n");
    tiny_pool_join(pool);
    printf("+ main: pool join complete\n");

//    printf("pool try exit\n");
//    tiny_pool_kill(pool);

    // TODO: tiny pool destroy

//    sleep(10);

    printf("+ main exit\n");

    return 0;
}