libhv/unittest/objectpool_test.cpp

#include <stdio.h>
#include <thread>

#include "hobjectpool.h"

#ifdef _WIN32
#include <windows.h>
#else
#include <unistd.h>
#endif

void msleep(unsigned int ms) {
#ifdef _WIN32
    Sleep(ms);
#else
    usleep(ms*1000);
#endif
}

class Task {
public:
    Task() {printf("Task()\n");}
    ~Task() {printf("~Task()\n");}

    void Do() {
        printf("%p start do...\n", this);
        msleep(4000);
        printf("%p end do\n", this);
    }
};

HObjectPool<Task> task_pool(1, 5);

void task_thread(int id) {
    printf("thread %d run...\n", id);
    HPoolObject<Task> pTask(task_pool);
    if (pTask) {
        pTask->Do();
    }
    else {
        printf("No available task in pool\n");
    }
    printf("thread %d exit\n", id);
}

int main(int argc, char** argv) {
    for (int i = 0; i < 10; ++i) {
        new std::thread(task_thread, i);
    }
    msleep(5000);
    for (int i = 10; i < 20; ++i) {
        new std::thread(task_thread, i);
    }
    msleep(10000);
    return 0;
}