mirror of
https://github.com/EnterpriseDB/repmgr.git
synced 2026-03-22 22:56:29 +00:00
Add event logging
This commit is contained in:
Ian Barwick
269
dbutils.c
269
dbutils.c
@@ -969,7 +969,274 @@ _create_update_node_record(PGconn *conn, char *action, t_node_info *node_info)
|
|||||||
}
|
}
|
||||||
|
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
log_info("ok");
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* ====================== */
|
||||||
|
/* event record functions */
|
||||||
|
/* ====================== */
|
||||||
|
|
||||||
|
/*
|
||||||
|
* create_event_record()
|
||||||
|
*
|
||||||
|
* If `conn` is not NULL, insert a record into the events table.
|
||||||
|
*
|
||||||
|
* If configuration parameter `event_notification_command` is set, also
|
||||||
|
* attempt to execute that command.
|
||||||
|
*
|
||||||
|
* Returns true if all operations succeeded, false if one or more failed.
|
||||||
|
*
|
||||||
|
* Note this function may be called with `conn` set to NULL in cases where
|
||||||
|
* the master node is not available and it's therefore not possible to write
|
||||||
|
* an event record. In this case, if `event_notification_command` is set, a
|
||||||
|
* user-defined notification to be generated; if not, this function will have
|
||||||
|
* no effect.
|
||||||
|
*/
|
||||||
|
bool
|
||||||
|
create_event_record(PGconn *conn, t_configuration_options *options, int node_id, char *event, bool successful, char *details)
|
||||||
|
{
|
||||||
|
t_event_info event_info = T_EVENT_INFO_INITIALIZER;
|
||||||
|
|
||||||
|
return _create_event_record(conn, options, node_id, event, successful, details, &event_info);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* create_event_record_extended()
|
||||||
|
*
|
||||||
|
* The caller may need to pass additional parameters to the event notification
|
||||||
|
* command (currently only the conninfo string of another node)
|
||||||
|
|
||||||
|
*/
|
||||||
|
bool
|
||||||
|
create_event_record_extended(PGconn *conn, t_configuration_options *options, int node_id, char *event, bool successful, char *details, t_event_info *event_info)
|
||||||
|
{
|
||||||
|
return _create_event_record(conn, options, node_id, event, successful, details, event_info);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool
|
||||||
|
_create_event_record(PGconn *conn, t_configuration_options *options, int node_id, char *event, bool successful, char *details, t_event_info *event_info)
|
||||||
|
{
|
||||||
|
PQExpBufferData query;
|
||||||
|
PGresult *res;
|
||||||
|
char event_timestamp[MAXLEN] = "";
|
||||||
|
bool success = true;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Only attempt to write a record if a connection handle was provided.
|
||||||
|
* Also check that the repmgr schema has been properly initialised - if
|
||||||
|
* not it means no configuration file was provided, which can happen with
|
||||||
|
* e.g. `repmgr standby clone`, and we won't know which schema to write to.
|
||||||
|
*/
|
||||||
|
if (conn != NULL && PQstatus(conn) == CONNECTION_OK)
|
||||||
|
{
|
||||||
|
int n_node_id = htonl(node_id);
|
||||||
|
char *t_successful = successful ? "TRUE" : "FALSE";
|
||||||
|
|
||||||
|
const char *values[4] = { (char *)&n_node_id,
|
||||||
|
event,
|
||||||
|
t_successful,
|
||||||
|
details
|
||||||
|
};
|
||||||
|
|
||||||
|
int lengths[4] = { sizeof(n_node_id),
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
0
|
||||||
|
};
|
||||||
|
|
||||||
|
int binary[4] = {1, 0, 0, 0};
|
||||||
|
|
||||||
|
initPQExpBuffer(&query);
|
||||||
|
appendPQExpBuffer(&query,
|
||||||
|
" INSERT INTO repmgr.events ( "
|
||||||
|
" node_id, "
|
||||||
|
" event, "
|
||||||
|
" successful, "
|
||||||
|
" details "
|
||||||
|
" ) "
|
||||||
|
" VALUES ($1, $2, $3, $4) "
|
||||||
|
" RETURNING event_timestamp ");
|
||||||
|
|
||||||
|
log_verbose(LOG_DEBUG, "create_event_record():\n %s", query.data);
|
||||||
|
|
||||||
|
res = PQexecParams(conn,
|
||||||
|
query.data,
|
||||||
|
4,
|
||||||
|
NULL,
|
||||||
|
values,
|
||||||
|
lengths,
|
||||||
|
binary,
|
||||||
|
0);
|
||||||
|
|
||||||
|
termPQExpBuffer(&query);
|
||||||
|
|
||||||
|
if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
|
||||||
|
{
|
||||||
|
/* we don't treat this as an error */
|
||||||
|
log_warning(_("unable to create event record:\n %s"),
|
||||||
|
PQerrorMessage(conn));
|
||||||
|
|
||||||
|
success = false;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
/* Store timestamp to send to the notification command */
|
||||||
|
strncpy(event_timestamp, PQgetvalue(res, 0, 0), MAXLEN);
|
||||||
|
}
|
||||||
|
|
||||||
|
PQclear(res);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* If no database connection provided, or the query failed, generate a
|
||||||
|
* current timestamp ourselves. This isn't quite the same
|
||||||
|
* format as PostgreSQL, but is close enough for diagnostic use.
|
||||||
|
*/
|
||||||
|
if (!strlen(event_timestamp))
|
||||||
|
{
|
||||||
|
time_t now;
|
||||||
|
struct tm ts;
|
||||||
|
|
||||||
|
time(&now);
|
||||||
|
ts = *localtime(&now);
|
||||||
|
strftime(event_timestamp, MAXLEN, "%Y-%m-%d %H:%M:%S%z", &ts);
|
||||||
|
}
|
||||||
|
|
||||||
|
log_verbose(LOG_DEBUG, "create_event_record(): Event timestamp is \"%s\"\n", event_timestamp);
|
||||||
|
|
||||||
|
/* an event notification command was provided - parse and execute it */
|
||||||
|
if (strlen(options->event_notification_command))
|
||||||
|
{
|
||||||
|
char parsed_command[MAXPGPATH];
|
||||||
|
const char *src_ptr;
|
||||||
|
char *dst_ptr;
|
||||||
|
char *end_ptr;
|
||||||
|
int r;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* If configuration option 'event_notifications' was provided,
|
||||||
|
* check if this event is one of the ones listed; if not listed,
|
||||||
|
* don't execute the notification script.
|
||||||
|
*
|
||||||
|
* (If 'event_notifications' was not provided, we assume the script
|
||||||
|
* should be executed for all events).
|
||||||
|
*/
|
||||||
|
if (options->event_notifications.head != NULL)
|
||||||
|
{
|
||||||
|
EventNotificationListCell *cell;
|
||||||
|
bool notify_ok = false;
|
||||||
|
|
||||||
|
for (cell = options->event_notifications.head; cell; cell = cell->next)
|
||||||
|
{
|
||||||
|
if (strcmp(event, cell->event_type) == 0)
|
||||||
|
{
|
||||||
|
notify_ok = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Event type not found in the 'event_notifications' list - return early
|
||||||
|
*/
|
||||||
|
if (notify_ok == false)
|
||||||
|
{
|
||||||
|
log_debug(_("Not executing notification script for event type '%s'\n"), event);
|
||||||
|
return success;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
dst_ptr = parsed_command;
|
||||||
|
end_ptr = parsed_command + MAXPGPATH - 1;
|
||||||
|
*end_ptr = '\0';
|
||||||
|
|
||||||
|
for(src_ptr = options->event_notification_command; *src_ptr; src_ptr++)
|
||||||
|
{
|
||||||
|
if (*src_ptr == '%')
|
||||||
|
{
|
||||||
|
switch (src_ptr[1])
|
||||||
|
{
|
||||||
|
case 'n':
|
||||||
|
/* %n: node id */
|
||||||
|
src_ptr++;
|
||||||
|
snprintf(dst_ptr, end_ptr - dst_ptr, "%i", node_id);
|
||||||
|
dst_ptr += strlen(dst_ptr);
|
||||||
|
break;
|
||||||
|
case 'a':
|
||||||
|
/* %a: node name */
|
||||||
|
src_ptr++;
|
||||||
|
if (event_info->node_name != NULL)
|
||||||
|
{
|
||||||
|
log_debug("node_name: %s\n", event_info->node_name);
|
||||||
|
strlcpy(dst_ptr, event_info->node_name, end_ptr - dst_ptr);
|
||||||
|
dst_ptr += strlen(dst_ptr);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 'e':
|
||||||
|
/* %e: event type */
|
||||||
|
src_ptr++;
|
||||||
|
strlcpy(dst_ptr, event, end_ptr - dst_ptr);
|
||||||
|
dst_ptr += strlen(dst_ptr);
|
||||||
|
break;
|
||||||
|
case 'd':
|
||||||
|
/* %d: details */
|
||||||
|
src_ptr++;
|
||||||
|
if (details != NULL)
|
||||||
|
{
|
||||||
|
strlcpy(dst_ptr, details, end_ptr - dst_ptr);
|
||||||
|
dst_ptr += strlen(dst_ptr);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 's':
|
||||||
|
/* %s: successful */
|
||||||
|
src_ptr++;
|
||||||
|
strlcpy(dst_ptr, successful ? "1" : "0", end_ptr - dst_ptr);
|
||||||
|
dst_ptr += strlen(dst_ptr);
|
||||||
|
break;
|
||||||
|
case 't':
|
||||||
|
/* %t: timestamp */
|
||||||
|
src_ptr++;
|
||||||
|
strlcpy(dst_ptr, event_timestamp, end_ptr - dst_ptr);
|
||||||
|
dst_ptr += strlen(dst_ptr);
|
||||||
|
break;
|
||||||
|
case 'c':
|
||||||
|
/* %c: conninfo for next available node */
|
||||||
|
src_ptr++;
|
||||||
|
if (event_info->conninfo_str != NULL)
|
||||||
|
{
|
||||||
|
log_debug("conninfo: %s\n", event_info->conninfo_str);
|
||||||
|
|
||||||
|
strlcpy(dst_ptr, event_info->conninfo_str, end_ptr - dst_ptr);
|
||||||
|
dst_ptr += strlen(dst_ptr);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
/* otherwise treat the % as not special */
|
||||||
|
if (dst_ptr < end_ptr)
|
||||||
|
*dst_ptr++ = *src_ptr;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (dst_ptr < end_ptr)
|
||||||
|
*dst_ptr++ = *src_ptr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
*dst_ptr = '\0';
|
||||||
|
|
||||||
|
log_debug("create_event_record(): executing\n%s", parsed_command);
|
||||||
|
|
||||||
|
r = system(parsed_command);
|
||||||
|
if (r != 0)
|
||||||
|
{
|
||||||
|
log_warning(_("unable to execute event notification command"));
|
||||||
|
log_info(_("parsed event notification command was:\n %s"), parsed_command);
|
||||||
|
success = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return success;
|
||||||
|
}
|
||||||
|
|||||||
@@ -156,5 +156,10 @@ int get_node_record(PGconn *conn, int node_id, t_node_info *node_info);
|
|||||||
bool create_node_record(PGconn *conn, char *repmgr_action, t_node_info *node_info);
|
bool create_node_record(PGconn *conn, char *repmgr_action, t_node_info *node_info);
|
||||||
bool update_node_record(PGconn *conn, char *repmgr_action, t_node_info *node_info);
|
bool update_node_record(PGconn *conn, char *repmgr_action, t_node_info *node_info);
|
||||||
|
|
||||||
|
/* event record functions */
|
||||||
|
bool create_event_record(PGconn *conn, t_configuration_options *options, int node_id, char *event, bool successful, char *details);
|
||||||
|
bool create_event_record_extended(PGconn *conn, t_configuration_options *options, int node_id, char *event, bool successful, char *details, t_event_info *event_info);
|
||||||
|
bool _create_event_record(PGconn *conn, t_configuration_options *options, int node_id, char *event, bool successful, char *details, t_event_info *event_info);
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|||||||
@@ -555,7 +555,31 @@ do_master_register(void)
|
|||||||
"master register",
|
"master register",
|
||||||
&node_info);
|
&node_info);
|
||||||
|
|
||||||
|
/* Log the event */
|
||||||
|
create_event_record(conn,
|
||||||
|
&config_file_options,
|
||||||
|
config_file_options.node_id,
|
||||||
|
"master_register",
|
||||||
|
record_created,
|
||||||
|
NULL);
|
||||||
|
|
||||||
|
if (record_created == false)
|
||||||
|
{
|
||||||
|
rollback_transaction(conn);
|
||||||
|
PQfinish(conn);
|
||||||
|
|
||||||
|
log_notice(_("unable to register master node - see preceding messages"));
|
||||||
|
exit(ERR_DB_QUERY);
|
||||||
|
}
|
||||||
|
|
||||||
commit_transaction(conn);
|
commit_transaction(conn);
|
||||||
|
PQfinish(conn);
|
||||||
|
|
||||||
|
log_notice(_("master node %s with id %i (conninfo: %s)"),
|
||||||
|
record_found ? "updated" : "registered",
|
||||||
|
config_file_options.node_id,
|
||||||
|
config_file_options.conninfo);
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user