LCOV - code coverage report
Current view: top level - src/backend/utils/activity - pgstat_database.c (source / functions) Hit Total Coverage
Test: PostgreSQL 17devel Lines: 148 156 94.9 %
Date: 2024-04-24 00:11:41 Functions: 16 17 94.1 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : /* -------------------------------------------------------------------------
       2             :  *
       3             :  * pgstat_database.c
       4             :  *    Implementation of database statistics.
       5             :  *
       6             :  * This file contains the implementation of database 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-2024, PostgreSQL Global Development Group
      12             :  *
      13             :  * IDENTIFICATION
      14             :  *    src/backend/utils/activity/pgstat_database.c
      15             :  * -------------------------------------------------------------------------
      16             :  */
      17             : 
      18             : #include "postgres.h"
      19             : 
      20             : #include "storage/procsignal.h"
      21             : #include "utils/pgstat_internal.h"
      22             : #include "utils/timestamp.h"
      23             : 
      24             : 
      25             : static bool pgstat_should_report_connstat(void);
      26             : 
      27             : 
      28             : PgStat_Counter pgStatBlockReadTime = 0;
      29             : PgStat_Counter pgStatBlockWriteTime = 0;
      30             : PgStat_Counter pgStatActiveTime = 0;
      31             : PgStat_Counter pgStatTransactionIdleTime = 0;
      32             : SessionEndType pgStatSessionEndCause = DISCONNECT_NORMAL;
      33             : 
      34             : 
      35             : static int  pgStatXactCommit = 0;
      36             : static int  pgStatXactRollback = 0;
      37             : static PgStat_Counter pgLastSessionReportTime = 0;
      38             : 
      39             : 
      40             : /*
      41             :  * Remove entry for the database being dropped.
      42             :  */
      43             : void
      44          68 : pgstat_drop_database(Oid databaseid)
      45             : {
      46          68 :     pgstat_drop_transactional(PGSTAT_KIND_DATABASE, databaseid, InvalidOid);
      47          68 : }
      48             : 
      49             : /*
      50             :  * Called from autovacuum.c to report startup of an autovacuum process.
      51             :  * We are called before InitPostgres is done, so can't rely on MyDatabaseId;
      52             :  * the db OID must be passed in, instead.
      53             :  */
      54             : void
      55          22 : pgstat_report_autovac(Oid dboid)
      56             : {
      57             :     PgStat_EntryRef *entry_ref;
      58             :     PgStatShared_Database *dbentry;
      59             : 
      60             :     /* can't get here in single user mode */
      61             :     Assert(IsUnderPostmaster);
      62             : 
      63             :     /*
      64             :      * End-of-vacuum is reported instantly. Report the start the same way for
      65             :      * consistency. Vacuum doesn't run frequently and is a long-lasting
      66             :      * operation so it doesn't matter if we get blocked here a little.
      67             :      */
      68          22 :     entry_ref = pgstat_get_entry_ref_locked(PGSTAT_KIND_DATABASE,
      69             :                                             dboid, InvalidOid, false);
      70             : 
      71          22 :     dbentry = (PgStatShared_Database *) entry_ref->shared_stats;
      72          22 :     dbentry->stats.last_autovac_time = GetCurrentTimestamp();
      73             : 
      74          22 :     pgstat_unlock_entry(entry_ref);
      75          22 : }
      76             : 
      77             : /*
      78             :  * Report a Hot Standby recovery conflict.
      79             :  */
      80             : void
      81          24 : pgstat_report_recovery_conflict(int reason)
      82             : {
      83             :     PgStat_StatDBEntry *dbentry;
      84             : 
      85             :     Assert(IsUnderPostmaster);
      86          24 :     if (!pgstat_track_counts)
      87           0 :         return;
      88             : 
      89          24 :     dbentry = pgstat_prep_database_pending(MyDatabaseId);
      90             : 
      91          24 :     switch (reason)
      92             :     {
      93           4 :         case PROCSIG_RECOVERY_CONFLICT_DATABASE:
      94             : 
      95             :             /*
      96             :              * Since we drop the information about the database as soon as it
      97             :              * replicates, there is no point in counting these conflicts.
      98             :              */
      99           4 :             break;
     100           2 :         case PROCSIG_RECOVERY_CONFLICT_TABLESPACE:
     101           2 :             dbentry->conflict_tablespace++;
     102           2 :             break;
     103           2 :         case PROCSIG_RECOVERY_CONFLICT_LOCK:
     104           2 :             dbentry->conflict_lock++;
     105           2 :             break;
     106           2 :         case PROCSIG_RECOVERY_CONFLICT_SNAPSHOT:
     107           2 :             dbentry->conflict_snapshot++;
     108           2 :             break;
     109           2 :         case PROCSIG_RECOVERY_CONFLICT_BUFFERPIN:
     110           2 :             dbentry->conflict_bufferpin++;
     111           2 :             break;
     112          10 :         case PROCSIG_RECOVERY_CONFLICT_LOGICALSLOT:
     113          10 :             dbentry->conflict_logicalslot++;
     114          10 :             break;
     115           2 :         case PROCSIG_RECOVERY_CONFLICT_STARTUP_DEADLOCK:
     116           2 :             dbentry->conflict_startup_deadlock++;
     117           2 :             break;
     118             :     }
     119             : }
     120             : 
     121             : /*
     122             :  * Report a detected deadlock.
     123             :  */
     124             : void
     125          10 : pgstat_report_deadlock(void)
     126             : {
     127             :     PgStat_StatDBEntry *dbent;
     128             : 
     129          10 :     if (!pgstat_track_counts)
     130           0 :         return;
     131             : 
     132          10 :     dbent = pgstat_prep_database_pending(MyDatabaseId);
     133          10 :     dbent->deadlocks++;
     134             : }
     135             : 
     136             : /*
     137             :  * Report one or more checksum failures.
     138             :  */
     139             : void
     140           4 : pgstat_report_checksum_failures_in_db(Oid dboid, int failurecount)
     141             : {
     142             :     PgStat_EntryRef *entry_ref;
     143             :     PgStatShared_Database *sharedent;
     144             : 
     145           4 :     if (!pgstat_track_counts)
     146           0 :         return;
     147             : 
     148             :     /*
     149             :      * Update the shared stats directly - checksum failures should never be
     150             :      * common enough for that to be a problem.
     151             :      */
     152             :     entry_ref =
     153           4 :         pgstat_get_entry_ref_locked(PGSTAT_KIND_DATABASE, dboid, InvalidOid, false);
     154             : 
     155           4 :     sharedent = (PgStatShared_Database *) entry_ref->shared_stats;
     156           4 :     sharedent->stats.checksum_failures += failurecount;
     157           4 :     sharedent->stats.last_checksum_failure = GetCurrentTimestamp();
     158             : 
     159           4 :     pgstat_unlock_entry(entry_ref);
     160             : }
     161             : 
     162             : /*
     163             :  * Report one checksum failure in the current database.
     164             :  */
     165             : void
     166           0 : pgstat_report_checksum_failure(void)
     167             : {
     168           0 :     pgstat_report_checksum_failures_in_db(MyDatabaseId, 1);
     169           0 : }
     170             : 
     171             : /*
     172             :  * Report creation of temporary file.
     173             :  */
     174             : void
     175        5292 : pgstat_report_tempfile(size_t filesize)
     176             : {
     177             :     PgStat_StatDBEntry *dbent;
     178             : 
     179        5292 :     if (!pgstat_track_counts)
     180           0 :         return;
     181             : 
     182        5292 :     dbent = pgstat_prep_database_pending(MyDatabaseId);
     183        5292 :     dbent->temp_bytes += filesize;
     184        5292 :     dbent->temp_files++;
     185             : }
     186             : 
     187             : /*
     188             :  * Notify stats system of a new connection.
     189             :  */
     190             : void
     191       20716 : pgstat_report_connect(Oid dboid)
     192             : {
     193             :     PgStat_StatDBEntry *dbentry;
     194             : 
     195       20716 :     if (!pgstat_should_report_connstat())
     196        2036 :         return;
     197             : 
     198       18680 :     pgLastSessionReportTime = MyStartTimestamp;
     199             : 
     200       18680 :     dbentry = pgstat_prep_database_pending(MyDatabaseId);
     201       18680 :     dbentry->sessions++;
     202             : }
     203             : 
     204             : /*
     205             :  * Notify the stats system of a disconnect.
     206             :  */
     207             : void
     208       23396 : pgstat_report_disconnect(Oid dboid)
     209             : {
     210             :     PgStat_StatDBEntry *dbentry;
     211             : 
     212       23396 :     if (!pgstat_should_report_connstat())
     213        4716 :         return;
     214             : 
     215       18680 :     dbentry = pgstat_prep_database_pending(MyDatabaseId);
     216             : 
     217       18680 :     switch (pgStatSessionEndCause)
     218             :     {
     219       18582 :         case DISCONNECT_NOT_YET:
     220             :         case DISCONNECT_NORMAL:
     221             :             /* we don't collect these */
     222       18582 :             break;
     223          60 :         case DISCONNECT_CLIENT_EOF:
     224          60 :             dbentry->sessions_abandoned++;
     225          60 :             break;
     226          14 :         case DISCONNECT_FATAL:
     227          14 :             dbentry->sessions_fatal++;
     228          14 :             break;
     229          24 :         case DISCONNECT_KILLED:
     230          24 :             dbentry->sessions_killed++;
     231          24 :             break;
     232             :     }
     233             : }
     234             : 
     235             : /*
     236             :  * Support function for the SQL-callable pgstat* functions. Returns
     237             :  * the collected statistics for one database or NULL. NULL doesn't mean
     238             :  * that the database doesn't exist, just that there are no statistics, so the
     239             :  * caller is better off to report ZERO instead.
     240             :  */
     241             : PgStat_StatDBEntry *
     242        2906 : pgstat_fetch_stat_dbentry(Oid dboid)
     243             : {
     244        2906 :     return (PgStat_StatDBEntry *)
     245        2906 :         pgstat_fetch_entry(PGSTAT_KIND_DATABASE, dboid, InvalidOid);
     246             : }
     247             : 
     248             : void
     249      565178 : AtEOXact_PgStat_Database(bool isCommit, bool parallel)
     250             : {
     251             :     /* Don't count parallel worker transaction stats */
     252      565178 :     if (!parallel)
     253             :     {
     254             :         /*
     255             :          * Count transaction commit or abort.  (We use counters, not just
     256             :          * bools, in case the reporting message isn't sent right away.)
     257             :          */
     258      562534 :         if (isCommit)
     259      517324 :             pgStatXactCommit++;
     260             :         else
     261       45210 :             pgStatXactRollback++;
     262             :     }
     263      565178 : }
     264             : 
     265             : /*
     266             :  * Subroutine for pgstat_report_stat(): Handle xact commit/rollback and I/O
     267             :  * timings.
     268             :  */
     269             : void
     270       51654 : pgstat_update_dbstats(TimestampTz ts)
     271             : {
     272             :     PgStat_StatDBEntry *dbentry;
     273             : 
     274             :     /*
     275             :      * If not connected to a database yet, don't attribute time to "shared
     276             :      * state" (InvalidOid is used to track stats for shared relations, etc.).
     277             :      */
     278       51654 :     if (!OidIsValid(MyDatabaseId))
     279        4974 :         return;
     280             : 
     281       46680 :     dbentry = pgstat_prep_database_pending(MyDatabaseId);
     282             : 
     283             :     /*
     284             :      * Accumulate xact commit/rollback and I/O timings to stats entry of the
     285             :      * current database.
     286             :      */
     287       46680 :     dbentry->xact_commit += pgStatXactCommit;
     288       46680 :     dbentry->xact_rollback += pgStatXactRollback;
     289       46680 :     dbentry->blk_read_time += pgStatBlockReadTime;
     290       46680 :     dbentry->blk_write_time += pgStatBlockWriteTime;
     291             : 
     292       46680 :     if (pgstat_should_report_connstat())
     293             :     {
     294             :         long        secs;
     295             :         int         usecs;
     296             : 
     297             :         /*
     298             :          * pgLastSessionReportTime is initialized to MyStartTimestamp by
     299             :          * pgstat_report_connect().
     300             :          */
     301       38874 :         TimestampDifference(pgLastSessionReportTime, ts, &secs, &usecs);
     302       38874 :         pgLastSessionReportTime = ts;
     303       38874 :         dbentry->session_time += (PgStat_Counter) secs * 1000000 + usecs;
     304       38874 :         dbentry->active_time += pgStatActiveTime;
     305       38874 :         dbentry->idle_in_transaction_time += pgStatTransactionIdleTime;
     306             :     }
     307             : 
     308       46680 :     pgStatXactCommit = 0;
     309       46680 :     pgStatXactRollback = 0;
     310       46680 :     pgStatBlockReadTime = 0;
     311       46680 :     pgStatBlockWriteTime = 0;
     312       46680 :     pgStatActiveTime = 0;
     313       46680 :     pgStatTransactionIdleTime = 0;
     314             : }
     315             : 
     316             : /*
     317             :  * We report session statistics only for normal backend processes.  Parallel
     318             :  * workers run in parallel, so they don't contribute to session times, even
     319             :  * though they use CPU time. Walsender processes could be considered here,
     320             :  * but they have different session characteristics from normal backends (for
     321             :  * example, they are always "active"), so they would skew session statistics.
     322             :  */
     323             : static bool
     324       90792 : pgstat_should_report_connstat(void)
     325             : {
     326       90792 :     return MyBackendType == B_BACKEND;
     327             : }
     328             : 
     329             : /*
     330             :  * Find or create a local PgStat_StatDBEntry entry for dboid.
     331             :  */
     332             : PgStat_StatDBEntry *
     333     1321832 : pgstat_prep_database_pending(Oid dboid)
     334             : {
     335             :     PgStat_EntryRef *entry_ref;
     336             : 
     337             :     /*
     338             :      * This should not report stats on database objects before having
     339             :      * connected to a database.
     340             :      */
     341             :     Assert(!OidIsValid(dboid) || OidIsValid(MyDatabaseId));
     342             : 
     343     1321832 :     entry_ref = pgstat_prep_pending_entry(PGSTAT_KIND_DATABASE, dboid, InvalidOid,
     344             :                                           NULL);
     345             : 
     346     1321832 :     return entry_ref->pending;
     347             : }
     348             : 
     349             : /*
     350             :  * Reset the database's reset timestamp, without resetting the contents of the
     351             :  * database stats.
     352             :  */
     353             : void
     354          16 : pgstat_reset_database_timestamp(Oid dboid, TimestampTz ts)
     355             : {
     356             :     PgStat_EntryRef *dbref;
     357             :     PgStatShared_Database *dbentry;
     358             : 
     359          16 :     dbref = pgstat_get_entry_ref_locked(PGSTAT_KIND_DATABASE, MyDatabaseId, InvalidOid,
     360             :                                         false);
     361             : 
     362          16 :     dbentry = (PgStatShared_Database *) dbref->shared_stats;
     363          16 :     dbentry->stats.stat_reset_timestamp = ts;
     364             : 
     365          16 :     pgstat_unlock_entry(dbref);
     366          16 : }
     367             : 
     368             : /*
     369             :  * Flush out pending stats for the entry
     370             :  *
     371             :  * If nowait is true, this function returns false if lock could not
     372             :  * immediately acquired, otherwise true is returned.
     373             :  */
     374             : bool
     375       85496 : pgstat_database_flush_cb(PgStat_EntryRef *entry_ref, bool nowait)
     376             : {
     377             :     PgStatShared_Database *sharedent;
     378             :     PgStat_StatDBEntry *pendingent;
     379             : 
     380       85496 :     pendingent = (PgStat_StatDBEntry *) entry_ref->pending;
     381       85496 :     sharedent = (PgStatShared_Database *) entry_ref->shared_stats;
     382             : 
     383       85496 :     if (!pgstat_lock_entry(entry_ref, nowait))
     384           0 :         return false;
     385             : 
     386             : #define PGSTAT_ACCUM_DBCOUNT(item)      \
     387             :     (sharedent)->stats.item += (pendingent)->item
     388             : 
     389       85496 :     PGSTAT_ACCUM_DBCOUNT(xact_commit);
     390       85496 :     PGSTAT_ACCUM_DBCOUNT(xact_rollback);
     391       85496 :     PGSTAT_ACCUM_DBCOUNT(blocks_fetched);
     392       85496 :     PGSTAT_ACCUM_DBCOUNT(blocks_hit);
     393             : 
     394       85496 :     PGSTAT_ACCUM_DBCOUNT(tuples_returned);
     395       85496 :     PGSTAT_ACCUM_DBCOUNT(tuples_fetched);
     396       85496 :     PGSTAT_ACCUM_DBCOUNT(tuples_inserted);
     397       85496 :     PGSTAT_ACCUM_DBCOUNT(tuples_updated);
     398       85496 :     PGSTAT_ACCUM_DBCOUNT(tuples_deleted);
     399             : 
     400             :     /* last_autovac_time is reported immediately */
     401             :     Assert(pendingent->last_autovac_time == 0);
     402             : 
     403       85496 :     PGSTAT_ACCUM_DBCOUNT(conflict_tablespace);
     404       85496 :     PGSTAT_ACCUM_DBCOUNT(conflict_lock);
     405       85496 :     PGSTAT_ACCUM_DBCOUNT(conflict_snapshot);
     406       85496 :     PGSTAT_ACCUM_DBCOUNT(conflict_logicalslot);
     407       85496 :     PGSTAT_ACCUM_DBCOUNT(conflict_bufferpin);
     408       85496 :     PGSTAT_ACCUM_DBCOUNT(conflict_startup_deadlock);
     409             : 
     410       85496 :     PGSTAT_ACCUM_DBCOUNT(temp_bytes);
     411       85496 :     PGSTAT_ACCUM_DBCOUNT(temp_files);
     412       85496 :     PGSTAT_ACCUM_DBCOUNT(deadlocks);
     413             : 
     414             :     /* checksum failures are reported immediately */
     415             :     Assert(pendingent->checksum_failures == 0);
     416             :     Assert(pendingent->last_checksum_failure == 0);
     417             : 
     418       85496 :     PGSTAT_ACCUM_DBCOUNT(blk_read_time);
     419       85496 :     PGSTAT_ACCUM_DBCOUNT(blk_write_time);
     420             : 
     421       85496 :     PGSTAT_ACCUM_DBCOUNT(sessions);
     422       85496 :     PGSTAT_ACCUM_DBCOUNT(session_time);
     423       85496 :     PGSTAT_ACCUM_DBCOUNT(active_time);
     424       85496 :     PGSTAT_ACCUM_DBCOUNT(idle_in_transaction_time);
     425       85496 :     PGSTAT_ACCUM_DBCOUNT(sessions_abandoned);
     426       85496 :     PGSTAT_ACCUM_DBCOUNT(sessions_fatal);
     427       85496 :     PGSTAT_ACCUM_DBCOUNT(sessions_killed);
     428             : #undef PGSTAT_ACCUM_DBCOUNT
     429             : 
     430       85496 :     pgstat_unlock_entry(entry_ref);
     431             : 
     432       85496 :     memset(pendingent, 0, sizeof(*pendingent));
     433             : 
     434       85496 :     return true;
     435             : }
     436             : 
     437             : void
     438          26 : pgstat_database_reset_timestamp_cb(PgStatShared_Common *header, TimestampTz ts)
     439             : {
     440          26 :     ((PgStatShared_Database *) header)->stats.stat_reset_timestamp = ts;
     441          26 : }

Generated by: LCOV version 1.14