memfile_lease_limits.h

Go to the documentation of this file.

// Copyright (C) 2022 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 MEMFILE_LEASE_LIMITS_H
#define MEMFILE_LEASE_LIMITS_H

#include <asiolink/io_address.h>
#include <cc/data.h>
#include <dhcp/classify.h>
#include <dhcpsrv/lease.h>
#include <dhcpsrv/subnet_id.h>
#include <stats/stats_mgr.h>

#include <unordered_map>

namespace isc {
namespace dhcp {

class ClassLeaseCounter {
public:
    typedef std::unordered_map<ClientClass, size_t> ClassCountMap;

    ClassLeaseCounter() = default;

    ~ClassLeaseCounter() = default;

    size_t getClassCount(const ClientClass& client_class,
                         const Lease::Type& ltype = Lease::TYPE_V4) const;

    void setClassCount(const ClientClass& client_class, size_t count,
                       const Lease::Type& ltype = Lease::TYPE_V4);

    void adjustClassCount(const ClientClass& client_class, int offset,
                          const Lease::Type& ltype = Lease::TYPE_V4);

    void adjustClassCounts(data::ConstElementPtr classes, int offset,
                           const Lease::Type& ltype = Lease::TYPE_V4);

    void addLease(LeasePtr lease);

    void updateLease(LeasePtr new_lease, LeasePtr old_lease);

    void removeLease(LeasePtr lease);

    void clear() {
        addresses_by_class_.clear();
        pds_by_class_.clear();
    }

    size_t size(const Lease::Type& ltype = Lease::TYPE_V4) const {
        return (getConstCountMap(ltype).size());
    }

    static data::ConstElementPtr getLeaseClientClasses(LeasePtr lease);

private:
    ClassCountMap& getCountMap(const Lease::Type& ltype = Lease::TYPE_V4) {
        return (ltype == Lease::TYPE_PD ? pds_by_class_ : addresses_by_class_);
    }

    const ClassCountMap& getConstCountMap(const Lease::Type& ltype = Lease::TYPE_V4) const {
        return (ltype == Lease::TYPE_PD ? pds_by_class_ : addresses_by_class_);
    }

    ClassCountMap addresses_by_class_;

    ClassCountMap pds_by_class_;
};

}  // namespace dhcp
}  // namespace isc

#endif // MEMFILE_LEASE_LIMITS_H