d2_process.h

Go to the documentation of this file.

// Copyright (C) 2013-2019 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 D2_PROCESS_H
#define D2_PROCESS_H

#include <d2/d2_queue_mgr.h>
#include <d2/d2_update_mgr.h>
#include <process/d_process.h>

namespace isc {
namespace d2 {

class D2Process : public process::DProcessBase {
public:

    enum ShutdownType {
      SD_NORMAL,
      SD_DRAIN_FIRST,
      SD_NOW
    };

    static const unsigned int QUEUE_RESTART_PERCENT;

    D2Process(const char* name, const asiolink::IOServicePtr& io_service);

    virtual void init();

    virtual void run();

    virtual isc::data::ConstElementPtr
        shutdown(isc::data::ConstElementPtr args);

    virtual isc::data::ConstElementPtr
    configure(isc::data::ConstElementPtr config_set,
              bool check_only = false);

    virtual ~D2Process();

protected:
    virtual void checkQueueStatus();

    virtual void reconfigureQueueMgr();

    virtual size_t runIO();

    virtual bool canShutdown() const;

    void setReconfQueueFlag(const bool value) {
        reconf_queue_flag_ = value;
    }

    void setShutdownType(const ShutdownType& value) {
        shutdown_type_ = value;
    }

    void reconfigureCommandChannel();

public:
    D2CfgMgrPtr getD2CfgMgr();

    const D2QueueMgrPtr& getD2QueueMgr() const {
        return (queue_mgr_);
    }

    const D2UpdateMgrPtr& getD2UpdateMgr() const {
        return (update_mgr_);
    }

    bool getReconfQueueFlag() const {
        return (reconf_queue_flag_);
    }

    ShutdownType getShutdownType() const {
        return (shutdown_type_);
    }

    static const char* getShutdownTypeStr(const ShutdownType& type);

private:
    D2QueueMgrPtr queue_mgr_;

    D2UpdateMgrPtr update_mgr_;

    bool reconf_queue_flag_;

    ShutdownType shutdown_type_;

    isc::data::ConstElementPtr current_control_socket_;

};

typedef boost::shared_ptr<D2Process> D2ProcessPtr;

}; // namespace isc::d2
}; // namespace isc

#endif