PostgreSQL Source Code  git master
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros
origin.c File Reference
#include "postgres.h"
#include <unistd.h>
#include <sys/stat.h>
#include "funcapi.h"
#include "miscadmin.h"
#include "access/genam.h"
#include "access/heapam.h"
#include "access/htup_details.h"
#include "access/xact.h"
#include "catalog/indexing.h"
#include "nodes/execnodes.h"
#include "replication/origin.h"
#include "replication/logical.h"
#include "storage/fd.h"
#include "storage/ipc.h"
#include "storage/lmgr.h"
#include "storage/copydir.h"
#include "utils/builtins.h"
#include "utils/fmgroids.h"
#include "utils/pg_lsn.h"
#include "utils/rel.h"
#include "utils/syscache.h"
#include "utils/tqual.h"
Include dependency graph for origin.c:

Go to the source code of this file.

Data Structures

struct  ReplicationState
 
struct  ReplicationStateOnDisk
 
struct  ReplicationStateCtl
 

Macros

#define REPLICATION_STATE_MAGIC   ((uint32) 0x1257DADE)
 
#define REPLICATION_ORIGIN_PROGRESS_COLS   4
 

Typedefs

typedef struct ReplicationState ReplicationState
 
typedef struct
ReplicationStateOnDisk 
ReplicationStateOnDisk
 
typedef struct ReplicationStateCtl ReplicationStateCtl
 

Functions

static void replorigin_check_prerequisites (bool check_slots, bool recoveryOK)
 
RepOriginId replorigin_by_name (char *roname, bool missing_ok)
 
RepOriginId replorigin_create (char *roname)
 
void replorigin_drop (RepOriginId roident)
 
bool replorigin_by_oid (RepOriginId roident, bool missing_ok, char **roname)
 
Size ReplicationOriginShmemSize (void)
 
void ReplicationOriginShmemInit (void)
 
void CheckPointReplicationOrigin (void)
 
void StartupReplicationOrigin (void)
 
void replorigin_redo (XLogReaderState *record)
 
void replorigin_advance (RepOriginId node, XLogRecPtr remote_commit, XLogRecPtr local_commit, bool go_backward, bool wal_log)
 
XLogRecPtr replorigin_get_progress (RepOriginId node, bool flush)
 
static void ReplicationOriginExitCleanup (int code, Datum arg)
 
void replorigin_session_setup (RepOriginId node)
 
void replorigin_session_reset (void)
 
void replorigin_session_advance (XLogRecPtr remote_commit, XLogRecPtr local_commit)
 
XLogRecPtr replorigin_session_get_progress (bool flush)
 
Datum pg_replication_origin_create (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_drop (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_oid (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_session_setup (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_session_reset (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_session_is_setup (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_session_progress (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_xact_setup (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_xact_reset (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_advance (PG_FUNCTION_ARGS)
 
Datum pg_replication_origin_progress (PG_FUNCTION_ARGS)
 
Datum pg_show_replication_origin_status (PG_FUNCTION_ARGS)
 

Variables

RepOriginId replorigin_session_origin = InvalidRepOriginId
 
XLogRecPtr replorigin_session_origin_lsn = InvalidXLogRecPtr
 
TimestampTz replorigin_session_origin_timestamp = 0
 
static ReplicationStatereplication_states
 
static ReplicationStateCtlreplication_states_ctl
 
static ReplicationStatesession_replication_state = NULL
 

Macro Definition Documentation

#define REPLICATION_ORIGIN_PROGRESS_COLS   4
#define REPLICATION_STATE_MAGIC   ((uint32) 0x1257DADE)

Definition at line 172 of file origin.c.

Referenced by CheckPointReplicationOrigin(), and StartupReplicationOrigin().

Typedef Documentation

Function Documentation

void CheckPointReplicationOrigin ( void  )

Definition at line 504 of file origin.c.

References CloseTransientFile(), COMP_CRC32C, durable_rename(), ereport, errcode_for_file_access(), errmsg(), FIN_CRC32C, i, INIT_CRC32C, InvalidRepOriginId, ReplicationState::local_lsn, ReplicationState::lock, LW_SHARED, LWLockAcquire(), LWLockRelease(), max_replication_slots, OpenTransientFile(), PANIC, PG_BINARY, ReplicationState::remote_lsn, ReplicationStateOnDisk::remote_lsn, REPLICATION_STATE_MAGIC, ReplicationState::roident, ReplicationStateOnDisk::roident, unlink(), write, and XLogFlush().

Referenced by CheckPointGuts().

505 {
506  const char *tmppath = "pg_logical/replorigin_checkpoint.tmp";
507  const char *path = "pg_logical/replorigin_checkpoint";
508  int tmpfd;
509  int i;
511  pg_crc32c crc;
512 
513  if (max_replication_slots == 0)
514  return;
515 
516  INIT_CRC32C(crc);
517 
518  /* make sure no old temp file is remaining */
519  if (unlink(tmppath) < 0 && errno != ENOENT)
520  ereport(PANIC,
522  errmsg("could not remove file \"%s\": %m",
523  tmppath)));
524 
525  /*
526  * no other backend can perform this at the same time, we're protected by
527  * CheckpointLock.
528  */
529  tmpfd = OpenTransientFile((char *) tmppath,
530  O_CREAT | O_EXCL | O_WRONLY | PG_BINARY,
531  S_IRUSR | S_IWUSR);
532  if (tmpfd < 0)
533  ereport(PANIC,
535  errmsg("could not create file \"%s\": %m",
536  tmppath)));
537 
538  /* write magic */
539  if ((write(tmpfd, &magic, sizeof(magic))) != sizeof(magic))
540  {
541  CloseTransientFile(tmpfd);
542  ereport(PANIC,
544  errmsg("could not write to file \"%s\": %m",
545  tmppath)));
546  }
547  COMP_CRC32C(crc, &magic, sizeof(magic));
548 
549  /* prevent concurrent creations/drops */
550  LWLockAcquire(ReplicationOriginLock, LW_SHARED);
551 
552  /* write actual data */
553  for (i = 0; i < max_replication_slots; i++)
554  {
555  ReplicationStateOnDisk disk_state;
556  ReplicationState *curstate = &replication_states[i];
557  XLogRecPtr local_lsn;
558 
559  if (curstate->roident == InvalidRepOriginId)
560  continue;
561 
562  LWLockAcquire(&curstate->lock, LW_SHARED);
563 
564  disk_state.roident = curstate->roident;
565 
566  disk_state.remote_lsn = curstate->remote_lsn;
567  local_lsn = curstate->local_lsn;
568 
569  LWLockRelease(&curstate->lock);
570 
571  /* make sure we only write out a commit that's persistent */
572  XLogFlush(local_lsn);
573 
574  if ((write(tmpfd, &disk_state, sizeof(disk_state))) !=
575  sizeof(disk_state))
576  {
577  CloseTransientFile(tmpfd);
578  ereport(PANIC,
580  errmsg("could not write to file \"%s\": %m",
581  tmppath)));
582  }
583 
584  COMP_CRC32C(crc, &disk_state, sizeof(disk_state));
585  }
586 
587  LWLockRelease(ReplicationOriginLock);
588 
589  /* write out the CRC */
590  FIN_CRC32C(crc);
591  if ((write(tmpfd, &crc, sizeof(crc))) != sizeof(crc))
592  {
593  CloseTransientFile(tmpfd);
594  ereport(PANIC,
596  errmsg("could not write to file \"%s\": %m",
597  tmppath)));
598  }
599 
600  CloseTransientFile(tmpfd);
601 
602  /* fsync, rename to permanent file, fsync file and directory */
603  durable_rename(tmppath, path, PANIC);
604 }
#define INIT_CRC32C(crc)
Definition: pg_crc32c.h:41
XLogRecPtr local_lsn
Definition: origin.c:120
#define write(a, b, c)
Definition: win32.h:19
XLogRecPtr remote_lsn
Definition: origin.c:139
uint32 pg_crc32c
Definition: pg_crc32c.h:38
RepOriginId roident
Definition: origin.c:138
#define PANIC
Definition: elog.h:53
void XLogFlush(XLogRecPtr record)
Definition: xlog.c:2754
#define PG_BINARY
Definition: c.h:1038
RepOriginId roident
Definition: origin.c:108
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
LWLock lock
Definition: origin.c:130
int OpenTransientFile(FileName fileName, int fileFlags, int fileMode)
Definition: fd.c:2144
int errcode_for_file_access(void)
Definition: elog.c:598
unsigned int uint32
Definition: c.h:268
int unlink(const char *filename)
#define ereport(elevel, rest)
Definition: elog.h:122
int durable_rename(const char *oldfile, const char *newfile, int elevel)
Definition: fd.c:593
int CloseTransientFile(int fd)
Definition: fd.c:2305
#define REPLICATION_STATE_MAGIC
Definition: origin.c:172
int max_replication_slots
Definition: slot.c:99
XLogRecPtr remote_lsn
Definition: origin.c:113
uint64 XLogRecPtr
Definition: xlogdefs.h:21
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
#define InvalidRepOriginId
Definition: origin.h:34
int errmsg(const char *fmt,...)
Definition: elog.c:797
int i
static ReplicationState * replication_states
Definition: origin.c:161
#define COMP_CRC32C(crc, data, len)
Definition: pg_crc32c.h:73
#define FIN_CRC32C(crc)
Definition: pg_crc32c.h:78
Datum pg_replication_origin_advance ( PG_FUNCTION_ARGS  )

Definition at line 1308 of file origin.c.

References InvalidXLogRecPtr, LockRelationOid(), name, PG_GETARG_LSN, PG_GETARG_TEXT_PP, PG_RETURN_VOID, ReplicationOriginRelationId, replorigin_advance(), replorigin_by_name(), replorigin_check_prerequisites(), RowExclusiveLock, text_to_cstring(), and UnlockRelationOid().

1309 {
1310  text *name = PG_GETARG_TEXT_PP(0);
1311  XLogRecPtr remote_commit = PG_GETARG_LSN(1);
1312  RepOriginId node;
1313 
1314  replorigin_check_prerequisites(true, false);
1315 
1316  /* lock to prevent the replication origin from vanishing */
1318 
1319  node = replorigin_by_name(text_to_cstring(name), false);
1320 
1321  /*
1322  * Can't sensibly pass a local commit to be flushed at checkpoint - this
1323  * xact hasn't committed yet. This is why this function should be used to
1324  * set up the initial replication state, but not for replay.
1325  */
1326  replorigin_advance(node, remote_commit, InvalidXLogRecPtr,
1327  true /* go backward */ , true /* wal log */ );
1328 
1330 
1331  PG_RETURN_VOID();
1332 }
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
void UnlockRelationOid(Oid relid, LOCKMODE lockmode)
Definition: lmgr.c:182
uint16 RepOriginId
Definition: xlogdefs.h:51
void replorigin_advance(RepOriginId node, XLogRecPtr remote_commit, XLogRecPtr local_commit, bool go_backward, bool wal_log)
Definition: origin.c:791
RepOriginId replorigin_by_name(char *roname, bool missing_ok)
Definition: origin.c:206
#define PG_GETARG_TEXT_PP(n)
Definition: fmgr.h:273
#define RowExclusiveLock
Definition: lockdefs.h:38
#define PG_GETARG_LSN(n)
Definition: pg_lsn.h:24
#define PG_RETURN_VOID()
Definition: fmgr.h:309
uint64 XLogRecPtr
Definition: xlogdefs.h:21
const char * name
Definition: encode.c:521
char * text_to_cstring(const text *t)
Definition: varlena.c:182
#define ReplicationOriginRelationId
Definition: c.h:439
void LockRelationOid(Oid relid, LOCKMODE lockmode)
Definition: lmgr.c:105
Datum pg_replication_origin_create ( PG_FUNCTION_ARGS  )

Definition at line 1139 of file origin.c.

References DatumGetPointer, name, pfree(), PG_GETARG_DATUM, PG_RETURN_OID, replorigin_check_prerequisites(), replorigin_create(), and text_to_cstring().

1140 {
1141  char *name;
1142  RepOriginId roident;
1143 
1144  replorigin_check_prerequisites(false, false);
1145 
1147  roident = replorigin_create(name);
1148 
1149  pfree(name);
1150 
1151  PG_RETURN_OID(roident);
1152 }
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:233
uint16 RepOriginId
Definition: xlogdefs.h:51
void pfree(void *pointer)
Definition: mcxt.c:950
RepOriginId replorigin_create(char *roname)
Definition: origin.c:235
const char * name
Definition: encode.c:521
#define DatumGetPointer(X)
Definition: postgres.h:555
char * text_to_cstring(const text *t)
Definition: varlena.c:182
Definition: c.h:439
#define PG_RETURN_OID(x)
Definition: fmgr.h:320
Datum pg_replication_origin_drop ( PG_FUNCTION_ARGS  )

Definition at line 1158 of file origin.c.

References Assert, DatumGetPointer, name, OidIsValid, pfree(), PG_GETARG_DATUM, PG_RETURN_VOID, replorigin_by_name(), replorigin_check_prerequisites(), replorigin_drop(), and text_to_cstring().

1159 {
1160  char *name;
1161  RepOriginId roident;
1162 
1163  replorigin_check_prerequisites(false, false);
1164 
1166 
1167  roident = replorigin_by_name(name, false);
1168  Assert(OidIsValid(roident));
1169 
1170  replorigin_drop(roident);
1171 
1172  pfree(name);
1173 
1174  PG_RETURN_VOID();
1175 }
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:233
uint16 RepOriginId
Definition: xlogdefs.h:51
#define OidIsValid(objectId)
Definition: c.h:538
RepOriginId replorigin_by_name(char *roname, bool missing_ok)
Definition: origin.c:206
void pfree(void *pointer)
Definition: mcxt.c:950
void replorigin_drop(RepOriginId roident)
Definition: origin.c:327
#define PG_RETURN_VOID()
Definition: fmgr.h:309
#define Assert(condition)
Definition: c.h:675
const char * name
Definition: encode.c:521
#define DatumGetPointer(X)
Definition: postgres.h:555
char * text_to_cstring(const text *t)
Definition: varlena.c:182
Definition: c.h:439
Datum pg_replication_origin_oid ( PG_FUNCTION_ARGS  )

Definition at line 1181 of file origin.c.

References DatumGetPointer, name, OidIsValid, pfree(), PG_GETARG_DATUM, PG_RETURN_NULL, PG_RETURN_OID, replorigin_by_name(), replorigin_check_prerequisites(), and text_to_cstring().

1182 {
1183  char *name;
1184  RepOriginId roident;
1185 
1186  replorigin_check_prerequisites(false, false);
1187 
1189  roident = replorigin_by_name(name, true);
1190 
1191  pfree(name);
1192 
1193  if (OidIsValid(roident))
1194  PG_RETURN_OID(roident);
1195  PG_RETURN_NULL();
1196 }
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:233
uint16 RepOriginId
Definition: xlogdefs.h:51
#define OidIsValid(objectId)
Definition: c.h:538
RepOriginId replorigin_by_name(char *roname, bool missing_ok)
Definition: origin.c:206
void pfree(void *pointer)
Definition: mcxt.c:950
const char * name
Definition: encode.c:521
#define DatumGetPointer(X)
Definition: postgres.h:555
char * text_to_cstring(const text *t)
Definition: varlena.c:182
Definition: c.h:439
#define PG_RETURN_OID(x)
Definition: fmgr.h:320
#define PG_RETURN_NULL()
Definition: fmgr.h:305
Datum pg_replication_origin_progress ( PG_FUNCTION_ARGS  )

Definition at line 1343 of file origin.c.

References Assert, DatumGetPointer, InvalidXLogRecPtr, name, OidIsValid, PG_GETARG_BOOL, PG_GETARG_DATUM, PG_RETURN_LSN, PG_RETURN_NULL, replorigin_by_name(), replorigin_check_prerequisites(), replorigin_get_progress(), and text_to_cstring().

1344 {
1345  char *name;
1346  bool flush;
1347  RepOriginId roident;
1348  XLogRecPtr remote_lsn = InvalidXLogRecPtr;
1349 
1350  replorigin_check_prerequisites(true, true);
1351 
1353  flush = PG_GETARG_BOOL(1);
1354 
1355  roident = replorigin_by_name(name, false);
1356  Assert(OidIsValid(roident));
1357 
1358  remote_lsn = replorigin_get_progress(roident, flush);
1359 
1360  if (remote_lsn == InvalidXLogRecPtr)
1361  PG_RETURN_NULL();
1362 
1363  PG_RETURN_LSN(remote_lsn);
1364 }
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:233
uint16 RepOriginId
Definition: xlogdefs.h:51
#define PG_GETARG_BOOL(n)
Definition: fmgr.h:239
#define OidIsValid(objectId)
Definition: c.h:538
RepOriginId replorigin_by_name(char *roname, bool missing_ok)
Definition: origin.c:206
#define PG_RETURN_LSN(x)
Definition: pg_lsn.h:25
uint64 XLogRecPtr
Definition: xlogdefs.h:21
#define Assert(condition)
Definition: c.h:675
const char * name
Definition: encode.c:521
#define DatumGetPointer(X)
Definition: postgres.h:555
char * text_to_cstring(const text *t)
Definition: varlena.c:182
Definition: c.h:439
XLogRecPtr replorigin_get_progress(RepOriginId node, bool flush)
Definition: origin.c:914
#define PG_RETURN_NULL()
Definition: fmgr.h:305
Datum pg_replication_origin_session_is_setup ( PG_FUNCTION_ARGS  )

Definition at line 1241 of file origin.c.

References InvalidRepOriginId, PG_RETURN_BOOL, replorigin_check_prerequisites(), and replorigin_session_origin.

1242 {
1243  replorigin_check_prerequisites(false, false);
1244 
1246 }
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:319
RepOriginId replorigin_session_origin
Definition: origin.c:150
#define InvalidRepOriginId
Definition: origin.h:34
Datum pg_replication_origin_session_progress ( PG_FUNCTION_ARGS  )

Definition at line 1257 of file origin.c.

References ereport, errcode(), errmsg(), ERROR, InvalidXLogRecPtr, NULL, PG_GETARG_BOOL, PG_RETURN_LSN, PG_RETURN_NULL, replorigin_check_prerequisites(), and replorigin_session_get_progress().

1258 {
1259  XLogRecPtr remote_lsn = InvalidXLogRecPtr;
1260  bool flush = PG_GETARG_BOOL(0);
1261 
1262  replorigin_check_prerequisites(true, false);
1263 
1265  ereport(ERROR,
1266  (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
1267  errmsg("no replication origin is configured")));
1268 
1269  remote_lsn = replorigin_session_get_progress(flush);
1270 
1271  if (remote_lsn == InvalidXLogRecPtr)
1272  PG_RETURN_NULL();
1273 
1274  PG_RETURN_LSN(remote_lsn);
1275 }
static ReplicationState * session_replication_state
Definition: origin.c:169
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
XLogRecPtr replorigin_session_get_progress(bool flush)
Definition: origin.c:1107
int errcode(int sqlerrcode)
Definition: elog.c:575
#define PG_GETARG_BOOL(n)
Definition: fmgr.h:239
#define PG_RETURN_LSN(x)
Definition: pg_lsn.h:25
#define ERROR
Definition: elog.h:43
#define ereport(elevel, rest)
Definition: elog.h:122
#define NULL
Definition: c.h:229
uint64 XLogRecPtr
Definition: xlogdefs.h:21
int errmsg(const char *fmt,...)
Definition: elog.c:797
#define PG_RETURN_NULL()
Definition: fmgr.h:305
Datum pg_replication_origin_session_reset ( PG_FUNCTION_ARGS  )

Definition at line 1224 of file origin.c.

References InvalidRepOriginId, InvalidXLogRecPtr, PG_RETURN_VOID, replorigin_check_prerequisites(), replorigin_session_origin, replorigin_session_origin_lsn, replorigin_session_origin_timestamp, and replorigin_session_reset().

1225 {
1226  replorigin_check_prerequisites(true, false);
1227 
1229 
1233 
1234  PG_RETURN_VOID();
1235 }
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
void replorigin_session_reset(void)
Definition: origin.c:1065
XLogRecPtr replorigin_session_origin_lsn
Definition: origin.c:151
TimestampTz replorigin_session_origin_timestamp
Definition: origin.c:152
#define PG_RETURN_VOID()
Definition: fmgr.h:309
RepOriginId replorigin_session_origin
Definition: origin.c:150
#define InvalidRepOriginId
Definition: origin.h:34
Datum pg_replication_origin_session_setup ( PG_FUNCTION_ARGS  )

Definition at line 1202 of file origin.c.

References DatumGetPointer, name, pfree(), PG_GETARG_DATUM, PG_RETURN_VOID, replorigin_by_name(), replorigin_check_prerequisites(), replorigin_session_origin, replorigin_session_setup(), and text_to_cstring().

1203 {
1204  char *name;
1205  RepOriginId origin;
1206 
1207  replorigin_check_prerequisites(true, false);
1208 
1210  origin = replorigin_by_name(name, false);
1211  replorigin_session_setup(origin);
1212 
1213  replorigin_session_origin = origin;
1214 
1215  pfree(name);
1216 
1217  PG_RETURN_VOID();
1218 }
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:233
uint16 RepOriginId
Definition: xlogdefs.h:51
void replorigin_session_setup(RepOriginId node)
Definition: origin.c:980
RepOriginId replorigin_by_name(char *roname, bool missing_ok)
Definition: origin.c:206
void pfree(void *pointer)
Definition: mcxt.c:950
#define PG_RETURN_VOID()
Definition: fmgr.h:309
RepOriginId replorigin_session_origin
Definition: origin.c:150
const char * name
Definition: encode.c:521
#define DatumGetPointer(X)
Definition: postgres.h:555
char * text_to_cstring(const text *t)
Definition: varlena.c:182
Definition: c.h:439
Datum pg_replication_origin_xact_reset ( PG_FUNCTION_ARGS  )

Definition at line 1296 of file origin.c.

References InvalidXLogRecPtr, PG_RETURN_VOID, replorigin_check_prerequisites(), replorigin_session_origin_lsn, and replorigin_session_origin_timestamp.

1297 {
1298  replorigin_check_prerequisites(true, false);
1299 
1302 
1303  PG_RETURN_VOID();
1304 }
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
XLogRecPtr replorigin_session_origin_lsn
Definition: origin.c:151
TimestampTz replorigin_session_origin_timestamp
Definition: origin.c:152
#define PG_RETURN_VOID()
Definition: fmgr.h:309
Datum pg_replication_origin_xact_setup ( PG_FUNCTION_ARGS  )

Definition at line 1278 of file origin.c.

References ereport, errcode(), errmsg(), ERROR, NULL, PG_GETARG_LSN, PG_GETARG_TIMESTAMPTZ, PG_RETURN_VOID, replorigin_check_prerequisites(), replorigin_session_origin_lsn, and replorigin_session_origin_timestamp.

1279 {
1280  XLogRecPtr location = PG_GETARG_LSN(0);
1281 
1282  replorigin_check_prerequisites(true, false);
1283 
1285  ereport(ERROR,
1286  (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
1287  errmsg("no replication origin is configured")));
1288 
1289  replorigin_session_origin_lsn = location;
1291 
1292  PG_RETURN_VOID();
1293 }
static ReplicationState * session_replication_state
Definition: origin.c:169
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
int errcode(int sqlerrcode)
Definition: elog.c:575
#define ERROR
Definition: elog.h:43
XLogRecPtr replorigin_session_origin_lsn
Definition: origin.c:151
TimestampTz replorigin_session_origin_timestamp
Definition: origin.c:152
#define ereport(elevel, rest)
Definition: elog.h:122
#define PG_GETARG_LSN(n)
Definition: pg_lsn.h:24
#define PG_RETURN_VOID()
Definition: fmgr.h:309
#define NULL
Definition: c.h:229
uint64 XLogRecPtr
Definition: xlogdefs.h:21
int errmsg(const char *fmt,...)
Definition: elog.c:797
#define PG_GETARG_TIMESTAMPTZ(n)
Definition: timestamp.h:36
Datum pg_show_replication_origin_status ( PG_FUNCTION_ARGS  )

Definition at line 1368 of file origin.c.

References ReturnSetInfo::allowedModes, CStringGetTextDatum, ReturnSetInfo::econtext, ExprContext::ecxt_per_query_memory, elog, ereport, errcode(), errmsg(), ERROR, get_call_result_type(), i, InvalidRepOriginId, IsA, ReplicationState::local_lsn, ReplicationState::lock, LSNGetDatum, LW_SHARED, LWLockAcquire(), LWLockRelease(), max_replication_slots, MemoryContextSwitchTo(), NULL, ObjectIdGetDatum, ReplicationState::remote_lsn, REPLICATION_ORIGIN_PROGRESS_COLS, replorigin_by_oid(), replorigin_check_prerequisites(), ReturnSetInfo::returnMode, ReplicationState::roident, ReturnSetInfo::setDesc, ReturnSetInfo::setResult, SFRM_Materialize, tuplestore_begin_heap(), tuplestore_donestoring, tuplestore_putvalues(), TYPEFUNC_COMPOSITE, values, and work_mem.

1369 {
1370  ReturnSetInfo *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo;
1371  TupleDesc tupdesc;
1372  Tuplestorestate *tupstore;
1373  MemoryContext per_query_ctx;
1374  MemoryContext oldcontext;
1375  int i;
1377 
1378  /* we we want to return 0 rows if slot is set to zero */
1379  replorigin_check_prerequisites(false, true);
1380 
1381  if (rsinfo == NULL || !IsA(rsinfo, ReturnSetInfo))
1382  ereport(ERROR,
1383  (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1384  errmsg("set-valued function called in context that cannot accept a set")));
1385  if (!(rsinfo->allowedModes & SFRM_Materialize))
1386  ereport(ERROR,
1387  (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1388  errmsg("materialize mode required, but it is not allowed in this context")));
1389  if (get_call_result_type(fcinfo, NULL, &tupdesc) != TYPEFUNC_COMPOSITE)
1390  elog(ERROR, "return type must be a row type");
1391 
1392  if (tupdesc->natts != REPLICATION_ORIGIN_PROGRESS_COLS)
1393  elog(ERROR, "wrong function definition");
1394 
1395  per_query_ctx = rsinfo->econtext->ecxt_per_query_memory;
1396  oldcontext = MemoryContextSwitchTo(per_query_ctx);
1397 
1398  tupstore = tuplestore_begin_heap(true, false, work_mem);
1399  rsinfo->returnMode = SFRM_Materialize;
1400  rsinfo->setResult = tupstore;
1401  rsinfo->setDesc = tupdesc;
1402 
1403  MemoryContextSwitchTo(oldcontext);
1404 
1405 
1406  /* prevent slots from being concurrently dropped */
1407  LWLockAcquire(ReplicationOriginLock, LW_SHARED);
1408 
1409  /*
1410  * Iterate through all possible replication_states, display if they are
1411  * filled. Note that we do not take any locks, so slightly corrupted/out
1412  * of date values are a possibility.
1413  */
1414  for (i = 0; i < max_replication_slots; i++)
1415  {
1419  char *roname;
1420 
1421  state = &replication_states[i];
1422 
1423  /* unused slot, nothing to display */
1424  if (state->roident == InvalidRepOriginId)
1425  continue;
1426 
1427  memset(values, 0, sizeof(values));
1428  memset(nulls, 1, sizeof(nulls));
1429 
1430  values[0] = ObjectIdGetDatum(state->roident);
1431  nulls[0] = false;
1432 
1433  /*
1434  * We're not preventing the origin to be dropped concurrently, so
1435  * silently accept that it might be gone.
1436  */
1437  if (replorigin_by_oid(state->roident, true,
1438  &roname))
1439  {
1440  values[1] = CStringGetTextDatum(roname);
1441  nulls[1] = false;
1442  }
1443 
1444  LWLockAcquire(&state->lock, LW_SHARED);
1445 
1446  values[2] = LSNGetDatum(state->remote_lsn);
1447  nulls[2] = false;
1448 
1449  values[3] = LSNGetDatum(state->local_lsn);
1450  nulls[3] = false;
1451 
1452  LWLockRelease(&state->lock);
1453 
1454  tuplestore_putvalues(tupstore, tupdesc, values, nulls);
1455  }
1456 
1457  tuplestore_donestoring(tupstore);
1458 
1459  LWLockRelease(ReplicationOriginLock);
1460 
1461 #undef REPLICATION_ORIGIN_PROGRESS_COLS
1462 
1463  return (Datum) 0;
1464 }
void tuplestore_putvalues(Tuplestorestate *state, TupleDesc tdesc, Datum *values, bool *isnull)
Definition: tuplestore.c:735
static void replorigin_check_prerequisites(bool check_slots, bool recoveryOK)
Definition: origin.c:175
#define IsA(nodeptr, _type_)
Definition: nodes.h:557
TypeFuncClass get_call_result_type(FunctionCallInfo fcinfo, Oid *resultTypeId, TupleDesc *resultTupleDesc)
Definition: funcapi.c:211
XLogRecPtr local_lsn
Definition: origin.c:120
#define tuplestore_donestoring(state)
Definition: tuplestore.h:60
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
Definition: palloc.h:109
int errcode(int sqlerrcode)
Definition: elog.c:575
#define LSNGetDatum(X)
Definition: pg_lsn.h:22
bool replorigin_by_oid(RepOriginId roident, bool missing_ok, char **roname)
Definition: origin.c:398
RepOriginId roident
Definition: origin.c:108
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
#define ObjectIdGetDatum(X)
Definition: postgres.h:513
#define ERROR
Definition: elog.h:43
LWLock lock
Definition: origin.c:130
#define ereport(elevel, rest)
Definition: elog.h:122
Tuplestorestate * tuplestore_begin_heap(bool randomAccess, bool interXact, int maxKBytes)
Definition: tuplestore.c:316
uintptr_t Datum
Definition: postgres.h:372
int work_mem
Definition: globals.c:112
int allowedModes
Definition: execnodes.h:267
SetFunctionReturnMode returnMode
Definition: execnodes.h:269
int max_replication_slots
Definition: slot.c:99
XLogRecPtr remote_lsn
Definition: origin.c:113
#define NULL
Definition: c.h:229
Definition: regguts.h:298
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
MemoryContext ecxt_per_query_memory
Definition: execnodes.h:201
#define InvalidRepOriginId
Definition: origin.h:34
Tuplestorestate * setResult
Definition: execnodes.h:272
static Datum values[MAXATTR]
Definition: bootstrap.c:162
ExprContext * econtext
Definition: execnodes.h:265
TupleDesc setDesc
Definition: execnodes.h:273
int errmsg(const char *fmt,...)
Definition: elog.c:797
int i
static ReplicationState * replication_states
Definition: origin.c:161
#define CStringGetTextDatum(s)
Definition: builtins.h:91
#define elog
Definition: elog.h:219
#define REPLICATION_ORIGIN_PROGRESS_COLS
static void ReplicationOriginExitCleanup ( int  code,
Datum  arg 
)
static

Definition at line 955 of file origin.c.

References ReplicationState::acquired_by, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), MyProcPid, and NULL.

Referenced by replorigin_session_setup().

956 {
957  LWLockAcquire(ReplicationOriginLock, LW_EXCLUSIVE);
958 
961  {
964  }
965 
966  LWLockRelease(ReplicationOriginLock);
967 }
static ReplicationState * session_replication_state
Definition: origin.c:169
int MyProcPid
Definition: globals.c:38
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
#define NULL
Definition: c.h:229
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
void ReplicationOriginShmemInit ( void  )

Definition at line 457 of file origin.c.

References i, LWLockInitialize(), LWLockRegisterTranche(), LWTRANCHE_REPLICATION_ORIGIN, max_replication_slots, MemSet, ReplicationOriginShmemSize(), ShmemInitStruct(), ReplicationStateCtl::states, and ReplicationStateCtl::tranche_id.

Referenced by CreateSharedMemoryAndSemaphores().

458 {
459  bool found;
460 
461  if (max_replication_slots == 0)
462  return;
463 
465  ShmemInitStruct("ReplicationOriginState",
467  &found);
469 
470  if (!found)
471  {
472  int i;
473 
475 
477 
478  for (i = 0; i < max_replication_slots; i++)
481  }
482 
484  "replication_origin");
485 }
#define MemSet(start, val, len)
Definition: c.h:857
void * ShmemInitStruct(const char *name, Size size, bool *foundPtr)
Definition: shmem.c:372
void LWLockInitialize(LWLock *lock, int tranche_id)
Definition: lwlock.c:667
Size ReplicationOriginShmemSize(void)
Definition: origin.c:437
static ReplicationStateCtl * replication_states_ctl
Definition: origin.c:162
int max_replication_slots
Definition: slot.c:99
ReplicationState states[FLEXIBLE_ARRAY_MEMBER]
Definition: origin.c:146
void LWLockRegisterTranche(int tranche_id, char *tranche_name)
Definition: lwlock.c:592
int i
static ReplicationState * replication_states
Definition: origin.c:161
Size ReplicationOriginShmemSize ( void  )

Definition at line 437 of file origin.c.

References add_size(), max_replication_slots, mul_size(), and offsetof.

Referenced by CreateSharedMemoryAndSemaphores(), and ReplicationOriginShmemInit().

438 {
439  Size size = 0;
440 
441  /*
442  * XXX: max_replication_slots is arguably the wrong thing to use, as here
443  * we keep the replay state of *remote* transactions. But for now it seems
444  * sufficient to reuse it, lest we introduce a separate guc.
445  */
446  if (max_replication_slots == 0)
447  return size;
448 
449  size = add_size(size, offsetof(ReplicationStateCtl, states));
450 
451  size = add_size(size,
453  return size;
454 }
Size mul_size(Size s1, Size s2)
Definition: shmem.c:492
Size add_size(Size s1, Size s2)
Definition: shmem.c:475
int max_replication_slots
Definition: slot.c:99
size_t Size
Definition: c.h:356
#define offsetof(type, field)
Definition: c.h:555
void replorigin_advance ( RepOriginId  node,
XLogRecPtr  remote_commit,
XLogRecPtr  local_commit,
bool  go_backward,
bool  wal_log 
)

Definition at line 791 of file origin.c.

References ReplicationState::acquired_by, Assert, DoNotReplicateId, ereport, errcode(), errhint(), errmsg(), ERROR, xl_replorigin_set::force, i, InvalidRepOriginId, InvalidXLogRecPtr, ReplicationState::local_lsn, ReplicationState::lock, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), max_replication_slots, xl_replorigin_set::node_id, NULL, xl_replorigin_set::remote_lsn, ReplicationState::remote_lsn, ReplicationState::roident, XLOG_REPLORIGIN_SET, XLogBeginInsert(), XLogInsert(), and XLogRegisterData().

Referenced by pg_replication_origin_advance(), replorigin_redo(), and xact_redo_commit().

794 {
795  int i;
796  ReplicationState *replication_state = NULL;
797  ReplicationState *free_state = NULL;
798 
799  Assert(node != InvalidRepOriginId);
800 
801  /* we don't track DoNotReplicateId */
802  if (node == DoNotReplicateId)
803  return;
804 
805  /*
806  * XXX: For the case where this is called by WAL replay, it'd be more
807  * efficient to restore into a backend local hashtable and only dump into
808  * shmem after recovery is finished. Let's wait with implementing that
809  * till it's shown to be a measurable expense
810  */
811 
812  /* Lock exclusively, as we may have to create a new table entry. */
813  LWLockAcquire(ReplicationOriginLock, LW_EXCLUSIVE);
814 
815  /*
816  * Search for either an existing slot for the origin, or a free one we can
817  * use.
818  */
819  for (i = 0; i < max_replication_slots; i++)
820  {
821  ReplicationState *curstate = &replication_states[i];
822 
823  /* remember where to insert if necessary */
824  if (curstate->roident == InvalidRepOriginId &&
825  free_state == NULL)
826  {
827  free_state = curstate;
828  continue;
829  }
830 
831  /* not our slot */
832  if (curstate->roident != node)
833  {
834  continue;
835  }
836 
837  /* ok, found slot */
838  replication_state = curstate;
839 
840  LWLockAcquire(&replication_state->lock, LW_EXCLUSIVE);
841 
842  /* Make sure it's not used by somebody else */
843  if (replication_state->acquired_by != 0)
844  {
845  ereport(ERROR,
846  (errcode(ERRCODE_OBJECT_IN_USE),
847  errmsg("replication origin with OID %d is already active for PID %d",
848  replication_state->roident,
849  replication_state->acquired_by)));
850  }
851 
852  break;
853  }
854 
855  if (replication_state == NULL && free_state == NULL)
856  ereport(ERROR,
857  (errcode(ERRCODE_CONFIGURATION_LIMIT_EXCEEDED),
858  errmsg("could not find free replication state slot for replication origin with OID %u",
859  node),
860  errhint("Increase max_replication_slots and try again.")));
861 
862  if (replication_state == NULL)
863  {
864  /* initialize new slot */
865  LWLockAcquire(&free_state->lock, LW_EXCLUSIVE);
866  replication_state = free_state;
867  Assert(replication_state->remote_lsn == InvalidXLogRecPtr);
868  Assert(replication_state->local_lsn == InvalidXLogRecPtr);
869  replication_state->roident = node;
870  }
871 
872  Assert(replication_state->roident != InvalidRepOriginId);
873 
874  /*
875  * If somebody "forcefully" sets this slot, WAL log it, so it's durable
876  * and the standby gets the message. Primarily this will be called during
877  * WAL replay (of commit records) where no WAL logging is necessary.
878  */
879  if (wal_log)
880  {
881  xl_replorigin_set xlrec;
882 
883  xlrec.remote_lsn = remote_commit;
884  xlrec.node_id = node;
885  xlrec.force = go_backward;
886 
887  XLogBeginInsert();
888  XLogRegisterData((char *) (&xlrec), sizeof(xlrec));
889 
890  XLogInsert(RM_REPLORIGIN_ID, XLOG_REPLORIGIN_SET);
891  }
892 
893  /*
894  * Due to - harmless - race conditions during a checkpoint we could see
895  * values here that are older than the ones we already have in memory.
896  * Don't overwrite those.
897  */
898  if (go_backward || replication_state->remote_lsn < remote_commit)
899  replication_state->remote_lsn = remote_commit;
900  if (local_commit != InvalidXLogRecPtr &&
901  (go_backward || replication_state->local_lsn < local_commit))
902  replication_state->local_lsn = local_commit;
903  LWLockRelease(&replication_state->lock);
904 
905  /*
906  * Release *after* changing the LSNs, slot isn't acquired and thus could
907  * otherwise be dropped anytime.
908  */
909  LWLockRelease(ReplicationOriginLock);
910 }
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
int errhint(const char *fmt,...)
Definition: elog.c:987
XLogRecPtr local_lsn
Definition: origin.c:120
#define DoNotReplicateId
Definition: origin.h:35
int errcode(int sqlerrcode)
Definition: elog.c:575
RepOriginId roident
Definition: origin.c:108
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
#define ERROR
Definition: elog.h:43
LWLock lock
Definition: origin.c:130
#define XLOG_REPLORIGIN_SET
Definition: origin.h:31
#define ereport(elevel, rest)
Definition: elog.h:122
void XLogRegisterData(char *data, int len)
Definition: xloginsert.c:323
XLogRecPtr XLogInsert(RmgrId rmid, uint8 info)
Definition: xloginsert.c:415
int max_replication_slots
Definition: slot.c:99
XLogRecPtr remote_lsn
Definition: origin.c:113
#define NULL
Definition: c.h:229
#define Assert(condition)
Definition: c.h:675
RepOriginId node_id
Definition: origin.h:22
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
XLogRecPtr remote_lsn
Definition: origin.h:21
#define InvalidRepOriginId
Definition: origin.h:34
int errmsg(const char *fmt,...)
Definition: elog.c:797
int i
static ReplicationState * replication_states
Definition: origin.c:161
void XLogBeginInsert(void)
Definition: xloginsert.c:120
RepOriginId replorigin_by_name ( char *  roname,
bool  missing_ok 
)

Definition at line 206 of file origin.c.

References CStringGetTextDatum, elog, ERROR, GETSTRUCT, HeapTupleIsValid, InvalidOid, ReleaseSysCache(), REPLORIGNAME, and SearchSysCache1.

Referenced by ApplyWorkerMain(), DropSubscription(), pg_replication_origin_advance(), pg_replication_origin_drop(), pg_replication_origin_oid(), pg_replication_origin_progress(), and pg_replication_origin_session_setup().

207 {
209  Oid roident = InvalidOid;
210  HeapTuple tuple;
211  Datum roname_d;
212 
213  roname_d = CStringGetTextDatum(roname);
214 
215  tuple = SearchSysCache1(REPLORIGNAME, roname_d);
216  if (HeapTupleIsValid(tuple))
217  {
218  ident = (Form_pg_replication_origin) GETSTRUCT(tuple);
219  roident = ident->roident;
220  ReleaseSysCache(tuple);
221  }
222  else if (!missing_ok)
223  elog(ERROR, "cache lookup failed for replication origin '%s'",
224  roname);
225 
226  return roident;
227 }
#define GETSTRUCT(TUP)
Definition: htup_details.h:656
unsigned int Oid
Definition: postgres_ext.h:31
#define SearchSysCache1(cacheId, key1)
Definition: syscache.h:152
#define ERROR
Definition: elog.h:43
FormData_pg_replication_origin * Form_pg_replication_origin
uintptr_t Datum
Definition: postgres.h:372
void ReleaseSysCache(HeapTuple tuple)
Definition: syscache.c:1116
#define InvalidOid
Definition: postgres_ext.h:36
#define HeapTupleIsValid(tuple)
Definition: htup.h:77
#define CStringGetTextDatum(s)
Definition: builtins.h:91
#define elog
Definition: elog.h:219
bool replorigin_by_oid ( RepOriginId  roident,
bool  missing_ok,
char **  roname 
)

Definition at line 398 of file origin.c.

References Assert, DoNotReplicateId, elog, ERROR, GETSTRUCT, HeapTupleIsValid, InvalidRepOriginId, NULL, ObjectIdGetDatum, OidIsValid, ReleaseSysCache(), REPLORIGIDENT, SearchSysCache1, and text_to_cstring().

Referenced by pg_show_replication_origin_status(), and pgoutput_begin_txn().

399 {
400  HeapTuple tuple;
402 
403  Assert(OidIsValid((Oid) roident));
404  Assert(roident != InvalidRepOriginId);
405  Assert(roident != DoNotReplicateId);
406 
408  ObjectIdGetDatum((Oid) roident));
409 
410  if (HeapTupleIsValid(tuple))
411  {
412  ric = (Form_pg_replication_origin) GETSTRUCT(tuple);
413  *roname = text_to_cstring(&ric->roname);
414  ReleaseSysCache(tuple);
415 
416  return true;
417  }
418  else
419  {
420  *roname = NULL;
421 
422  if (!missing_ok)
423  elog(ERROR, "cache lookup failed for replication origin with oid %u",
424  roident);
425 
426  return false;
427  }
428 }
#define GETSTRUCT(TUP)
Definition: htup_details.h:656
#define DoNotReplicateId
Definition: origin.h:35
unsigned int Oid
Definition: postgres_ext.h:31
#define OidIsValid(objectId)
Definition: c.h:538
#define SearchSysCache1(cacheId, key1)
Definition: syscache.h:152
#define ObjectIdGetDatum(X)
Definition: postgres.h:513
#define ERROR
Definition: elog.h:43
FormData_pg_replication_origin * Form_pg_replication_origin
void ReleaseSysCache(HeapTuple tuple)
Definition: syscache.c:1116
#define HeapTupleIsValid(tuple)
Definition: htup.h:77
#define NULL
Definition: c.h:229
#define Assert(condition)
Definition: c.h:675
#define InvalidRepOriginId
Definition: origin.h:34
char * text_to_cstring(const text *t)
Definition: varlena.c:182
#define elog
Definition: elog.h:219
static void replorigin_check_prerequisites ( bool  check_slots,
bool  recoveryOK 
)
static

Definition at line 175 of file origin.c.

References ereport, errcode(), errmsg(), ERROR, max_replication_slots, RecoveryInProgress(), and superuser().

Referenced by pg_replication_origin_advance(), pg_replication_origin_create(), pg_replication_origin_drop(), pg_replication_origin_oid(), pg_replication_origin_progress(), pg_replication_origin_session_is_setup(), pg_replication_origin_session_progress(), pg_replication_origin_session_reset(), pg_replication_origin_session_setup(), pg_replication_origin_xact_reset(), pg_replication_origin_xact_setup(), and pg_show_replication_origin_status().

176 {
177  if (!superuser())
178  ereport(ERROR,
179  (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
180  errmsg("only superusers can query or manipulate replication origins")));
181 
182  if (check_slots && max_replication_slots == 0)
183  ereport(ERROR,
184  (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
185  errmsg("cannot query or manipulate replication origin when max_replication_slots = 0")));
186 
187  if (!recoveryOK && RecoveryInProgress())
188  ereport(ERROR,
189  (errcode(ERRCODE_READ_ONLY_SQL_TRANSACTION),
190  errmsg("cannot manipulate replication origins during recovery")));
191 
192 }
int errcode(int sqlerrcode)
Definition: elog.c:575
bool superuser(void)
Definition: superuser.c:47
bool RecoveryInProgress(void)
Definition: xlog.c:7860
#define ERROR
Definition: elog.h:43
#define ereport(elevel, rest)
Definition: elog.h:122
int max_replication_slots
Definition: slot.c:99
int errmsg(const char *fmt,...)
Definition: elog.c:797
RepOriginId replorigin_create ( char *  roname)

Definition at line 235 of file origin.c.

References Anum_pg_replication_origin_roident, Anum_pg_replication_origin_roname, Assert, BTEqualStrategyNumber, CatalogTupleInsert(), CHECK_FOR_INTERRUPTS, CommandCounterIncrement(), CStringGetTextDatum, ereport, errcode(), errmsg(), ERROR, ExclusiveLock, heap_close, heap_form_tuple(), heap_freetuple(), heap_open(), HeapTupleIsValid, InitDirtySnapshot, InvalidOid, IsTransactionState(), Natts_pg_replication_origin, NULL, ObjectIdGetDatum, PG_UINT16_MAX, RelationGetDescr, ReplicationOriginIdentIndex, ReplicationOriginRelationId, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), and values.

Referenced by ApplyWorkerMain(), CreateSubscription(), and pg_replication_origin_create().

236 {
237  Oid roident;
238  HeapTuple tuple = NULL;
239  Relation rel;
240  Datum roname_d;
241  SnapshotData SnapshotDirty;
242  SysScanDesc scan;
243  ScanKeyData key;
244 
245  roname_d = CStringGetTextDatum(roname);
246 
248 
249  /*
250  * We need the numeric replication origin to be 16bit wide, so we cannot
251  * rely on the normal oid allocation. Instead we simply scan
252  * pg_replication_origin for the first unused id. That's not particularly
253  * efficient, but this should be a fairly infrequent operation - we can
254  * easily spend a bit more code on this when it turns out it needs to be
255  * faster.
256  *
257  * We handle concurrency by taking an exclusive lock (allowing reads!)
258  * over the table for the duration of the search. Because we use a "dirty
259  * snapshot" we can read rows that other in-progress sessions have
260  * written, even though they would be invisible with normal snapshots. Due
261  * to the exclusive lock there's no danger that new rows can appear while
262  * we're checking.
263  */
264  InitDirtySnapshot(SnapshotDirty);
265 
267 
268  for (roident = InvalidOid + 1; roident < PG_UINT16_MAX; roident++)
269  {
270  bool nulls[Natts_pg_replication_origin];
272  bool collides;
273 
275 
276  ScanKeyInit(&key,
278  BTEqualStrategyNumber, F_OIDEQ,
279  ObjectIdGetDatum(roident));
280 
282  true /* indexOK */ ,
283  &SnapshotDirty,
284  1, &key);
285 
286  collides = HeapTupleIsValid(systable_getnext(scan));
287 
288  systable_endscan(scan);
289 
290  if (!collides)
291  {
292  /*
293  * Ok, found an unused roident, insert the new row and do a CCI,
294  * so our callers can look it up if they want to.
295  */
296  memset(&nulls, 0, sizeof(nulls));
297 
299  values[Anum_pg_replication_origin_roname - 1] = roname_d;
300 
301  tuple = heap_form_tuple(RelationGetDescr(rel), values, nulls);
302  CatalogTupleInsert(rel, tuple);
304  break;
305  }
306  }
307 
308  /* now release lock again, */
310 
311  if (tuple == NULL)
312  ereport(ERROR,
313  (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
314  errmsg("could not find free replication origin OID")));
315 
316  heap_freetuple(tuple);
317  return roident;
318 }
void systable_endscan(SysScanDesc sysscan)
Definition: genam.c:499
#define Anum_pg_replication_origin_roident
#define RelationGetDescr(relation)
Definition: rel.h:429
#define ExclusiveLock
Definition: lockdefs.h:44
int errcode(int sqlerrcode)
Definition: elog.c:575
HeapTuple heap_form_tuple(TupleDesc tupleDescriptor, Datum *values, bool *isnull)
Definition: heaptuple.c:692
#define heap_close(r, l)
Definition: heapam.h:97
void heap_freetuple(HeapTuple htup)
Definition: heaptuple.c:1374
unsigned int Oid
Definition: postgres_ext.h:31
SysScanDesc systable_beginscan(Relation heapRelation, Oid indexId, bool indexOK, Snapshot snapshot, int nkeys, ScanKey key)
Definition: genam.c:328
HeapTuple systable_getnext(SysScanDesc sysscan)
Definition: genam.c:416
#define Anum_pg_replication_origin_roname
#define ReplicationOriginIdentIndex
Definition: indexing.h:333
#define ObjectIdGetDatum(X)
Definition: postgres.h:513
#define ERROR
Definition: elog.h:43
#define PG_UINT16_MAX
Definition: c.h:338
Oid CatalogTupleInsert(Relation heapRel, HeapTuple tup)
Definition: indexing.c:162
#define InitDirtySnapshot(snapshotdata)
Definition: tqual.h:100
#define ereport(elevel, rest)
Definition: elog.h:122
uintptr_t Datum
Definition: postgres.h:372
void CommandCounterIncrement(void)
Definition: xact.c:922
Relation heap_open(Oid relationId, LOCKMODE lockmode)
Definition: heapam.c:1287
#define InvalidOid
Definition: postgres_ext.h:36
#define HeapTupleIsValid(tuple)
Definition: htup.h:77
#define NULL
Definition: c.h:229
#define Assert(condition)
Definition: c.h:675
bool IsTransactionState(void)
Definition: xact.c:350
#define Natts_pg_replication_origin
static Datum values[MAXATTR]
Definition: bootstrap.c:162
#define ReplicationOriginRelationId
int errmsg(const char *fmt,...)
Definition: elog.c:797
void ScanKeyInit(ScanKey entry, AttrNumber attributeNumber, StrategyNumber strategy, RegProcedure procedure, Datum argument)
Definition: scankey.c:76
#define CStringGetTextDatum(s)
Definition: builtins.h:91
#define CHECK_FOR_INTERRUPTS()
Definition: miscadmin.h:97
#define BTEqualStrategyNumber
Definition: stratnum.h:31
void replorigin_drop ( RepOriginId  roident)

Definition at line 327 of file origin.c.

References ReplicationState::acquired_by, Assert, CatalogTupleDelete(), CommandCounterIncrement(), elog, ereport, errcode(), errmsg(), ERROR, ExclusiveLock, heap_close, heap_open(), HeapTupleIsValid, i, InvalidRepOriginId, InvalidXLogRecPtr, IsTransactionState(), ReplicationState::local_lsn, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), max_replication_slots, xl_replorigin_drop::node_id, NULL, ObjectIdGetDatum, ReleaseSysCache(), ReplicationState::remote_lsn, ReplicationOriginRelationId, REPLORIGIDENT, ReplicationState::roident, SearchSysCache1, HeapTupleData::t_self, XLOG_REPLORIGIN_DROP, XLogBeginInsert(), XLogInsert(), and XLogRegisterData().

Referenced by DropSubscription(), and pg_replication_origin_drop().

328 {
329  HeapTuple tuple = NULL;
330  Relation rel;
331  int i;
332 
334 
336 
337  /* cleanup the slot state info */
338  LWLockAcquire(ReplicationOriginLock, LW_EXCLUSIVE);
339 
340  for (i = 0; i < max_replication_slots; i++)
341  {
343 
344  /* found our slot */
345  if (state->roident == roident)
346  {
347  if (state->acquired_by != 0)
348  {
349  ereport(ERROR,
350  (errcode(ERRCODE_OBJECT_IN_USE),
351  errmsg("could not drop replication origin with OID %d, in use by PID %d",
352  state->roident,
353  state->acquired_by)));
354  }
355 
356  /* first WAL log */
357  {
358  xl_replorigin_drop xlrec;
359 
360  xlrec.node_id = roident;
361  XLogBeginInsert();
362  XLogRegisterData((char *) (&xlrec), sizeof(xlrec));
363  XLogInsert(RM_REPLORIGIN_ID, XLOG_REPLORIGIN_DROP);
364  }
365 
366  /* then reset the in-memory entry */
367  state->roident = InvalidRepOriginId;
368  state->remote_lsn = InvalidXLogRecPtr;
369  state->local_lsn = InvalidXLogRecPtr;
370  break;
371  }
372  }
373  LWLockRelease(ReplicationOriginLock);
374 
376  if (!HeapTupleIsValid(tuple))
377  elog(ERROR, "cache lookup failed for replication origin with oid %u",
378  roident);
379 
380  CatalogTupleDelete(rel, &tuple->t_self);
381  ReleaseSysCache(tuple);
382 
384 
385  /* now release lock again, */
387 }
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
XLogRecPtr local_lsn
Definition: origin.c:120
#define ExclusiveLock
Definition: lockdefs.h:44
int errcode(int sqlerrcode)
Definition: elog.c:575
void CatalogTupleDelete(Relation heapRel, ItemPointer tid)
Definition: indexing.c:255
#define heap_close(r, l)
Definition: heapam.h:97
#define SearchSysCache1(cacheId, key1)
Definition: syscache.h:152
RepOriginId roident
Definition: origin.c:108
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
#define ObjectIdGetDatum(X)
Definition: postgres.h:513
#define ERROR
Definition: elog.h:43
ItemPointerData t_self
Definition: htup.h:65
#define ereport(elevel, rest)
Definition: elog.h:122
void XLogRegisterData(char *data, int len)
Definition: xloginsert.c:323
XLogRecPtr XLogInsert(RmgrId rmid, uint8 info)
Definition: xloginsert.c:415
#define XLOG_REPLORIGIN_DROP
Definition: origin.h:32
void CommandCounterIncrement(void)
Definition: xact.c:922
void ReleaseSysCache(HeapTuple tuple)
Definition: syscache.c:1116
Relation heap_open(Oid relationId, LOCKMODE lockmode)
Definition: heapam.c:1287
int max_replication_slots
Definition: slot.c:99
XLogRecPtr remote_lsn
Definition: origin.c:113
#define HeapTupleIsValid(tuple)
Definition: htup.h:77
RepOriginId node_id
Definition: origin.h:28
#define NULL
Definition: c.h:229
#define Assert(condition)
Definition: c.h:675
Definition: regguts.h:298
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
bool IsTransactionState(void)
Definition: xact.c:350
#define InvalidRepOriginId
Definition: origin.h:34
#define ReplicationOriginRelationId
int errmsg(const char *fmt,...)
Definition: elog.c:797
int i
static ReplicationState * replication_states
Definition: origin.c:161
#define elog
Definition: elog.h:219
void XLogBeginInsert(void)
Definition: xloginsert.c:120
XLogRecPtr replorigin_get_progress ( RepOriginId  node,
bool  flush 
)

Definition at line 914 of file origin.c.

References i, InvalidXLogRecPtr, ReplicationState::local_lsn, ReplicationState::lock, LW_SHARED, LWLockAcquire(), LWLockRelease(), max_replication_slots, ReplicationState::remote_lsn, ReplicationState::roident, and XLogFlush().

Referenced by pg_replication_origin_progress().

915 {
916  int i;
917  XLogRecPtr local_lsn = InvalidXLogRecPtr;
918  XLogRecPtr remote_lsn = InvalidXLogRecPtr;
919 
920  /* prevent slots from being concurrently dropped */
921  LWLockAcquire(ReplicationOriginLock, LW_SHARED);
922 
923  for (i = 0; i < max_replication_slots; i++)
924  {
926 
927  state = &replication_states[i];
928 
929  if (state->roident == node)
930  {
931  LWLockAcquire(&state->lock, LW_SHARED);
932 
933  remote_lsn = state->remote_lsn;
934  local_lsn = state->local_lsn;
935 
936  LWLockRelease(&state->lock);
937 
938  break;
939  }
940  }
941 
942  LWLockRelease(ReplicationOriginLock);
943 
944  if (flush && local_lsn != InvalidXLogRecPtr)
945  XLogFlush(local_lsn);
946 
947  return remote_lsn;
948 }
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
XLogRecPtr local_lsn
Definition: origin.c:120
void XLogFlush(XLogRecPtr record)
Definition: xlog.c:2754
RepOriginId roident
Definition: origin.c:108
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
LWLock lock
Definition: origin.c:130
int max_replication_slots
Definition: slot.c:99
XLogRecPtr remote_lsn
Definition: origin.c:113
uint64 XLogRecPtr
Definition: xlogdefs.h:21
Definition: regguts.h:298
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
int i
static ReplicationState * replication_states
Definition: origin.c:161
void replorigin_redo ( XLogReaderState record)

Definition at line 730 of file origin.c.

References elog, XLogReaderState::EndRecPtr, xl_replorigin_set::force, i, InvalidRepOriginId, InvalidXLogRecPtr, ReplicationState::local_lsn, max_replication_slots, xl_replorigin_set::node_id, xl_replorigin_drop::node_id, PANIC, xl_replorigin_set::remote_lsn, ReplicationState::remote_lsn, replorigin_advance(), ReplicationState::roident, XLOG_REPLORIGIN_DROP, XLOG_REPLORIGIN_SET, XLogRecGetData, XLogRecGetInfo, and XLR_INFO_MASK.

731 {
732  uint8 info = XLogRecGetInfo(record) & ~XLR_INFO_MASK;
733 
734  switch (info)
735  {
736  case XLOG_REPLORIGIN_SET:
737  {
738  xl_replorigin_set *xlrec =
739  (xl_replorigin_set *) XLogRecGetData(record);
740 
742  xlrec->remote_lsn, record->EndRecPtr,
743  xlrec->force /* backward */ ,
744  false /* WAL log */ );
745  break;
746  }
748  {
749  xl_replorigin_drop *xlrec;
750  int i;
751 
752  xlrec = (xl_replorigin_drop *) XLogRecGetData(record);
753 
754  for (i = 0; i < max_replication_slots; i++)
755  {
757 
758  /* found our slot */
759  if (state->roident == xlrec->node_id)
760  {
761  /* reset entry */
762  state->roident = InvalidRepOriginId;
763  state->remote_lsn = InvalidXLogRecPtr;
764  state->local_lsn = InvalidXLogRecPtr;
765  break;
766  }
767  }
768  break;
769  }
770  default:
771  elog(PANIC, "replorigin_redo: unknown op code %u", info);
772  }
773 }
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
XLogRecPtr local_lsn
Definition: origin.c:120
unsigned char uint8
Definition: c.h:266
void replorigin_advance(RepOriginId node, XLogRecPtr remote_commit, XLogRecPtr local_commit, bool go_backward, bool wal_log)
Definition: origin.c:791
#define PANIC
Definition: elog.h:53
RepOriginId roident
Definition: origin.c:108
XLogRecPtr EndRecPtr
Definition: xlogreader.h:115
#define XLogRecGetData(decoder)
Definition: xlogreader.h:218
#define XLOG_REPLORIGIN_SET
Definition: origin.h:31
#define XLogRecGetInfo(decoder)
Definition: xlogreader.h:214
#define XLOG_REPLORIGIN_DROP
Definition: origin.h:32
int max_replication_slots
Definition: slot.c:99
XLogRecPtr remote_lsn
Definition: origin.c:113
RepOriginId node_id
Definition: origin.h:28
#define XLR_INFO_MASK
Definition: xlogrecord.h:62
Definition: regguts.h:298
RepOriginId node_id
Definition: origin.h:22
XLogRecPtr remote_lsn
Definition: origin.h:21
#define InvalidRepOriginId
Definition: origin.h:34
int i
static ReplicationState * replication_states
Definition: origin.c:161
#define elog
Definition: elog.h:219
void replorigin_session_advance ( XLogRecPtr  remote_commit,
XLogRecPtr  local_commit 
)

Definition at line 1089 of file origin.c.

References Assert, InvalidRepOriginId, ReplicationState::local_lsn, ReplicationState::lock, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), NULL, ReplicationState::remote_lsn, and ReplicationState::roident.

Referenced by RecordTransactionCommit(), and RecordTransactionCommitPrepared().

1090 {
1093 
1095  if (session_replication_state->local_lsn < local_commit)
1096  session_replication_state->local_lsn = local_commit;
1097  if (session_replication_state->remote_lsn < remote_commit)
1098  session_replication_state->remote_lsn = remote_commit;
1100 }
static ReplicationState * session_replication_state
Definition: origin.c:169
XLogRecPtr local_lsn
Definition: origin.c:120
RepOriginId roident
Definition: origin.c:108
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
LWLock lock
Definition: origin.c:130
XLogRecPtr remote_lsn
Definition: origin.c:113
#define NULL
Definition: c.h:229
#define Assert(condition)
Definition: c.h:675
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
#define InvalidRepOriginId
Definition: origin.h:34
XLogRecPtr replorigin_session_get_progress ( bool  flush)

Definition at line 1107 of file origin.c.

References Assert, InvalidXLogRecPtr, ReplicationState::local_lsn, ReplicationState::lock, LW_SHARED, LWLockAcquire(), LWLockRelease(), NULL, ReplicationState::remote_lsn, and XLogFlush().

Referenced by ApplyWorkerMain(), and pg_replication_origin_session_progress().

1108 {
1109  XLogRecPtr remote_lsn;
1110  XLogRecPtr local_lsn;
1111 
1113 
1115  remote_lsn = session_replication_state->remote_lsn;
1116  local_lsn = session_replication_state->local_lsn;
1118 
1119  if (flush && local_lsn != InvalidXLogRecPtr)
1120  XLogFlush(local_lsn);
1121 
1122  return remote_lsn;
1123 }
static ReplicationState * session_replication_state
Definition: origin.c:169
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
XLogRecPtr local_lsn
Definition: origin.c:120
void XLogFlush(XLogRecPtr record)
Definition: xlog.c:2754
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
LWLock lock
Definition: origin.c:130
XLogRecPtr remote_lsn
Definition: origin.c:113
#define NULL
Definition: c.h:229
uint64 XLogRecPtr
Definition: xlogdefs.h:21
#define Assert(condition)
Definition: c.h:675
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
void replorigin_session_reset ( void  )

Definition at line 1065 of file origin.c.

References ReplicationState::acquired_by, Assert, ereport, errcode(), errmsg(), ERROR, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), max_replication_slots, and NULL.

Referenced by pg_replication_origin_session_reset().

1066 {
1068 
1070  ereport(ERROR,
1071  (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
1072  errmsg("no replication origin is configured")));
1073 
1074  LWLockAcquire(ReplicationOriginLock, LW_EXCLUSIVE);
1075 
1078 
1079  LWLockRelease(ReplicationOriginLock);
1080 }
static ReplicationState * session_replication_state
Definition: origin.c:169
int errcode(int sqlerrcode)
Definition: elog.c:575
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
#define ERROR
Definition: elog.h:43
#define ereport(elevel, rest)
Definition: elog.h:122
int max_replication_slots
Definition: slot.c:99
#define NULL
Definition: c.h:229
#define Assert(condition)
Definition: c.h:675
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
int errmsg(const char *fmt,...)
Definition: elog.c:797
void replorigin_session_setup ( RepOriginId  node)

Definition at line 980 of file origin.c.

References ReplicationState::acquired_by, Assert, ereport, errcode(), errhint(), errmsg(), ERROR, i, InvalidRepOriginId, InvalidXLogRecPtr, ReplicationState::local_lsn, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), max_replication_slots, MyProcPid, NULL, on_shmem_exit(), ReplicationState::remote_lsn, ReplicationOriginExitCleanup(), and ReplicationState::roident.

Referenced by ApplyWorkerMain(), and pg_replication_origin_session_setup().

981 {
982  static bool registered_cleanup;
983  int i;
984  int free_slot = -1;
985 
986  if (!registered_cleanup)
987  {
989  registered_cleanup = true;
990  }
991 
993 
995  ereport(ERROR,
996  (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE),
997  errmsg("cannot setup replication origin when one is already setup")));
998 
999  /* Lock exclusively, as we may have to create a new table entry. */
1000  LWLockAcquire(ReplicationOriginLock, LW_EXCLUSIVE);
1001 
1002  /*
1003  * Search for either an existing slot for the origin, or a free one we can
1004  * use.
1005  */
1006  for (i = 0; i < max_replication_slots; i++)
1007  {
1008  ReplicationState *curstate = &replication_states[i];
1009 
1010  /* remember where to insert if necessary */
1011  if (curstate->roident == InvalidRepOriginId &&
1012  free_slot == -1)
1013  {
1014  free_slot = i;
1015  continue;
1016  }
1017 
1018  /* not our slot */
1019  if (curstate->roident != node)
1020  continue;
1021 
1022  else if (curstate->acquired_by != 0)
1023  {
1024  ereport(ERROR,
1025  (errcode(ERRCODE_OBJECT_IN_USE),
1026  errmsg("replication identifier %d is already active for PID %d",
1027  curstate->roident, curstate->acquired_by)));
1028  }
1029 
1030  /* ok, found slot */
1031  session_replication_state = curstate;
1032  }
1033 
1034 
1035  if (session_replication_state == NULL && free_slot == -1)
1036  ereport(ERROR,
1037  (errcode(ERRCODE_CONFIGURATION_LIMIT_EXCEEDED),
1038  errmsg("could not find free replication state slot for replication origin with OID %u",
1039  node),
1040  errhint("Increase max_replication_slots and try again.")));
1041  else if (session_replication_state == NULL)
1042  {
1043  /* initialize new slot */
1048  }
1049 
1050 
1052 
1054 
1055  LWLockRelease(ReplicationOriginLock);
1056 }
static ReplicationState * session_replication_state
Definition: origin.c:169
#define InvalidXLogRecPtr
Definition: xlogdefs.h:28
int MyProcPid
Definition: globals.c:38
int errhint(const char *fmt,...)
Definition: elog.c:987
XLogRecPtr local_lsn
Definition: origin.c:120
static void ReplicationOriginExitCleanup(int code, Datum arg)
Definition: origin.c:955
int errcode(int sqlerrcode)
Definition: elog.c:575
RepOriginId roident
Definition: origin.c:108
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1715
#define ERROR
Definition: elog.h:43
void on_shmem_exit(pg_on_exit_callback function, Datum arg)
Definition: ipc.c:348
#define ereport(elevel, rest)
Definition: elog.h:122
int max_replication_slots
Definition: slot.c:99
XLogRecPtr remote_lsn
Definition: origin.c:113
#define NULL
Definition: c.h:229
#define Assert(condition)
Definition: c.h:675
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1111
#define InvalidRepOriginId
Definition: origin.h:34
int errmsg(const char *fmt,...)
Definition: elog.c:797
int i
static ReplicationState * replication_states
Definition: origin.c:161
void StartupReplicationOrigin ( void  )

Definition at line 615 of file origin.c.

References Assert, CloseTransientFile(), COMP_CRC32C, DEBUG2, elog, ereport, errcode(), errcode_for_file_access(), errmsg(), fd(), FIN_CRC32C, INIT_CRC32C, LOG, max_replication_slots, OpenTransientFile(), PANIC, PG_BINARY, read, ReplicationState::remote_lsn, ReplicationStateOnDisk::remote_lsn, REPLICATION_STATE_MAGIC, ReplicationState::roident, and ReplicationStateOnDisk::roident.

Referenced by StartupXLOG().

616 {
617  const char *path = "pg_logical/replorigin_checkpoint";
618  int fd;
619  int readBytes;
621  int last_state = 0;
622  pg_crc32c file_crc;
623  pg_crc32c crc;
624 
625  /* don't want to overwrite already existing state */
626 #ifdef USE_ASSERT_CHECKING
627  static bool already_started = false;
628 
629  Assert(!already_started);
630  already_started = true;
631 #endif
632 
633  if (max_replication_slots == 0)
634  return;
635 
636  INIT_CRC32C(crc);
637 
638  elog(DEBUG2, "starting up replication origin progress state");
639 
640  fd = OpenTransientFile((char *) path, O_RDONLY | PG_BINARY, 0);
641 
642  /*
643  * might have had max_replication_slots == 0 last run, or we just brought
644  * up a standby.
645  */
646  if (fd < 0 && errno == ENOENT)
647  return;
648  else if (fd < 0)
649  ereport(PANIC,
651  errmsg("could not open file \"%s\": %m",
652  path)));
653 
654  /* verify magic, that is written even if nothing was active */
655  readBytes = read(fd, &magic, sizeof(magic));
656  if (readBytes != sizeof(magic))
657  ereport(PANIC,
658  (errmsg("could not read file \"%s\": %m",
659  path)));
660  COMP_CRC32C(crc, &magic, sizeof(magic));
661 
662  if (magic != REPLICATION_STATE_MAGIC)
663  ereport(PANIC,
664  (errmsg("replication checkpoint has wrong magic %u instead of %u",
665  magic, REPLICATION_STATE_MAGIC)));
666 
667  /* we can skip locking here, no other access is possible */
668 
669  /* recover individual states, until there are no more to be found */
670  while (true)
671  {
672  ReplicationStateOnDisk disk_state;
673 
674  readBytes = read(fd, &disk_state, sizeof(disk_state));
675 
676  /* no further data */
677  if (readBytes == sizeof(crc))
678  {
679  /* not pretty, but simple ... */
680  file_crc = *(pg_crc32c *) &disk_state;
681  break;
682  }
683 
684  if (readBytes < 0)
685  {
686  ereport(PANIC,
688  errmsg("could not read file \"%s\": %m",
689  path)));
690  }
691 
692  if (readBytes != sizeof(disk_state))
693  {
694  ereport(PANIC,
696  errmsg("could not read file \"%s\": read %d of %zu",
697  path, readBytes, sizeof(disk_state))));
698  }
699 
700  COMP_CRC32C(crc, &disk_state, sizeof(disk_state));
701 
702  if (last_state == max_replication_slots)
703  ereport(PANIC,
704  (errcode(ERRCODE_CONFIGURATION_LIMIT_EXCEEDED),
705  errmsg("could not find free replication state, increase max_replication_slots")));
706 
707  /* copy data to shared memory */
708  replication_states[last_state].roident = disk_state.roident;
709  replication_states[last_state].remote_lsn = disk_state.remote_lsn;
710  last_state++;
711 
712  elog(LOG, "recovered replication state of node %u to %X/%X",
713  disk_state.roident,
714  (uint32) (disk_state.remote_lsn >> 32),
715  (uint32) disk_state.remote_lsn);
716  }
717 
718  /* now check checksum */
719  FIN_CRC32C(crc);
720  if (file_crc != crc)
721  ereport(PANIC,
722  (errcode(ERRCODE_CONFIGURATION_LIMIT_EXCEEDED),
723  errmsg("replication slot checkpoint has wrong checksum %u, expected %u",
724  crc, file_crc)));
725 
726  CloseTransientFile(fd);
727 }
#define INIT_CRC32C(crc)
Definition: pg_crc32c.h:41
XLogRecPtr remote_lsn
Definition: origin.c:139
uint32 pg_crc32c
Definition: pg_crc32c.h:38
RepOriginId roident
Definition: origin.c:138
int errcode(int sqlerrcode)
Definition: elog.c:575
#define LOG
Definition: elog.h:26
#define PANIC
Definition: elog.h:53
static int fd(const char *x, int i)
Definition: preproc-init.c:105
#define PG_BINARY
Definition: c.h:1038
RepOriginId roident
Definition: origin.c:108
#define DEBUG2
Definition: elog.h:24
int OpenTransientFile(FileName fileName, int fileFlags, int fileMode)
Definition: fd.c:2144
int errcode_for_file_access(void)
Definition: elog.c:598
unsigned int uint32
Definition: c.h:268
#define ereport(elevel, rest)
Definition: elog.h:122
int CloseTransientFile(int fd)
Definition: fd.c:2305
#define REPLICATION_STATE_MAGIC
Definition: origin.c:172
int max_replication_slots
Definition: slot.c:99
XLogRecPtr remote_lsn
Definition: origin.c:113
#define Assert(condition)
Definition: c.h:675
int errmsg(const char *fmt,...)
Definition: elog.c:797
static ReplicationState * replication_states
Definition: origin.c:161
#define elog
Definition: elog.h:219
#define COMP_CRC32C(crc, data, len)
Definition: pg_crc32c.h:73
#define FIN_CRC32C(crc)
Definition: pg_crc32c.h:78
#define read(a, b, c)
Definition: win32.h:18

Variable Documentation

ReplicationState* replication_states
static

Definition at line 161 of file origin.c.

ReplicationStateCtl* replication_states_ctl
static

Definition at line 162 of file origin.c.

ReplicationState* session_replication_state = NULL
static

Definition at line 169 of file origin.c.