1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/*-------------------------------------------------------------------------
*
* pglogical_monitoring.c
* support for monitoring and progress tracking
*
* Copyright (c) 2017, PostgreSQL Global Development Group
*
* IDENTIFICATION
* pglogical_monitoring.c
*
*-------------------------------------------------------------------------
*/
#include "postgres.h"
#include "fmgr.h"
#include "miscadmin.h"
#include "replication/slot.h"
#include "utils/pg_lsn.h"
#include "storage/ipc.h"
#include "storage/proc.h"
#include "pgstat.h"
#include "pglogical.h"
PG_FUNCTION_INFO_V1(pglogical_wait_slot_confirm_lsn);
/*
* Wait for the confirmed_flush_lsn of the specified slot, or all logical slots
* if none given, to pass the supplied value. If no position is supplied the
* write position is used.
*
* No timeout is offered, use a statement_timeout.
*/
Datum
pglogical_wait_slot_confirm_lsn(PG_FUNCTION_ARGS)
{
XLogRecPtr target_lsn;
Name slot_name;
int i;
if (PG_ARGISNULL(0))
slot_name = NULL;
else
slot_name = PG_GETARG_NAME(0);
if (PG_ARGISNULL(1))
{
if (XLogRecPtrIsInvalid(XactLastCommitEnd))
target_lsn = GetXLogInsertRecPtr();
else
target_lsn = XactLastCommitEnd;
}
else
target_lsn = PG_GETARG_LSN(1);
elog(DEBUG1, "waiting for %s to pass confirmed_flush position %X/%X",
slot_name == NULL ? "all local slots" : NameStr(*slot_name),
(uint32)(target_lsn>>32), (uint32)target_lsn);
do
{
XLogRecPtr oldest_confirmed_lsn = InvalidXLogRecPtr;
int oldest_slot_pos = -1;
int rc;
LWLockAcquire(ReplicationSlotControlLock, LW_SHARED);
for (i = 0; i < max_replication_slots; i++)
{
ReplicationSlot *s = &ReplicationSlotCtl->replication_slots[i];
if (!s->in_use)
continue;
if (slot_name != NULL && strncmp(NameStr(*slot_name), NameStr(s->data.name), NAMEDATALEN) != 0)
continue;
if (oldest_confirmed_lsn == InvalidXLogRecPtr
|| (s->data.confirmed_flush != InvalidXLogRecPtr && s->data.confirmed_flush < oldest_confirmed_lsn))
{
oldest_confirmed_lsn = s->data.confirmed_flush;
oldest_slot_pos = i;
}
}
if (oldest_slot_pos >= 0)
elog(DEBUG2, "oldest confirmed lsn is %X/%X on slot '%s', %u bytes left until %X/%X",
(uint32)(oldest_confirmed_lsn>>32), (uint32)oldest_confirmed_lsn,
NameStr(ReplicationSlotCtl->replication_slots[oldest_slot_pos].data.name),
(uint32)(target_lsn - oldest_confirmed_lsn),
(uint32)(target_lsn>>32), (uint32)target_lsn);
LWLockRelease(ReplicationSlotControlLock);
if (oldest_confirmed_lsn >= target_lsn)
break;
rc = WaitLatch(&MyProc->procLatch,
WL_LATCH_SET | WL_TIMEOUT | WL_POSTMASTER_DEATH,
1000);
ResetLatch(&MyProc->procLatch);
if (rc & WL_POSTMASTER_DEATH)
proc_exit(1);
CHECK_FOR_INTERRUPTS();
} while(1);
PG_RETURN_VOID();
}
|
