mir/kea
2
0
Fork 0
mirror of https://gitlab.isc.org/isc-projects/kea synced 2025-08-31 05:55:28 +00:00
Code Issues Releases Wiki Activity

[5477] Addressed review comments

Browse Source 
src/bin/dhcp4/ctrl_dhcp4_srv.*
src/bin/dhcp6/ctrl_dhcp6_srv.*
    Changed dbReconnect() to accept ReconnectCtlPtr
    Added commentary for dbReconnect and dbLostCallback

src/lib/dhcpsrv/database_connection.h
    Removed extraneous typedef

many files:
    Changed DatabaseConnection::Callback to ::DbLostCallback

src/lib/dhcpsrv/tests/database_connection_unittest.cc
    Added commentary to text fixture and tests
This commit is contained in:
Thomas Markwalder
2018-03-07 11:10:34 -05:00
parent 8e55883c27
commit 8f5ce70600
19 changed files with 142 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/bin/dhcp4/ctrl_dhcp4_srv.cc
View File

@@ -650,8 +650,7 @@ ControlledDhcpv4Srv::checkConfig(isc::data::ConstElementPtr config) {
}
ControlledDhcpv4Srv::ControlledDhcpv4Srv(uint16_t port /*= DHCP4_SERVER_PORT*/)
: Dhcpv4Srv(port), io_service_(), timer_mgr_(TimerMgr::instance()),
db_reconnect_ctl_(NULL) {
: Dhcpv4Srv(port), io_service_(), timer_mgr_(TimerMgr::instance()) {
if (getInstance()) {
isc_throw(InvalidOperation,
"There is another Dhcpv4Srv instance already.");
@@ -795,7 +794,7 @@ ControlledDhcpv4Srv::deleteExpiredReclaimedLeases(const uint32_t secs) {
}
void
ControlledDhcpv4Srv::dbReconnect() {
ControlledDhcpv4Srv::dbReconnect(ReconnectCtlPtr db_reconnect_ctl) {
bool reopened = false;
// Re-open lease and host database with new parameters.
@@ -817,24 +816,25 @@ ControlledDhcpv4Srv::dbReconnect() {
network_state_.enableService();
// Toss the reconnect control, we're done with it
db_reconnect_ctl_.reset();
db_reconnect_ctl.reset();
} else {
if (!db_reconnect_ctl_->checkRetries()) {
if (!db_reconnect_ctl->checkRetries()) {
LOG_ERROR(dhcp4_logger, DHCP4_DB_RECONNECT_RETRIES_EXHAUSTED)
.arg(db_reconnect_ctl_->maxRetries());
.arg(db_reconnect_ctl->maxRetries());
shutdown();
return;
}
LOG_INFO(dhcp4_logger, DHCP4_DB_RECONNECT_ATTEMPT_SCHEDULE)
.arg(db_reconnect_ctl_->maxRetries() - db_reconnect_ctl_->retriesLeft() + 1)
.arg(db_reconnect_ctl_->maxRetries())
.arg(db_reconnect_ctl_->retryInterval());
.arg(db_reconnect_ctl->maxRetries() - db_reconnect_ctl->retriesLeft() + 1)
.arg(db_reconnect_ctl->maxRetries())
.arg(db_reconnect_ctl->retryInterval());
if (!TimerMgr::instance()->isTimerRegistered("Dhcp4DbReconnectTimer")) {
TimerMgr::instance()->registerTimer("Dhcp4DbReconnectTimer",
boost::bind(&ControlledDhcpv4Srv::dbReconnect, this),
db_reconnect_ctl_->retryInterval() * 1000,
boost::bind(&ControlledDhcpv4Srv::dbReconnect, this,
db_reconnect_ctl),
db_reconnect_ctl->retryInterval() * 1000,
asiolink::IntervalTimer::ONE_SHOT);
}
@@ -862,11 +862,8 @@ ControlledDhcpv4Srv::dbLostCallback(ReconnectCtlPtr db_reconnect_ctl) {
return(false);
}
// Save the reconnect control
db_reconnect_ctl_ = db_reconnect_ctl;
// Invoke reconnect method
dbReconnect();
dbReconnect(db_reconnect_ctl);
return(true);
}

39
src/bin/dhcp4/ctrl_dhcp4_srv.h
View File

@@ -319,9 +319,43 @@ private:
void deleteExpiredReclaimedLeases(const uint32_t secs);
/// @brief Attempts to reconnect the server to the DB backend managers
void dbReconnect();
///
/// This is a self-rescheduling function that attempts to reconnect to the
/// server's DB backends after connectivity to one or more have been
/// lost. Upon entry it will attempt to reconnect via @ref CfgDdbAccess::
/// createManagers. If this is succesful, DHCP servicing is re-enabled and
/// server returns to normal operation.
///
/// If reconnection fails and the maximum number of retries has not been
/// exhausted, it will schedule a call to itself to occur at the
/// configured retry interval. DHCP service remains disabled.
///
/// If the maximum number of retries has been exhausted an error is logged
/// and the server shuts down.
///
/// @param db_reconnect_ctl pointer to the ReconnectCtl containing the
/// configured reconnect parameters
///
void dbReconnect(ReconnectCtlPtr db_reconnect_ctl);
/// @brief Callback DB backends should invoke upon loss of connectivity
///
/// This function is invoked by DB backends when they detect a loss of
/// connectivity. The parameter, db_reconnect_ctl, conveys the configured
/// maximum number of reconnect retries as well as the interval to wait
/// between retry attempts.
///
/// If either value is zero, reconnect is presumed to be disabled and
/// the function will returns false. This instructs the DB backend
/// layer (the caller) to treat the connectivity loss as fatal.
///
/// Otherwise, the function saves db_reconnect_ctl and invokes
/// dbReconnect to initiate the reconnect process.
///
/// @param db_reconnect_ctl pointer to the ReconnectCtl containing the
/// configured reconnect parameters
///
/// @return false if reconnect is not configured, true otherwise
bool dbLostCallback(ReconnectCtlPtr db_reconnect_ctl);
/// @brief Static pointer to the sole instance of the DHCP server.
@@ -338,9 +372,6 @@ private:
/// Shared pointer to the instance of timer @c TimerMgr is held here to
/// make sure that the @c TimerMgr outlives instance of this class.
TimerMgrPtr timer_mgr_;
/// @brief Pointer the current DB reconnect control values
ReconnectCtlPtr db_reconnect_ctl_;
};
}; // namespace isc::dhcp

27
src/bin/dhcp6/ctrl_dhcp6_srv.cc
View File

@@ -673,8 +673,7 @@ ControlledDhcpv6Srv::checkConfig(isc::data::ConstElementPtr config) {
}
ControlledDhcpv6Srv::ControlledDhcpv6Srv(uint16_t port)
: Dhcpv6Srv(port), io_service_(), timer_mgr_(TimerMgr::instance()),
db_reconnect_ctl_(NULL) {
: Dhcpv6Srv(port), io_service_(), timer_mgr_(TimerMgr::instance()) {
if (server_) {
isc_throw(InvalidOperation,
"There is another Dhcpv6Srv instance already.");
@@ -817,7 +816,7 @@ ControlledDhcpv6Srv::deleteExpiredReclaimedLeases(const uint32_t secs) {
}
void
ControlledDhcpv6Srv::dbReconnect() {
ControlledDhcpv6Srv::dbReconnect(ReconnectCtlPtr db_reconnect_ctl) {
bool reopened = false;
// Re-open lease and host database with new parameters.
@@ -838,24 +837,25 @@ ControlledDhcpv6Srv::dbReconnect() {
network_state_.enableService();
// Toss the reconnct control, we're done with it
db_reconnect_ctl_.reset();
db_reconnect_ctl.reset();
} else {
if (!db_reconnect_ctl_->checkRetries()) {
if (!db_reconnect_ctl->checkRetries()) {
LOG_ERROR(dhcp6_logger, DHCP6_DB_RECONNECT_RETRIES_EXHAUSTED)
.arg(db_reconnect_ctl_->maxRetries());
.arg(db_reconnect_ctl->maxRetries());
shutdown();
return;
}
LOG_INFO(dhcp6_logger, DHCP6_DB_RECONNECT_ATTEMPT_SCHEDULE)
.arg(db_reconnect_ctl_->maxRetries() - db_reconnect_ctl_->retriesLeft() + 1)
.arg(db_reconnect_ctl_->maxRetries())
.arg(db_reconnect_ctl_->retryInterval());
.arg(db_reconnect_ctl->maxRetries() - db_reconnect_ctl->retriesLeft() + 1)
.arg(db_reconnect_ctl->maxRetries())
.arg(db_reconnect_ctl->retryInterval());
if (!TimerMgr::instance()->isTimerRegistered("Dhcp6DbReconnectTimer")) {
TimerMgr::instance()->registerTimer("Dhcp6DbReconnectTimer",
boost::bind(&ControlledDhcpv6Srv::dbReconnect, this),
db_reconnect_ctl_->retryInterval() * 1000,
boost::bind(&ControlledDhcpv6Srv::dbReconnect, this,
db_reconnect_ctl),
db_reconnect_ctl->retryInterval() * 1000,
asiolink::IntervalTimer::ONE_SHOT);
}
@@ -883,11 +883,8 @@ ControlledDhcpv6Srv::dbLostCallback(ReconnectCtlPtr db_reconnect_ctl) {
return(false);
}
// Save the reconnect control
db_reconnect_ctl_ = db_reconnect_ctl;
// Invoke reconnect method
dbReconnect();
dbReconnect(db_reconnect_ctl);
return(true);
}

36
src/bin/dhcp6/ctrl_dhcp6_srv.h
View File

@@ -319,9 +319,40 @@ private:
void deleteExpiredReclaimedLeases(const uint32_t secs);
/// @brief Attempts to reconnect the server to the DB backend managers
void dbReconnect();
///
/// This is a self-rescheduling function that attempts to reconnect to the
/// server's DB backends after connectivity to one or more have been
/// lost. Upon entry it will attempt to reconnect via @ref CfgDdbAccess::
/// createManagers. If this is succesful, DHCP servicing is re-enabled and
/// server returns to normal operation.
///
/// If reconnection fails and the maximum number of retries has not been
/// exhausted, it will schedule a call to itself to occur at the
/// configured retry interval. DHCP service remains disabled.
///
/// If the maximum number of retries has been exhausted an error is logged
/// and the server shuts down.
/// @param db_reconnect_ctl pointer to the ReconnectCtl containing the
/// configured reconnect parameters
///
void dbReconnect(ReconnectCtlPtr db_reconnect_ctl);
/// @brief Callback DB backends should invoke upon loss of connectivity
///
/// This function is invoked by DB backends when they detect a loss of
/// connectivity. The parameter, db_reconnect_ctl, conveys the configured
/// maximum number of reconnect retries as well as the interval to wait
/// between retry attempts.
///
/// If either value is zero, reconnect is presumed to be disabled and
/// the function will returns false. This instructs the DB backend
/// layer (the caller) to treat the connectivity loss as fatal.
///
/// Otherwise, the function saves db_reconnect_ctl and invokes
/// dbReconnect to initiate the reconnect process.
///
/// @param db_reconnect_ctl pointer to the ReconnectCtl containing the
/// configured reconnect parameters
bool dbLostCallback(ReconnectCtlPtr db_reconnect_ctl);
/// @brief Static pointer to the sole instance of the DHCP server.
@@ -338,9 +369,6 @@ private:
/// Shared pointer to the instance of timer @c TimerMgr is held here to
/// make sure that the @c TimerMgr outlives instance of this class.
TimerMgrPtr timer_mgr_;
/// @brief Pointer the current DB reconnect control values
ReconnectCtlPtr db_reconnect_ctl_;
};
}; // namespace isc::dhcp

2
src/lib/dhcpsrv/cfg_db_access.cc
View File

@@ -38,7 +38,7 @@ CfgDbAccess::getHostDbAccessString() const {
void
CfgDbAccess::createManagers(DatabaseConnection::Callback db_lost_callback) const {
CfgDbAccess::createManagers(DatabaseConnection::DbLostCallback db_lost_callback) const {
// Recreate lease manager.
LeaseMgrFactory::destroy();
LeaseMgrFactory::create(getLeaseDbAccessString(), db_lost_callback);

2
src/lib/dhcpsrv/cfg_db_access.h
View File

@@ -68,7 +68,7 @@ public:
/// @param db_lost_callback function to invoke if connectivity to
/// either the lease or host managers, once established, is subsequently
/// lost.
void createManagers(DatabaseConnection::Callback db_lost_callback = NULL) const;
void createManagers(DatabaseConnection::DbLostCallback db_lost_callback = NULL) const;
/// @brief Unparse an access string
///

7
src/lib/dhcpsrv/database_connection.h
View File

@@ -141,8 +141,7 @@ public:
typedef std::map<std::string, std::string> ParameterMap;
/// @brief Defines a callback prototype for propogating events upward
/// typedef boost::function<bool (const ParameterMap& parameters)> Callback;
typedef boost::function<bool (ReconnectCtlPtr db_retry)> Callback;
typedef boost::function<bool (ReconnectCtlPtr db_retry)> DbLostCallback;
/// @brief Constructor
///
@@ -151,7 +150,7 @@ public:
/// @param db_lost_callback Optional call back function to invoke if a
/// successfully open connection subsequently fails
DatabaseConnection(const ParameterMap& parameters,
Callback db_lost_callback = NULL)
DbLostCallback db_lost_callback = NULL)
:parameters_(parameters), db_lost_callback_(db_lost_callback) {
}
@@ -221,7 +220,7 @@ private:
protected:
/// @brief Optional function to invoke if the connectivity is lost
Callback db_lost_callback_;
DbLostCallback db_lost_callback_;
};
}; // end of isc::dhcp namespace

2
src/lib/dhcpsrv/host_data_source_factory.cc
View File

@@ -46,7 +46,7 @@ HostDataSourceFactory::getHostDataSourcePtr() {
void
HostDataSourceFactory::create(const std::string& dbaccess,
DatabaseConnection::Callback db_lost_callback) {
DatabaseConnection::DbLostCallback db_lost_callback) {
// Parse the access string and create a redacted string for logging.
DatabaseConnection::ParameterMap parameters =
DatabaseConnection::parse(dbaccess);

2
src/lib/dhcpsrv/host_data_source_factory.h
View File

@@ -67,7 +67,7 @@ public:
/// @throw isc::dhcp::InvalidType The "type" keyword in dbaccess does not
/// identify a supported backend.
static void create(const std::string& dbaccess,
DatabaseConnection::Callback db_lost_callback = NULL);
DatabaseConnection::DbLostCallback db_lost_callback = NULL);
/// @brief Destroy host data source
///

2
src/lib/dhcpsrv/host_mgr.cc
View File

@@ -38,7 +38,7 @@ HostMgr::getHostMgrPtr() {
void
HostMgr::create(const std::string& access,
DatabaseConnection::Callback db_lost_callback) {
DatabaseConnection::DbLostCallback db_lost_callback) {
getHostMgrPtr().reset(new HostMgr());
if (!access.empty()) {

2
src/lib/dhcpsrv/host_mgr.h
View File

@@ -73,7 +73,7 @@ public:
/// @param db_lost_callback function to invoke if connectivity to
/// the host database is lost.
static void create(const std::string& access = "",
DatabaseConnection::Callback db_lost_callback = NULL);
DatabaseConnection::DbLostCallback db_lost_callback = NULL);
/// @brief Returns a sole instance of the @c HostMgr.
///

2
src/lib/dhcpsrv/lease_mgr_factory.cc
View File

@@ -42,7 +42,7 @@ LeaseMgrFactory::getLeaseMgrPtr() {
void
LeaseMgrFactory::create(const std::string& dbaccess,
DatabaseConnection::Callback db_lost_callback) {
DatabaseConnection::DbLostCallback db_lost_callback) {
const std::string type = "type";
// Parse the access string and create a redacted string for logging.

2
src/lib/dhcpsrv/lease_mgr_factory.h
View File

@@ -71,7 +71,7 @@ public:
/// @throw isc::dhcp::InvalidType The "type" keyword in dbaccess does not
/// identify a supported backend.
static void create(const std::string& dbaccess,
DatabaseConnection::Callback db_lost_callback = NULL);
DatabaseConnection::DbLostCallback db_lost_callback = NULL);
/// @brief Destroy lease manager
///

2
src/lib/dhcpsrv/pgsql_connection.h
View File

@@ -304,7 +304,7 @@ public:
}
PgSqlConnection(const ParameterMap& parameters,
Callback db_lost_callback)
DbLostCallback db_lost_callback)
: DatabaseConnection(parameters, db_lost_callback) {
}

6
src/lib/dhcpsrv/pgsql_host_data_source.cc
View File

@@ -1285,7 +1285,7 @@ public:
/// This constructor opens database connection and initializes prepared
/// statements used in the queries.
PgSqlHostDataSourceImpl(const PgSqlConnection::ParameterMap& parameters,
DatabaseConnection::Callback db_lost_callback);
DatabaseConnection::DbLostCallback db_lost_callback);
/// @brief Destructor.
~PgSqlHostDataSourceImpl();
@@ -1701,7 +1701,7 @@ TaggedStatementArray tagged_statements = { {
PgSqlHostDataSourceImpl::
PgSqlHostDataSourceImpl(const PgSqlConnection::ParameterMap& parameters,
DatabaseConnection::Callback db_lost_callback)
DatabaseConnection::DbLostCallback db_lost_callback)
: host_exchange_(new PgSqlHostWithOptionsExchange(PgSqlHostWithOptionsExchange::DHCP4_ONLY)),
host_ipv6_exchange_(new PgSqlHostIPv6Exchange(PgSqlHostWithOptionsExchange::DHCP6_ONLY)),
host_ipv46_exchange_(new PgSqlHostIPv6Exchange(PgSqlHostWithOptionsExchange::
@@ -1929,7 +1929,7 @@ PgSqlHostDataSourceImpl::checkReadOnly() const {
PgSqlHostDataSource::
PgSqlHostDataSource(const PgSqlConnection::ParameterMap& parameters,
DatabaseConnection::Callback db_lost_callback)
DatabaseConnection::DbLostCallback db_lost_callback)
: impl_(new PgSqlHostDataSourceImpl(parameters, db_lost_callback)) {
}

2
src/lib/dhcpsrv/pgsql_host_data_source.h
View File

@@ -61,7 +61,7 @@ public:
/// @throw isc::dhcp::DbOperationError An operation on the open database has
/// failed.
PgSqlHostDataSource(const DatabaseConnection::ParameterMap& parameters,
DatabaseConnection::Callback db_lost_callback = NULL);
DatabaseConnection::DbLostCallback db_lost_callback = NULL);
/// @brief Virtual destructor.
/// Frees database resources and closes the database connection through

2
src/lib/dhcpsrv/pgsql_lease_mgr.cc
View File

@@ -821,7 +821,7 @@ protected:
};
PgSqlLeaseMgr::PgSqlLeaseMgr(const DatabaseConnection::ParameterMap& parameters,
DatabaseConnection::Callback db_lost_callback)
DatabaseConnection::DbLostCallback db_lost_callback)
: LeaseMgr(), exchange4_(new PgSqlLease4Exchange()),
exchange6_(new PgSqlLease6Exchange()), conn_(parameters, db_lost_callback) {
conn_.openDatabase();

2
src/lib/dhcpsrv/pgsql_lease_mgr.h
View File

@@ -59,7 +59,7 @@ public:
/// @throw isc::dhcp::DbOperationError An operation on the open database has
/// failed.
PgSqlLeaseMgr(const DatabaseConnection::ParameterMap& parameters,
DatabaseConnection::Callback db_lost_callback = NULL);
DatabaseConnection::DbLostCallback db_lost_callback = NULL);
/// @brief Destructor (closes database)
virtual ~PgSqlLeaseMgr();

37
src/lib/dhcpsrv/tests/database_connection_unittest.cc
View File

@@ -13,21 +13,28 @@
using namespace isc::dhcp;
/// @brief Test fixture for excersizing DbLostCallback invocation
class DatabaseConnectionCallbackTest : public ::testing::Test {
public:
/// Constructor
DatabaseConnectionCallbackTest()
: db_reconnect_ctl_(0) {
}
bool dbLostCallback(ReconnectCtlPtr db_conn_retry) {
if (!db_conn_retry) {
isc_throw(isc::BadValue, "db_conn_retry should not null");
/// @brief Callback to register with a DatabaseConnection
///
/// @param db_reconnect_ctl ReconnectCtl containing reconnect
/// parameters
bool dbLostCallback(ReconnectCtlPtr db_reconnect_ctl) {
if (!db_reconnect_ctl) {
isc_throw(isc::BadValue, "db_reconnect_ctl should not null");
}
db_reconnect_ctl_ = db_conn_retry;
db_reconnect_ctl_ = db_reconnect_ctl;
return (true);
}
/// @brief Retainer for the control passed into the callback
ReconnectCtlPtr db_reconnect_ctl_;
};
@@ -49,6 +56,11 @@ TEST(DatabaseConnectionTest, getParameter) {
EXPECT_THROW(datasrc.getParameter("param3"), isc::BadValue);
}
/// @brief NoDbLostCallback
///
/// This test verifies that DatabaseConnection::invokeDbLostCallback
/// returns a false if there is connection has no registered
/// DbLostCallback.
TEST_F(DatabaseConnectionCallbackTest, NoDbLostCallback) {
DatabaseConnection::ParameterMap pmap;
pmap[std::string("max-reconnect-tries")] = std::string("3");
@@ -61,12 +73,25 @@ TEST_F(DatabaseConnectionCallbackTest, NoDbLostCallback) {
EXPECT_FALSE(db_reconnect_ctl_);
}
/// @brief NoDbLostCallback
///
/// This test verifies that DatabaseConnection::invokeDbLostCallback
/// safely invokes the registered DbLostCallback. It also tests
/// operation of DbReconnectCtl retry accounting methods.
TEST_F(DatabaseConnectionCallbackTest, dbLostCallback) {
/// Create a Database configuration that includes the reconnect
/// control parameters.
DatabaseConnection::ParameterMap pmap;
pmap[std::string("max-reconnect-tries")] = std::string("3");
pmap[std::string("reconnect-wait-time")] = std::string("60");
/// Create the connection with a DbLostCallback.
DatabaseConnection datasrc(pmap, boost::bind(&DatabaseConnectionCallbackTest
::dbLostCallback, this, _1));
/// We should be able to invoke the callback and glean
/// the correct reconnect contorl parameters from it.
bool ret;
ASSERT_NO_THROW(ret = datasrc.invokeDbLostCallback());
EXPECT_TRUE(ret);
@@ -75,11 +100,15 @@ TEST_F(DatabaseConnectionCallbackTest, dbLostCallback) {
ASSERT_EQ(3, db_reconnect_ctl_->retriesLeft());
EXPECT_EQ(60, db_reconnect_ctl_->retryInterval());
/// Verify that checkRetries() correctly decrements
/// down to zero, and that retriesLeft() returns
/// the correct value.
for (int i = 3; i > 1 ; --i) {
ASSERT_EQ(i, db_reconnect_ctl_->retriesLeft());
ASSERT_TRUE(db_reconnect_ctl_->checkRetries());
}
/// Retries are exhausted, verify that's reflected.
EXPECT_FALSE(db_reconnect_ctl_->checkRetries());
EXPECT_EQ(0, db_reconnect_ctl_->retriesLeft());
EXPECT_EQ(3, db_reconnect_ctl_->maxRetries());