added possibility to execute after_commands on master nodes

README.md

@@ -1,4 +1,4 @@
-pg_keeper 1.0
+pg_keeper 1.1
 ===========
 
 pg_keeper is a simplified clustering module for PostgreSQL, to promote a standby server to master in a 2 servers cluster.
@@ -51,10 +51,15 @@ Note that the parameters with (*) are mandatory options.
 
   - Specifies how many times pg_keeper try polling to master server in order to promote standby server. 4 times by default.
 
-- pg_keeper.after_command
+- pg_keeper.master_after_command
+
+  - Specifies shell command that will be called after connection to standby is lost.
+
+- pg_keeper.standby_after_command
 
   - Specifies shell command that will be called after promoted. Setting stonith command to this parameter is useful for preventing the split-brain syndrome.
 
+
 ## Tested platforms
 pg_keeper has been built and tested on following platforms:
 

master.c

@@ -199,6 +199,9 @@ KeeperMainMaster(void)
 				 */
 				updateStatus(KEEPER_MASTER_ASYNC);
 				standby_connected = false;
+                /* If after command is given, execute it */
+				if (pgkeeper_master_after_command)
+					doAfterCommand(pgkeeper_master_after_command);
 			}
 		}
 		/* nothing to do if in async mode */

pg_keeper.c

@@ -56,6 +56,8 @@ int	pgkeeper_keepalives_time;
 int	pgkeeper_keepalives_count;
 char *pgkeeper_partner_conninfo;
 char *pgkeeper_my_conninfo;
+char *pgkeeper_master_after_command;
+char *pgkeeper_standby_after_command;
 
 KeeperShmem	*keeperShmem;
 
@@ -122,10 +124,21 @@ _PG_init(void)
 							   NULL,
 							   NULL);
 
-	DefineCustomStringVariable("pg_keeper.after_command",
-							   "Shell command that will be called after promoted",
+	DefineCustomStringVariable("pg_keeper.master_after_command",
+							   "Shell command that will be called after master loses standby connection",
 							   NULL,
-							   &pgkeeper_after_command,
+							   &pgkeeper_master_after_command,
+							   NULL,
+							   PGC_SIGHUP,
+							   GUC_NOT_IN_SAMPLE,
+							   NULL,
+							   NULL,
+							   NULL);
+
+	DefineCustomStringVariable("pg_keeper.standby_after_command",
+							   "Shell command that will be called after standby is promoted",
+							   NULL,
+							   &pgkeeper_standby_after_command,
 							   NULL,
 							   PGC_SIGHUP,
 							   GUC_NOT_IN_SAMPLE,
@@ -396,3 +409,27 @@ updateStatus(KeeperStatus status)
 	/* Then, update process title */
 	set_ps_display(getStatusPsString(status), false);
 }
+
+/*
+ * Attempt to execute an external shell command after promotion.
+ */
+void
+doAfterCommand(const char *command)
+{
+	int	rc;
+
+	Assert(command);
+
+	ereport(LOG,
+			(errmsg("executing after command \"%s\"",
+					command)));
+
+	rc = system(command);
+
+	if (rc != 0)
+	{
+		ereport(LOG,
+				(errmsg("failed to execute after command \"%s\"",
+						command)));
+	}
+}

pg_keeper.h

@@ -58,6 +58,7 @@ sig_atomic_t got_sighup;
 sig_atomic_t got_sigterm;
 
 extern void updateStatus(KeeperStatus status);
+extern void doAfterCommand(const char *command);
 
 /* master.c */
 extern bool KeeperMainMaster(void);
@@ -72,4 +73,5 @@ extern int	pgkeeper_keepalives_time;
 extern int	pgkeeper_keepalives_count;
 extern char *pgkeeper_partner_conninfo;
 extern char *pgkeeper_my_conninfo;
-extern char *pgkeeper_after_command;
+extern char *pgkeeper_standby_after_command;
+extern char *pgkeeper_master_after_command;

standby.c

@@ -34,10 +34,6 @@ bool	KeeperMainStandby(void);
 void	setupKeeperStandby(void);
 
 static void doPromote(void);
-static void doAfterCommand(void);
-
-/* GUC variables */
-char	*pgkeeper_after_command;
 
 /* Variables for heartbeat */
 static int retry_count;
@@ -130,8 +126,8 @@ KeeperMainStandby(void)
 			doPromote();
 
 			/* If after command is given, execute it */
-			if (pgkeeper_after_command)
-				doAfterCommand();
+			if (pgkeeper_standby_after_command)
+				doAfterCommand(pgkeeper_standby_after_command);
 
 			return true;
 		}
@@ -169,28 +165,3 @@ doPromote(void)
 	ereport(LOG,
 			(errmsg("pg_keeper promoted standby server to primary server")));
 }
-
-/*
- * Attempt to execute an external shell command after promotion.
- */
-static void
-doAfterCommand(void)
-{
-	int	rc;
-
-	Assert(pgkeeper_after_command);
-
-	ereport(LOG,
-			(errmsg("executing after promoting command \"%s\"",
-					pgkeeper_after_command)));
-
-	rc = system(pgkeeper_after_command);
-
-	if (rc != 0)
-	{
-		ereport(LOG,
-				(errmsg("failed to execute after promoting command \"%s\"",
-						pgkeeper_after_command)));
-	}
-}
-