libhv/hthread.h

#ifndef H_THREAD_H
#define H_THREAD_H

#include "hdef.h"
#include "hplatform.h"
#include <thread>
#include <atomic>

#ifdef _MSC_VER
inline uint32 getpid(){
    return GetCurrentProcessId();
}
#endif

inline uint32 gettid(){
#ifdef _MSC_VER
    return GetCurrentThreadId();
#else
    return pthread_self();
#endif
}

/************************************************
 * HThread
 * Status: STOP,RUNNING,PAUSE
 * Control: start,stop,pause,resume
 * first-level virtual: doTask
 * second-level virtual: run
************************************************/
class HThread{
public:
    HThread() {
        status = STOP;
    }

    virtual ~HThread() {

    }

    virtual int start() {
        if (status == STOP) {
            status = RUNNING;
            dotask_cnt = 0;

            thread = std::thread([this]{
                doPrepare();
                run();
                doFinish();
            });
        }
        return 0;
    }

    virtual int stop() {
        if (status != STOP) {
            status = STOP;
            thread.join(); // wait thread exit
        }
        return 0;
    }

    virtual int pause() {
        if (status == RUNNING) {
            status = PAUSE;
        }
        return 0;
    }

    virtual int resume() {
        if (status == PAUSE) {
            status = RUNNING;
        }
        return 0;
    }

    virtual void run() {
        while (status != STOP) {
            while (status == PAUSE) {
                std::this_thread::yield();
            }

            doTask();
            dotask_cnt++;

            std::this_thread::yield();
        }
    }

    virtual void doPrepare() {}
    virtual void doTask() {}
    virtual void doFinish() {}

    std::thread thread;
    enum Status {
        STOP,
        RUNNING,
        PAUSE,
    };
    std::atomic<Status> status;
    uint32 dotask_cnt;
};

#endif // H_THREAD_H