worker_thread.h

Go to the documentation of this file.

#ifndef __WORKER_THREAD_H
#define __WORKER_THREAD_H

#include "thread_wrapper.h"

#include <algorithm>
#include <chrono>
#include <mutex>
#include <condition_variable>
#include <atomic>

class worker_thread_t : public thread_wrapper_t {
public:
    worker_thread_t(int inverval_ms = -1);

    virtual ~worker_thread_t();

    void wakeup(bool wait = false, int rounds_to_wait = -1);

    void stop();

    void wait_for_round(long round = 0);

    long get_rounds_completed() const {
        return rounds_completed;
    };

    bool is_busy() const {
        return worker_busy;
    }

protected:

    virtual void do_work() = 0;

    bool should_exit() const {
        return stop_requested;
    }

    void notify_one();

    void notify_all();

    void quit();

private:
    virtual void run();

    std::mutex cond_mutex;

    std::condition_variable wakeup_condvar;

    std::condition_variable done_condvar;

    int interval_msec;

    std::atomic<bool> stop_requested;

    bool wakeup_requested;

    bool worker_busy;

    long rounds_completed;
};

class log_worker_thread_t : public worker_thread_t {
public:
    log_worker_thread_t(int interval_ms = -1)
            : worker_thread_t(interval_ms),
              endLSN(lsn_t::null) {}

    virtual ~log_worker_thread_t() {}

    void wakeup_until_lsn(lsn_t lsn, bool wait = false, int rounds_to_wait = -1);

    lsn_t getEndLSN() {
        return endLSN;
    }

private:
    std::atomic<lsn_t> endLSN;
};

#endif // __WORKER_THREAD_H