feiq/feiqlib/msgqueuethread.h

#ifndef MSGQUEUETHREAD_H
#define MSGQUEUETHREAD_H

#include <mutex>
#include <queue>
#include <functional>
#include <condition_variable>
#include <thread>
using namespace std;

//TODO:实现移到cpp中
template<class Msg>
class MsgQueueThread
{
    typedef function<void (shared_ptr<Msg>)> Handler;
public:
    void setHandler(Handler handler)
    {
        mHandler = handler;
    }

    void start()
    {
        if (mRun)
            return;

        mRun=true;
        thread thd(&MsgQueueThread::loop, this);
        mThread.swap(thd);
    }

    void stop()
    {
        if (!mRun)
            return;

        mRun=false;
        unique_lock<mutex> lock(mQueueMutex);
        while(!mQueue.empty())
            mQueue.pop();
        mQueueCnd.notify_all();
        lock.unlock();
        mThread.join();
    }

    void sendMessage(shared_ptr<Msg> msg)
    {
        unique_lock<mutex> lock(mQueueMutex);
        mQueue.push(msg);
        mQueueCnd.notify_one();
    }

private:
    void loop()
    {
        while (mRun) {
            unique_lock<mutex> lock(mQueueMutex);
            if (mQueue.empty())
                mQueueCnd.wait(lock);
            if (!mRun)
                return;

            auto msg = mQueue.front();
            mQueue.pop();
            lock.unlock();//队列已经没有利用价值了，放了它

            if (mHandler)
                mHandler(msg);
        }

    }

private:
    condition_variable mQueueCnd;
    mutex mQueueMutex;
    queue<shared_ptr<Msg>> mQueue;
    bool mRun=false;
    Handler mHandler;
    thread mThread;
};

#endif // MSGQUEUETHREAD_H