github/repmgr
1
0
Fork 0
mirror of https://github.com/EnterpriseDB/repmgr.git synced 2026-03-27 00:46:29 +00:00
Code Issues Packages Projects Releases 1 Wiki Activity

poke it around until it works less badly

Browse Source
This commit is contained in:
Ian Barwick
2017-06-29 09:35:09 +09:00
parent fa86fe4ad8
commit 3514e20367
7 changed files with 81 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
repmgr.c
View File

@@ -30,6 +30,8 @@
#include "voting.h"
#define UNKNOWN_NODE_ID -1
#define MAXFNAMELEN 64
#define TRANCHE_NAME "repmgrd"
@@ -46,6 +48,7 @@ typedef struct repmgrdSharedState
LWLockId lock; /* protects search/modification */
NodeState node_state;
NodeVotingStatus voting_status;
int candidate_node_id;
} repmgrdSharedState;
static repmgrdSharedState *shared_state = NULL;
@@ -67,6 +70,8 @@ PG_FUNCTION_INFO_V1(get_voting_status);
Datum set_voting_status_initiated(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(set_voting_status_initiated);
Datum other_node_is_candidate(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(other_node_is_candidate);
/*
* Module load callback
*/
@@ -130,7 +135,7 @@ repmgr_shmem_startup(void)
if (!found)
{
/* First time through ... */
/* Initialise shared memory struct */
#if (PG_VERSION_NUM >= 90600)
shared_state->lock = &(GetNamedLWLockTranche(TRANCHE_NAME))->lock;
#else
@@ -138,6 +143,7 @@ repmgr_shmem_startup(void)
#endif
shared_state->voting_status = VS_NO_VOTE;
shared_state->candidate_node_id = UNKNOWN_NODE_ID;
}
LWLockRelease(AddinShmemInitLock);
@@ -225,3 +231,25 @@ set_voting_status_initiated(PG_FUNCTION_ARGS)
PG_RETURN_VOID();
}
Datum
other_node_is_candidate(PG_FUNCTION_ARGS)
{
int requesting_node_id = PG_GETARG_INT32(0);
LWLockAcquire(shared_state->lock, LW_SHARED);
if (shared_state->candidate_node_id != UNKNOWN_NODE_ID)
{
elog(INFO, "node %i requesting candidature, but node %i already candidate",
requesting_node_id,
shared_state->candidate_node_id);
PG_RETURN_BOOL(false);
}
shared_state->candidate_node_id = requesting_node_id;
LWLockRelease(shared_state->lock);
elog(INFO, "node %i is candidate", requesting_node_id);
PG_RETURN_BOOL(true);
}