pgsql_lease_mgr.h

Go to the documentation of this file.

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

#include <dhcp/hwaddr.h>
#include <dhcpsrv/dhcpsrv_exceptions.h>
#include <dhcpsrv/lease_mgr.h>
#include <pgsql/pgsql_connection.h>
#include <pgsql/pgsql_exchange.h>

#include <boost/scoped_ptr.hpp>
#include <boost/utility.hpp>

#include <vector>

namespace isc {
namespace dhcp {

// Forward definitions (needed for shared_ptr definitions)
// See pgsql_lease_mgr.cc file for actual class definitions
class PgSqlLease4Exchange;
class PgSqlLease6Exchange;

class PgSqlLeaseMgr : public LeaseMgr {
public:

    PgSqlLeaseMgr(const db::DatabaseConnection::ParameterMap& parameters);

    virtual ~PgSqlLeaseMgr();

    static std::string getDBVersion();

    virtual bool addLease(const Lease4Ptr& lease);

    virtual bool addLease(const Lease6Ptr& lease);

    virtual Lease4Ptr getLease4(const isc::asiolink::IOAddress& addr) const;

    virtual Lease4Collection getLease4(const isc::dhcp::HWAddr& hwaddr) const;

    virtual Lease4Ptr getLease4(const isc::dhcp::HWAddr& hwaddr,
                                SubnetID subnet_id) const;

    virtual Lease4Collection getLease4(const ClientId& clientid) const;

    virtual Lease4Ptr getLease4(const ClientId& client_id, const HWAddr& hwaddr,
                                SubnetID subnet_id) const;

    virtual Lease4Ptr getLease4(const ClientId& clientid,
                                SubnetID subnet_id) const;

    virtual Lease4Collection getLeases4(SubnetID subnet_id) const;

    virtual Lease4Collection getLeases4() const;

    virtual Lease4Collection
    getLeases4(const asiolink::IOAddress& lower_bound_address,
               const LeasePageSize& page_size) const;

    virtual Lease6Ptr getLease6(Lease::Type type,
                                const isc::asiolink::IOAddress& addr) const;

    virtual Lease6Collection getLeases6(Lease::Type type, const DUID& duid,
                                       uint32_t iaid) const;

    virtual Lease6Collection getLeases6(Lease::Type type, const DUID& duid,
                                        uint32_t iaid, SubnetID subnet_id) const;

    virtual Lease6Collection getLeases6(SubnetID subnet_id) const;

    virtual Lease6Collection getLeases6() const;

    virtual Lease6Collection getLeases6(const DUID& duid) const;
    
    virtual Lease6Collection
    getLeases6(const asiolink::IOAddress& lower_bound_address,
               const LeasePageSize& page_size) const;

    virtual void getExpiredLeases4(Lease4Collection& expired_leases,
                                   const size_t max_leases) const;

    virtual void getExpiredLeases6(Lease6Collection& expired_leases,
                                   const size_t max_leases) const;

    virtual void updateLease4(const Lease4Ptr& lease4);

    virtual void updateLease6(const Lease6Ptr& lease6);

    virtual bool deleteLease(const isc::asiolink::IOAddress& addr);

    virtual uint64_t deleteExpiredReclaimedLeases4(const uint32_t secs);

    virtual uint64_t deleteExpiredReclaimedLeases6(const uint32_t secs);

    virtual LeaseStatsQueryPtr startLeaseStatsQuery4();

    virtual LeaseStatsQueryPtr startSubnetLeaseStatsQuery4(const SubnetID& subnet_id);


    virtual LeaseStatsQueryPtr startSubnetRangeLeaseStatsQuery4(const SubnetID& first_subnet_id,
                                                                const SubnetID& last_subnet_id);
    virtual LeaseStatsQueryPtr startLeaseStatsQuery6();

    virtual LeaseStatsQueryPtr startSubnetLeaseStatsQuery6(const SubnetID& subnet_id);

    virtual LeaseStatsQueryPtr startSubnetRangeLeaseStatsQuery6(const SubnetID& first_subnet_id,
                                                                const SubnetID& last_subnet_id);
    virtual size_t wipeLeases4(const SubnetID& subnet_id);

    virtual size_t wipeLeases6(const SubnetID& subnet_id);

    virtual std::string getType() const {
        return (std::string("postgresql"));
    }

    virtual std::string getName() const;

    virtual std::string getDescription() const;

    virtual std::pair<uint32_t, uint32_t> getVersion() const;

    virtual void commit();

    virtual void rollback();

    enum StatementIndex {
        DELETE_LEASE4,              // Delete from lease4 by address
        DELETE_LEASE4_STATE_EXPIRED,// Delete expired lease4s in certain state.
        DELETE_LEASE6,              // Delete from lease6 by address
        DELETE_LEASE6_STATE_EXPIRED,// Delete expired lease6s in certain state.
        GET_LEASE4,                 // Get all IPv4 leases
        GET_LEASE4_ADDR,            // Get lease4 by address
        GET_LEASE4_CLIENTID,        // Get lease4 by client ID
        GET_LEASE4_CLIENTID_SUBID,  // Get lease4 by client ID & subnet ID
        GET_LEASE4_HWADDR,          // Get lease4 by HW address
        GET_LEASE4_HWADDR_SUBID,    // Get lease4 by HW address & subnet ID
        GET_LEASE4_PAGE,            // Get page of leases beginning with an address
        GET_LEASE4_SUBID,           // Get IPv4 leases by subnet ID
        GET_LEASE4_EXPIRE,          // Get expired lease4
        GET_LEASE6,                 // Get all IPv6 leases
        GET_LEASE6_ADDR,            // Get lease6 by address
        GET_LEASE6_DUID_IAID,       // Get lease6 by DUID and IAID
        GET_LEASE6_DUID_IAID_SUBID, // Get lease6 by DUID, IAID and subnet ID
        GET_LEASE6_PAGE,            // Get page of IPv6 leases beginning with an address
        GET_LEASE6_SUBID,           // Get IPv6 leases by subnet ID
        GET_LEASE6_DUID,           // Get IPv6 leases by DUID
        GET_LEASE6_EXPIRE,          // Get expired lease6
        INSERT_LEASE4,              // Add entry to lease4 table
        INSERT_LEASE6,              // Add entry to lease6 table
        UPDATE_LEASE4,              // Update a Lease4 entry
        UPDATE_LEASE6,              // Update a Lease6 entry
        ALL_LEASE4_STATS,           // Fetches IPv4 lease statistics
        SUBNET_LEASE4_STATS,        // Fetched IPv4 lease stats for a single subnet.
        SUBNET_RANGE_LEASE4_STATS,  // Fetched IPv4 lease stats for a subnet range.
        ALL_LEASE6_STATS,           // Fetches IPv6 lease statistics
        SUBNET_LEASE6_STATS,        // Fetched IPv6 lease stats for a single subnet.
        SUBNET_RANGE_LEASE6_STATS,  // Fetched IPv6 lease stats for a subnet range.
        NUM_STATEMENTS              // Number of statements
    };

private:

    bool addLeaseCommon(StatementIndex stindex, db::PsqlBindArray& bind_array);

    template <typename Exchange, typename LeaseCollection>
    void getLeaseCollection(StatementIndex stindex, db::PsqlBindArray& bind_array,
                            Exchange& exchange, LeaseCollection& result,
                            bool single = false) const;

    void getLeaseCollection(StatementIndex stindex, db::PsqlBindArray& bind_array,
                            Lease4Collection& result) const {
        getLeaseCollection(stindex, bind_array, exchange4_, result);
    }

    void getLeaseCollection(StatementIndex stindex, db::PsqlBindArray& bind_array,
                            Lease6Collection& result) const {
        getLeaseCollection(stindex, bind_array, exchange6_, result);
    }

    void getLease(StatementIndex stindex, db::PsqlBindArray& bind_array,
                  Lease4Ptr& result) const;

    void getLease(StatementIndex stindex, db::PsqlBindArray& bind_array,
                  Lease6Ptr& result) const;

    template<typename LeaseCollection>
    void getExpiredLeasesCommon(LeaseCollection& expired_leases,
                                const size_t max_leases,
                                StatementIndex statement_index) const;

    template <typename LeasePtr>
    void updateLeaseCommon(StatementIndex stindex, db::PsqlBindArray& bind_array,
                           const LeasePtr& lease);

    uint64_t deleteLeaseCommon(StatementIndex stindex,
                               db::PsqlBindArray& bind_array);

    uint64_t deleteExpiredReclaimedLeasesCommon(const uint32_t secs,
                                                StatementIndex statement_index);

    boost::scoped_ptr<PgSqlLease4Exchange> exchange4_; 
    boost::scoped_ptr<PgSqlLease6Exchange> exchange6_; 

    db::PgSqlConnection conn_;
};

}  // namespace dhcp
}  // namespace isc

#endif // PGSQL_LEASE_MGR_H