database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-11-03 09:13:20 +03:00
Code Activity

Add worker type argument to logical replication worker functions.

Browse Source 
Extend logicalrep_worker_stop, logicalrep_worker_wakeup, and
logicalrep_worker_find to accept a worker type argument. This change
enables differentiation between logical replication worker types, such as
apply workers and table sync workers. While preserving existing behavior,
it lays the groundwork for upcoming patch to add sequence synchronization
workers.

Author: Vignesh C <vignesh21@gmail.com>
Reviewed-by: shveta malik <shveta.malik@gmail.com>
Reviewed-by: Peter Smith <smithpb2250@gmail.com>
Reviewed-by: Chao Li <li.evan.chao@gmail.com>
Reviewed-by: Hayato Kuroda <kuroda.hayato@fujitsu.com>
Reviewed-by: Amit Kapila <amit.kapila16@gmail.com>
Discussion: https://postgr.es/m/CAA4eK1LC+KJiAkSrpE_NwvNdidw9F2os7GERUeSxSKv71gXysQ@mail.gmail.com
This commit is contained in:
Amit Kapila
2025-10-28 05:47:50 +00:00
parent 8767b449a3
commit 3e8e05596a
6 changed files with 49 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
src/backend/replication/logical/launcher.c
View File

@@ -245,20 +245,25 @@ WaitForReplicationWorkerAttach(LogicalRepWorker *worker,
}
/*
* Walks the workers array and searches for one that matches given
* subscription id and relid.
* Walks the workers array and searches for one that matches given worker type,
* subscription id, and relation id.
*
* We are only interested in the leader apply worker or table sync worker.
* For apply workers, the relid should be set to InvalidOid, as they manage
* changes across all tables. For table sync workers, the relid should be set
* to the OID of the relation being synchronized.
*/
LogicalRepWorker *
logicalrep_worker_find(Oid subid, Oid relid, bool only_running)
logicalrep_worker_find(LogicalRepWorkerType wtype, Oid subid, Oid relid,
bool only_running)
{
int i;
LogicalRepWorker *res = NULL;
/* relid must be valid only for table sync workers */
Assert((wtype == WORKERTYPE_TABLESYNC) == OidIsValid(relid));
Assert(LWLockHeldByMe(LogicalRepWorkerLock));
/* Search for attached worker for a given subscription id. */
/* Search for an attached worker that matches the specified criteria. */
for (i = 0; i < max_logical_replication_workers; i++)
{
LogicalRepWorker *w = &LogicalRepCtx->workers[i];
@@ -268,7 +273,7 @@ logicalrep_worker_find(Oid subid, Oid relid, bool only_running)
continue;
if (w->in_use && w->subid == subid && w->relid == relid &&
(!only_running || w->proc))
w->type == wtype && (!only_running || w->proc))
{
res = w;
break;
@@ -627,16 +632,20 @@ logicalrep_worker_stop_internal(LogicalRepWorker *worker, int signo)
}
/*
* Stop the logical replication worker for subid/relid, if any.
* Stop the logical replication worker that matches the specified worker type,
* subscription id, and relation id.
*/
void
logicalrep_worker_stop(Oid subid, Oid relid)
logicalrep_worker_stop(LogicalRepWorkerType wtype, Oid subid, Oid relid)
{
LogicalRepWorker *worker;
/* relid must be valid only for table sync workers */
Assert((wtype == WORKERTYPE_TABLESYNC) == OidIsValid(relid));
LWLockAcquire(LogicalRepWorkerLock, LW_SHARED);
worker = logicalrep_worker_find(subid, relid, false);
worker = logicalrep_worker_find(wtype, subid, relid, false);
if (worker)
{
@@ -694,16 +703,20 @@ logicalrep_pa_worker_stop(ParallelApplyWorkerInfo *winfo)
}
/*
* Wake up (using latch) any logical replication worker for specified sub/rel.
* Wake up (using latch) any logical replication worker that matches the
* specified worker type, subscription id, and relation id.
*/
void
logicalrep_worker_wakeup(Oid subid, Oid relid)
logicalrep_worker_wakeup(LogicalRepWorkerType wtype, Oid subid, Oid relid)
{
LogicalRepWorker *worker;
/* relid must be valid only for table sync workers */
Assert((wtype == WORKERTYPE_TABLESYNC) == OidIsValid(relid));
LWLockAcquire(LogicalRepWorkerLock, LW_SHARED);
worker = logicalrep_worker_find(subid, relid, true);
worker = logicalrep_worker_find(wtype, subid, relid, true);
if (worker)
logicalrep_worker_wakeup_ptr(worker);
@@ -1260,7 +1273,8 @@ ApplyLauncherMain(Datum main_arg)
continue;
LWLockAcquire(LogicalRepWorkerLock, LW_SHARED);
w = logicalrep_worker_find(sub->oid, InvalidOid, false);
w = logicalrep_worker_find(WORKERTYPE_APPLY, sub->oid, InvalidOid,
false);
if (w != NULL)
{