TLA  Line data    Source code

       1                 : /* -------------------------------------------------------------------------
       2                 :  *
       3                 :  * pgstat_subscription.c
       4                 :  *    Implementation of subscription statistics.
       5                 :  *
       6                 :  * This file contains the implementation of subscription statistics. It is kept
       7                 :  * separate from pgstat.c to enforce the line between the statistics access /
       8                 :  * storage implementation and the details about individual types of
       9                 :  * statistics.
      10                 :  *
      11                 :  * Copyright (c) 2001-2023, PostgreSQL Global Development Group
      12                 :  *
      13                 :  * IDENTIFICATION
      14                 :  *    src/backend/utils/activity/pgstat_subscription.c
      15                 :  * -------------------------------------------------------------------------
      16                 :  */
      17                 : 
      18                 : #include "postgres.h"
      19                 : 
      20                 : #include "utils/pgstat_internal.h"
      21                 : 
      22                 : 
      23                 : /*
      24                 :  * Report a subscription error.
      25                 :  */
      26                 : void
      27 CBC          53 : pgstat_report_subscription_error(Oid subid, bool is_apply_error)
      28                 : {
      29                 :     PgStat_EntryRef *entry_ref;
      30                 :     PgStat_BackendSubEntry *pending;
      31                 : 
      32              53 :     entry_ref = pgstat_prep_pending_entry(PGSTAT_KIND_SUBSCRIPTION,
      33                 :                                           InvalidOid, subid, NULL);
      34              53 :     pending = entry_ref->pending;
      35                 : 
      36              53 :     if (is_apply_error)
      37              45 :         pending->apply_error_count++;
      38                 :     else
      39               8 :         pending->sync_error_count++;
      40              53 : }
      41                 : 
      42                 : /*
      43                 :  * Report creating the subscription.
      44 ECB             :  */
      45                 : void
      46 GIC         107 : pgstat_create_subscription(Oid subid)
      47 ECB             : {
      48                 :     /* Ensures that stats are dropped if transaction rolls back */
      49 GIC         107 :     pgstat_create_transactional(PGSTAT_KIND_SUBSCRIPTION,
      50                 :                                 InvalidOid, subid);
      51                 : 
      52                 :     /* Create and initialize the subscription stats entry */
      53 GNC         107 :     pgstat_get_entry_ref(PGSTAT_KIND_SUBSCRIPTION, InvalidOid, subid,
      54                 :                          true, NULL);
      55             107 :     pgstat_reset_entry(PGSTAT_KIND_SUBSCRIPTION, InvalidOid, subid, 0);
      56 GIC         107 : }
      57 ECB             : 
      58                 : /*
      59                 :  * Report dropping the subscription.
      60                 :  *
      61                 :  * Ensures that stats are dropped if transaction commits.
      62                 :  */
      63                 : void
      64 GIC          34 : pgstat_drop_subscription(Oid subid)
      65                 : {
      66              34 :     pgstat_drop_transactional(PGSTAT_KIND_SUBSCRIPTION,
      67                 :                               InvalidOid, subid);
      68 CBC          34 : }
      69                 : 
      70 ECB             : /*
      71                 :  * Support function for the SQL-callable pgstat* functions. Returns
      72                 :  * the collected statistics for one subscription or NULL.
      73                 :  */
      74                 : PgStat_StatSubEntry *
      75 GIC          29 : pgstat_fetch_stat_subscription(Oid subid)
      76                 : {
      77              29 :     return (PgStat_StatSubEntry *)
      78              29 :         pgstat_fetch_entry(PGSTAT_KIND_SUBSCRIPTION, InvalidOid, subid);
      79 ECB             : }
      80                 : 
      81                 : /*
      82                 :  * Flush out pending stats for the entry
      83                 :  *
      84                 :  * If nowait is true, this function returns false if lock could not
      85                 :  * immediately acquired, otherwise true is returned.
      86                 :  */
      87                 : bool
      88 GIC          53 : pgstat_subscription_flush_cb(PgStat_EntryRef *entry_ref, bool nowait)
      89                 : {
      90                 :     PgStat_BackendSubEntry *localent;
      91                 :     PgStatShared_Subscription *shsubent;
      92 ECB             : 
      93 GIC          53 :     localent = (PgStat_BackendSubEntry *) entry_ref->pending;
      94              53 :     shsubent = (PgStatShared_Subscription *) entry_ref->shared_stats;
      95                 : 
      96                 :     /* localent always has non-zero content */
      97 ECB             : 
      98 CBC          53 :     if (!pgstat_lock_entry(entry_ref, nowait))
      99 UIC           0 :         return false;
     100                 : 
     101                 : #define SUB_ACC(fld) shsubent->stats.fld += localent->fld
     102 CBC          53 :     SUB_ACC(apply_error_count);
     103 GBC          53 :     SUB_ACC(sync_error_count);
     104                 : #undef SUB_ACC
     105                 : 
     106 CBC          53 :     pgstat_unlock_entry(entry_ref);
     107              53 :     return true;
     108                 : }
     109                 : 
     110 ECB             : void
     111 CBC         119 : pgstat_subscription_reset_timestamp_cb(PgStatShared_Common *header, TimestampTz ts)
     112                 : {
     113 GIC         119 :     ((PgStatShared_Subscription *) header)->stats.stat_reset_timestamp = ts;
     114             119 : }