pgstat_subscription.c

Go to the documentation of this file.

/* -------------------------------------------------------------------------
 *
 * pgstat_subscription.c
 *    Implementation of subscription statistics.
 *
 * This file contains the implementation of subscription statistics. It is kept
 * separate from pgstat.c to enforce the line between the statistics access /
 * storage implementation and the details about individual types of
 * statistics.
 *
 * Copyright (c) 2001-2025, PostgreSQL Global Development Group
 *
 * IDENTIFICATION
 *    src/backend/utils/activity/pgstat_subscription.c
 * -------------------------------------------------------------------------
 */
 
#include "postgres.h"
 
#include "replication/worker_internal.h"
#include "utils/pgstat_internal.h"
 
 
/*
 * Report a subscription error.
 */
void
pgstat_report_subscription_error(Oid subid, LogicalRepWorkerType wtype)
{
    PgStat_EntryRef *entry_ref;
    PgStat_BackendSubEntry *pending;
 
    entry_ref = pgstat_prep_pending_entry(PGSTAT_KIND_SUBSCRIPTION,
                                          InvalidOid, subid, NULL);
    pending = entry_ref->pending;
 
    switch (wtype)
    {
        case WORKERTYPE_APPLY:
            pending->apply_error_count++;
            break;
 
        case WORKERTYPE_SEQUENCESYNC:
            pending->sync_seq_error_count++;
            break;
 
        case WORKERTYPE_TABLESYNC:
            pending->sync_table_error_count++;
            break;
 
        default:
            /* Should never happen. */
            Assert(0);
            break;
    }
}
 
/*
 * Report a subscription conflict.
 */
void
pgstat_report_subscription_conflict(Oid subid, ConflictType type)
{
    PgStat_EntryRef *entry_ref;
    PgStat_BackendSubEntry *pending;
 
    entry_ref = pgstat_prep_pending_entry(PGSTAT_KIND_SUBSCRIPTION,
                                          InvalidOid, subid, NULL);
    pending = entry_ref->pending;
    pending->conflict_count[type]++;
}
 
/*
 * Report creating the subscription.
 */
void
pgstat_create_subscription(Oid subid)
{
    /* Ensures that stats are dropped if transaction rolls back */
    pgstat_create_transactional(PGSTAT_KIND_SUBSCRIPTION,
                                InvalidOid, subid);
 
    /* Create and initialize the subscription stats entry */
    pgstat_get_entry_ref(PGSTAT_KIND_SUBSCRIPTION, InvalidOid, subid,
                         true, NULL);
    pgstat_reset_entry(PGSTAT_KIND_SUBSCRIPTION, InvalidOid, subid, 0);
}
 
/*
 * Report dropping the subscription.
 *
 * Ensures that stats are dropped if transaction commits.
 */
void
pgstat_drop_subscription(Oid subid)
{
    pgstat_drop_transactional(PGSTAT_KIND_SUBSCRIPTION,
                              InvalidOid, subid);
}
 
/*
 * Support function for the SQL-callable pgstat* functions. Returns
 * the collected statistics for one subscription or NULL.
 */
PgStat_StatSubEntry *
pgstat_fetch_stat_subscription(Oid subid)
{
    return (PgStat_StatSubEntry *)
        pgstat_fetch_entry(PGSTAT_KIND_SUBSCRIPTION, InvalidOid, subid);
}
 
/*
 * Flush out pending stats for the entry
 *
 * If nowait is true and the lock could not be immediately acquired, returns
 * false without flushing the entry.  Otherwise returns true.
 */
bool
pgstat_subscription_flush_cb(PgStat_EntryRef *entry_ref, bool nowait)
{
    PgStat_BackendSubEntry *localent;
    PgStatShared_Subscription *shsubent;
 
    localent = (PgStat_BackendSubEntry *) entry_ref->pending;
    shsubent = (PgStatShared_Subscription *) entry_ref->shared_stats;
 
    /* localent always has non-zero content */
 
    if (!pgstat_lock_entry(entry_ref, nowait))
        return false;
 
#define SUB_ACC(fld) shsubent->stats.fld += localent->fld
    SUB_ACC(apply_error_count);
    SUB_ACC(sync_seq_error_count);
    SUB_ACC(sync_table_error_count);
    for (int i = 0; i < CONFLICT_NUM_TYPES; i++)
        SUB_ACC(conflict_count[i]);
#undef SUB_ACC
 
    pgstat_unlock_entry(entry_ref);
    return true;
}
 
void
pgstat_subscription_reset_timestamp_cb(PgStatShared_Common *header, TimestampTz ts)
{
    ((PgStatShared_Subscription *) header)->stats.stat_reset_timestamp = ts;
}