watched_thread.h

Go to the documentation of this file.

// Copyright (C) 2018-2020 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#ifndef WATCHED_THREAD_H
#define WATCHED_THREAD_H

#include <util/watch_socket.h>

#include <boost/shared_ptr.hpp>

#include <functional>
#include <mutex>
#include <thread>

namespace isc {
namespace util {

typedef boost::shared_ptr<std::thread> ThreadPtr;

class WatchedThread {
public:
    enum WatchType {
        ERROR = 0,
        READY = 1,
        TERMINATE = 2
    };

    WatchedThread(){};

    virtual ~WatchedThread(){}

    int getWatchFd(WatchType watch_type);

    void markReady(WatchType watch_type);

    bool isReady(WatchType watch_type);

    void clearReady(WatchType watch_type);

    bool shouldTerminate();

    void start(const std::function<void()>& thread_main);

    bool isRunning() {
        return (thread_ != 0);
    }

    void stop();

    void setError(const std::string& error_msg);

    std::string getLastError();

private:

    void setErrorInternal(const std::string& error_msg);

    std::string last_error_;

    std::mutex mutex_;

    WatchSocket sockets_[TERMINATE + 1];

    ThreadPtr thread_ ;
};

typedef boost::shared_ptr<WatchedThread> WatchedThreadPtr;

}  // namespace util
}  // namespace isc

#endif // WATCHED_THREAD_H