d2_update_mgr.h

Go to the documentation of this file.

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


#include <asiolink/io_service.h>
#include <exceptions/exceptions.h>
#include <d2/d2_log.h>
#include <d2/d2_queue_mgr.h>
#include <d2/d2_cfg_mgr.h>
#include <d2/nc_trans.h>

#include <boost/noncopyable.hpp>
#include <boost/shared_ptr.hpp>
#include <map>

namespace isc {
namespace d2 {

class D2UpdateMgrError : public isc::Exception {
public:
    D2UpdateMgrError(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) { };
};

typedef std::map<TransactionKey, NameChangeTransactionPtr> TransactionList;

class D2UpdateMgr : public boost::noncopyable {
public:
    static const size_t MAX_TRANSACTIONS_DEFAULT = 32;

    // @todo This structure is not yet used. It is here in anticipation of
    // enabled statistics capture.
    struct Stats {
        uint64_t start_time_;
        uint64_t stop_time_;
        uint64_t update_count_;
        uint64_t min_update_time_;
        uint64_t max_update_time_;
        uint64_t server_rejects_;
        uint64_t server_timeouts_;
    };

    D2UpdateMgr(D2QueueMgrPtr& queue_mgr, D2CfgMgrPtr& cfg_mgr,
                asiolink::IOServicePtr& io_service,
                const size_t max_transactions = MAX_TRANSACTIONS_DEFAULT);

    virtual ~D2UpdateMgr();

    void sweep();

protected:
    void checkFinishedTransactions();

    void pickNextJob();

    void makeTransaction(isc::dhcp_ddns::NameChangeRequestPtr& ncr);

public:
    const asiolink::IOServicePtr& getIOService() {
        return (io_service_);
    }

    size_t getMaxTransactions() const {
        return (max_transactions_);
    }

    void setMaxTransactions(const size_t max_transactions);

    TransactionList::iterator findTransaction(const TransactionKey& key);

    TransactionList::iterator transactionListEnd();

    TransactionList::iterator transactionListBegin();

    bool hasTransaction(const TransactionKey& key);

    void removeTransaction(const TransactionKey& key);

    void clearTransactionList();

    size_t getQueueCount() const;

    size_t getTransactionCount() const;

private:
    D2QueueMgrPtr queue_mgr_;

    D2CfgMgrPtr cfg_mgr_;

    asiolink::IOServicePtr io_service_;

    size_t max_transactions_;

    TransactionList transaction_list_;
};

typedef boost::shared_ptr<D2UpdateMgr> D2UpdateMgrPtr;


} // namespace isc::d2
} // namespace isc
#endif