database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-11-10 17:42:29 +03:00
Code Activity

Add a message type header to the CopyData messages sent from primary

Browse Source 
to standby in streaming replication. While we only have one message type
at the moment, adding a message type header makes this easier to extend.
This commit is contained in:
Heikki Linnakangas
2010-02-03 09:47:19 +00:00
parent 47c5b8f558
commit 808969d0e7
5 changed files with 117 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
src/backend/replication/walreceiver.c
View File

@@ -29,7 +29,7 @@
*
*
* IDENTIFICATION
* $PostgreSQL: pgsql/src/backend/replication/walreceiver.c,v 1.2 2010/01/27 15:27:51 heikki Exp $
* $PostgreSQL: pgsql/src/backend/replication/walreceiver.c,v 1.3 2010/02/03 09:47:19 heikki Exp $
*
*-------------------------------------------------------------------------
*/
@@ -135,6 +135,7 @@ static void WalRcvQuickDieHandler(SIGNAL_ARGS);
/* Prototypes for private functions */
static void WalRcvDie(int code, Datum arg);
static void XLogWalRcvProcessMsg(unsigned char type, char *buf, Size len);
static void XLogWalRcvWrite(char *buf, Size nbytes, XLogRecPtr recptr);
static void XLogWalRcvFlush(void);
@@ -258,7 +259,7 @@ WalReceiverMain(void)
/* Loop until end-of-streaming or error */
for (;;)
{
XLogRecPtr recptr;
unsigned char type;
char *buf;
int len;
@@ -287,17 +288,17 @@ WalReceiverMain(void)
}
/* Wait a while for data to arrive */
if (walrcv_receive(NAPTIME_PER_CYCLE, &recptr, &buf, &len))
if (walrcv_receive(NAPTIME_PER_CYCLE, &type, &buf, &len))
{
/* Write received WAL records to disk */
XLogWalRcvWrite(buf, len, recptr);
/* Accept the received data, and process it */
XLogWalRcvProcessMsg(type, buf, len);
/* Receive any more WAL records we can without sleeping */
while(walrcv_receive(0, &recptr, &buf, &len))
XLogWalRcvWrite(buf, len, recptr);
/* Receive any more data we can without sleeping */
while(walrcv_receive(0, &type, &buf, &len))
XLogWalRcvProcessMsg(type, buf, len);
/*
* Now that we've written some records, flush them to disk and
* If we've written some records, flush them to disk and
* let the startup process know about them.
*/
XLogWalRcvFlush();
@@ -375,6 +376,36 @@ WalRcvQuickDieHandler(SIGNAL_ARGS)
exit(2);
}
/*
* Accept the message from XLOG stream, and process it.
*/
static void
XLogWalRcvProcessMsg(unsigned char type, char *buf, Size len)
{
switch (type)
{
case 'w': /* WAL records */
{
XLogRecPtr recptr;
if (len < sizeof(XLogRecPtr))
ereport(ERROR,
(errmsg("invalid WAL message received from primary")));
recptr = *((XLogRecPtr *) buf);
buf += sizeof(XLogRecPtr);
len -= sizeof(XLogRecPtr);
XLogWalRcvWrite(buf, len, recptr);
break;
}
default:
ereport(ERROR,
(errcode(ERRCODE_PROTOCOL_VIOLATION),
errmsg("invalid replication message type %d",
type)));
}
}
/*
* Write XLOG data to disk.
*/