SetShouldConnect(true);
- std::vector<DbObject::Ptr> active_dbobjs;
+ bool reconnect = false;
- {
- bool reconnect = false;
-
- if (GetConnected()) {
- /* Check if we're really still connected */
- try {
- Query("SELECT 1");
- return;
- } catch (const std::exception&) {
- PQfinish(m_Connection);
- SetConnected(false);
- reconnect = true;
- }
+ if (GetConnected()) {
+ /* Check if we're really still connected */
+ try {
+ Query("SELECT 1");
+ return;
+ } catch (const std::exception&) {
+ PQfinish(m_Connection);
+ SetConnected(false);
+ reconnect = true;
}
+ }
- ClearIDCache();
+ ClearIDCache();
- String ihost, iport, iuser, ipasswd, idb;
- const char *host, *port, *user , *passwd, *db;
+ String ihost, iport, iuser, ipasswd, idb;
+ const char *host, *port, *user , *passwd, *db;
- ihost = GetHost();
- iport = GetPort();
- iuser = GetUser();
- ipasswd = GetPassword();
- idb = GetDatabase();
+ ihost = GetHost();
+ iport = GetPort();
+ iuser = GetUser();
+ ipasswd = GetPassword();
+ idb = GetDatabase();
- host = (!ihost.IsEmpty()) ? ihost.CStr() : NULL;
- port = (!iport.IsEmpty()) ? iport.CStr() : NULL;
- user = (!iuser.IsEmpty()) ? iuser.CStr() : NULL;
- passwd = (!ipasswd.IsEmpty()) ? ipasswd.CStr() : NULL;
- db = (!idb.IsEmpty()) ? idb.CStr() : NULL;
+ host = (!ihost.IsEmpty()) ? ihost.CStr() : NULL;
+ port = (!iport.IsEmpty()) ? iport.CStr() : NULL;
+ user = (!iuser.IsEmpty()) ? iuser.CStr() : NULL;
+ passwd = (!ipasswd.IsEmpty()) ? ipasswd.CStr() : NULL;
+ db = (!idb.IsEmpty()) ? idb.CStr() : NULL;
- m_Connection = PQsetdbLogin(host, port, NULL, NULL, db, user, passwd);
+ m_Connection = PQsetdbLogin(host, port, NULL, NULL, db, user, passwd);
- if (!m_Connection)
- return;
+ if (!m_Connection)
+ return;
- if (PQstatus(m_Connection) != CONNECTION_OK) {
- String message = PQerrorMessage(m_Connection);
- PQfinish(m_Connection);
- SetConnected(false);
+ if (PQstatus(m_Connection) != CONNECTION_OK) {
+ String message = PQerrorMessage(m_Connection);
+ PQfinish(m_Connection);
+ SetConnected(false);
- Log(LogCritical, "IdoPgsqlConnection")
- << "Connection to database '" << db << "' with user '" << user << "' on '" << host << ":" << port
- << "' failed: \"" << message << "\"";
+ Log(LogCritical, "IdoPgsqlConnection")
+ << "Connection to database '" << db << "' with user '" << user << "' on '" << host << ":" << port
+ << "' failed: \"" << message << "\"";
- BOOST_THROW_EXCEPTION(std::runtime_error(message));
- }
+ BOOST_THROW_EXCEPTION(std::runtime_error(message));
+ }
- SetConnected(true);
+ SetConnected(true);
- IdoPgsqlResult result;
+ IdoPgsqlResult result;
- /* explicitely require legacy mode for string escaping in PostgreSQL >= 9.1
- * changing standard_conforming_strings to on by default
- */
- if (PQserverVersion(m_Connection) >= 90100)
- result = Query("SET standard_conforming_strings TO off");
+ /* explicitely require legacy mode for string escaping in PostgreSQL >= 9.1
+ * changing standard_conforming_strings to on by default
+ */
+ if (PQserverVersion(m_Connection) >= 90100)
+ result = Query("SET standard_conforming_strings TO off");
- String dbVersionName = "idoutils";
- result = Query("SELECT version FROM " + GetTablePrefix() + "dbversion WHERE name=E'" + Escape(dbVersionName) + "'");
+ String dbVersionName = "idoutils";
+ result = Query("SELECT version FROM " + GetTablePrefix() + "dbversion WHERE name=E'" + Escape(dbVersionName) + "'");
- Dictionary::Ptr row = FetchRow(result, 0);
+ Dictionary::Ptr row = FetchRow(result, 0);
- if (!row) {
- PQfinish(m_Connection);
- SetConnected(false);
+ if (!row) {
+ PQfinish(m_Connection);
+ SetConnected(false);
- Log(LogCritical, "IdoPgsqlConnection", "Schema does not provide any valid version! Verify your schema installation.");
+ Log(LogCritical, "IdoPgsqlConnection", "Schema does not provide any valid version! Verify your schema installation.");
- Application::Exit(EXIT_FAILURE);
- }
+ Application::Exit(EXIT_FAILURE);
+ }
- String version = row->Get("version");
+ String version = row->Get("version");
- SetSchemaVersion(version);
+ SetSchemaVersion(version);
- if (Utility::CompareVersion(IDO_COMPAT_SCHEMA_VERSION, version) < 0) {
- PQfinish(m_Connection);
- SetConnected(false);
+ if (Utility::CompareVersion(IDO_COMPAT_SCHEMA_VERSION, version) < 0) {
+ PQfinish(m_Connection);
+ SetConnected(false);
- Log(LogCritical, "IdoPgsqlConnection")
- << "Schema version '" << version << "' does not match the required version '"
- << IDO_COMPAT_SCHEMA_VERSION << "' (or newer)! Please check the upgrade documentation.";
+ Log(LogCritical, "IdoPgsqlConnection")
+ << "Schema version '" << version << "' does not match the required version '"
+ << IDO_COMPAT_SCHEMA_VERSION << "' (or newer)! Please check the upgrade documentation.";
- Application::Exit(EXIT_FAILURE);
- }
+ Application::Exit(EXIT_FAILURE);
+ }
- String instanceName = GetInstanceName();
+ String instanceName = GetInstanceName();
- result = Query("SELECT instance_id FROM " + GetTablePrefix() + "instances WHERE instance_name = E'" + Escape(instanceName) + "'");
- row = FetchRow(result, 0);
+ result = Query("SELECT instance_id FROM " + GetTablePrefix() + "instances WHERE instance_name = E'" + Escape(instanceName) + "'");
+ row = FetchRow(result, 0);
- if (!row) {
- Query("INSERT INTO " + GetTablePrefix() + "instances (instance_name, instance_description) VALUES (E'" + Escape(instanceName) + "', E'" + Escape(GetInstanceDescription()) + "')");
- m_InstanceID = GetSequenceValue(GetTablePrefix() + "instances", "instance_id");
- } else {
- m_InstanceID = DbReference(row->Get("instance_id"));
- }
+ if (!row) {
+ Query("INSERT INTO " + GetTablePrefix() + "instances (instance_name, instance_description) VALUES (E'" + Escape(instanceName) + "', E'" + Escape(GetInstanceDescription()) + "')");
+ m_InstanceID = GetSequenceValue(GetTablePrefix() + "instances", "instance_id");
+ } else {
+ m_InstanceID = DbReference(row->Get("instance_id"));
+ }
- Endpoint::Ptr my_endpoint = Endpoint::GetLocalEndpoint();
+ Endpoint::Ptr my_endpoint = Endpoint::GetLocalEndpoint();
- /* we have an endpoint in a cluster setup, so decide if we can proceed here */
- if (my_endpoint && GetHAMode() == HARunOnce) {
- /* get the current endpoint writing to programstatus table */
- result = Query("SELECT UNIX_TIMESTAMP(status_update_time) AS status_update_time, endpoint_name FROM " +
- GetTablePrefix() + "programstatus WHERE instance_id = " + Convert::ToString(m_InstanceID));
- row = FetchRow(result, 0);
+ /* we have an endpoint in a cluster setup, so decide if we can proceed here */
+ if (my_endpoint && GetHAMode() == HARunOnce) {
+ /* get the current endpoint writing to programstatus table */
+ result = Query("SELECT UNIX_TIMESTAMP(status_update_time) AS status_update_time, endpoint_name FROM " +
+ GetTablePrefix() + "programstatus WHERE instance_id = " + Convert::ToString(m_InstanceID));
+ row = FetchRow(result, 0);
- String endpoint_name;
+ String endpoint_name;
- if (row)
- endpoint_name = row->Get("endpoint_name");
- else
- Log(LogNotice, "IdoPgsqlConnection", "Empty program status table");
+ if (row)
+ endpoint_name = row->Get("endpoint_name");
+ else
+ Log(LogNotice, "IdoPgsqlConnection", "Empty program status table");
- /* if we did not write into the database earlier, another instance is active */
- if (endpoint_name != my_endpoint->GetName()) {
- double status_update_time;
+ /* if we did not write into the database earlier, another instance is active */
+ if (endpoint_name != my_endpoint->GetName()) {
+ double status_update_time;
- if (row)
- status_update_time = row->Get("status_update_time");
- else
- status_update_time = 0;
+ if (row)
+ status_update_time = row->Get("status_update_time");
+ else
+ status_update_time = 0;
- double status_update_age = Utility::GetTime() - status_update_time;
+ double status_update_age = Utility::GetTime() - status_update_time;
- Log(LogNotice, "IdoPgsqlConnection")
- << "Last update by '" << endpoint_name << "' was " << status_update_age << "s ago.";
+ Log(LogNotice, "IdoPgsqlConnection")
+ << "Last update by '" << endpoint_name << "' was " << status_update_age << "s ago.";
- if (status_update_age < GetFailoverTimeout()) {
- PQfinish(m_Connection);
- SetConnected(false);
- SetShouldConnect(false);
+ if (status_update_age < GetFailoverTimeout()) {
+ PQfinish(m_Connection);
+ SetConnected(false);
+ SetShouldConnect(false);
- return;
- }
+ return;
+ }
- /* activate the IDO only, if we're authoritative in this zone */
- if (IsPaused()) {
- Log(LogNotice, "IdoPgsqlConnection")
- << "Local endpoint '" << my_endpoint->GetName() << "' is not authoritative, bailing out.";
+ /* activate the IDO only, if we're authoritative in this zone */
+ if (IsPaused()) {
+ Log(LogNotice, "IdoPgsqlConnection")
+ << "Local endpoint '" << my_endpoint->GetName() << "' is not authoritative, bailing out.";
- PQfinish(m_Connection);
- SetConnected(false);
+ PQfinish(m_Connection);
+ SetConnected(false);
- return;
- }
+ return;
}
-
- Log(LogNotice, "IdoPgsqlConnection", "Enabling IDO connection.");
}
- Log(LogInformation, "IdoPgsqlConnection")
- << "pgSQL IDO instance id: " << static_cast<long>(m_InstanceID) << " (schema version: '" + version + "')";
+ Log(LogNotice, "IdoPgsqlConnection", "Enabling IDO connection.");
+ }
- /* record connection */
- Query("INSERT INTO " + GetTablePrefix() + "conninfo " +
- "(instance_id, connect_time, last_checkin_time, agent_name, agent_version, connect_type, data_start_time) VALUES ("
- + Convert::ToString(static_cast<long>(m_InstanceID)) + ", NOW(), NOW(), E'icinga2 db_ido_pgsql', E'" + Escape(Application::GetAppVersion())
- + "', E'" + (reconnect ? "RECONNECT" : "INITIAL") + "', NOW())");
+ Log(LogInformation, "IdoPgsqlConnection")
+ << "pgSQL IDO instance id: " << static_cast<long>(m_InstanceID) << " (schema version: '" + version + "')";
- /* clear config tables for the initial config dump */
- PrepareDatabase();
+ /* record connection */
+ Query("INSERT INTO " + GetTablePrefix() + "conninfo " +
+ "(instance_id, connect_time, last_checkin_time, agent_name, agent_version, connect_type, data_start_time) VALUES ("
+ + Convert::ToString(static_cast<long>(m_InstanceID)) + ", NOW(), NOW(), E'icinga2 db_ido_pgsql', E'" + Escape(Application::GetAppVersion())
+ + "', E'" + (reconnect ? "RECONNECT" : "INITIAL") + "', NOW())");
- std::ostringstream q1buf;
- q1buf << "SELECT object_id, objecttype_id, name1, name2, is_active FROM " + GetTablePrefix() + "objects WHERE instance_id = " << static_cast<long>(m_InstanceID);
- result = Query(q1buf.str());
+ /* clear config tables for the initial config dump */
+ PrepareDatabase();
- int index = 0;
- while ((row = FetchRow(result, index))) {
- index++;
+ std::ostringstream q1buf;
+ q1buf << "SELECT object_id, objecttype_id, name1, name2, is_active FROM " + GetTablePrefix() + "objects WHERE instance_id = " << static_cast<long>(m_InstanceID);
+ result = Query(q1buf.str());
- DbType::Ptr dbtype = DbType::GetByID(row->Get("objecttype_id"));
+ std::vector<DbObject::Ptr> activeDbObjs;
- if (!dbtype)
- continue;
+ int index = 0;
+ while ((row = FetchRow(result, index))) {
+ index++;
- DbObject::Ptr dbobj = dbtype->GetOrCreateObjectByName(row->Get("name1"), row->Get("name2"));
- SetObjectID(dbobj, DbReference(row->Get("object_id")));
- SetObjectActive(dbobj, row->Get("is_active"));
+ DbType::Ptr dbtype = DbType::GetByID(row->Get("objecttype_id"));
- if (GetObjectActive(dbobj))
- active_dbobjs.push_back(dbobj);
- }
+ if (!dbtype)
+ continue;
- Query("BEGIN");
+ DbObject::Ptr dbobj = dbtype->GetOrCreateObjectByName(row->Get("name1"), row->Get("name2"));
+ SetObjectID(dbobj, DbReference(row->Get("object_id")));
+ bool active = row->Get("is_active");
+ SetObjectActive(dbobj, active);
+
+ if (active)
+ activeDbObjs.push_back(dbobj);
}
- UpdateAllObjects();
+ Query("BEGIN");
- /* deactivate all deleted configuration objects */
- BOOST_FOREACH(const DbObject::Ptr& dbobj, active_dbobjs) {
+ BOOST_FOREACH(const DbObject::Ptr& dbobj, activeDbObjs) {
if (dbobj->GetObject() == NULL) {
Log(LogNotice, "IdoPgsqlConnection")
<< "Deactivate deleted object name1: '" << dbobj->GetName1()
<< "' name2: '" << dbobj->GetName2() + "'.";
DeactivateObject(dbobj);
+ } else {
+ dbobj->SendConfigUpdate();
+ dbobj->SendStatusUpdate();
}
}
+ UpdateAllObjects();
+
/* delete all customvariables without current session token */
ClearCustomVarTable("customvariables");
ClearCustomVarTable("customvariablestatus");
projects / icinga2 / commitdiff