event_task.h

Go to the documentation of this file.

/*
 * Copyright (c) 2013-2014 ADTECH GmbH
 * Licensed under MIT (https://github.com/adtechlabs/libtasks/blob/master/COPYING)
 *
 * Author: Andreas Pohl
 */

#ifndef _TASKS_EVENT_TASK_H_
#define _TASKS_EVENT_TASK_H_

#include <tasks/task.h>
#include <tasks/error_base.h>
#include <tasks/dispatcher.h>

namespace tasks {

typedef std::function<void(struct ev_loop*)> task_func_t;

class event_task : public task, public error_base {
  public:
    typedef std::function<void(worker* worker, const tasks_exception& e)> error_func_worker_t;
    typedef std::function<void(const tasks_exception& e)> error_func_void_t;
    struct error_func_t {
        error_func_t(error_func_worker_t f) : m_type(0), m_f_worker(f) {}
        error_func_t(error_func_void_t f) : m_type(1), m_f_void(f) {}

        void operator()(worker* worker, const tasks_exception& e) {
            switch (m_type) {
                case 0:
                    m_f_worker(worker, e);
                    break;
                case 1:
                    m_f_void(e);
                    break;
            }
        }

        int m_type = 0;
        error_func_worker_t m_f_worker;
        error_func_void_t m_f_void;
    };

    virtual ~event_task() {}

    virtual bool handle_event(worker* worker, int events) = 0;

    virtual void init_watcher() = 0;

    virtual void stop_watcher(worker* worker) = 0;

    virtual void start_watcher(worker* worker) = 0;

    inline worker* assigned_worker() const { return m_worker; }

    void assign_worker(worker* worker);

    void notify_error(worker* worker = nullptr);

    inline void on_error(error_func_worker_t f) {
        m_error_funcs.push_back(error_func_t(f));
    }

    inline void on_error(error_func_void_t f) {
        m_error_funcs.push_back(error_func_t(f));
    }

  private:
    worker* m_worker = nullptr;
    std::vector<error_func_t> m_error_funcs;
};

}  // tasks

#endif  // _TASKS_EVENT_TASK_H_