PostgreSQL Source Code  git master
pgstat_shmem.c File Reference
#include "postgres.h"
#include "pgstat.h"
#include "storage/shmem.h"
#include "utils/memutils.h"
#include "utils/pgstat_internal.h"
#include "lib/simplehash.h"
Include dependency graph for pgstat_shmem.c:

Go to the source code of this file.

Data Structures

struct  PgStat_EntryRefHashEntry
 

Macros

#define PGSTAT_ENTRY_REF_HASH_SIZE   128
 
#define SH_PREFIX   pgstat_entry_ref_hash
 
#define SH_ELEMENT_TYPE   PgStat_EntryRefHashEntry
 
#define SH_KEY_TYPE   PgStat_HashKey
 
#define SH_KEY   key
 
#define SH_HASH_KEY(tb, key)    pgstat_hash_hash_key(&key, sizeof(PgStat_HashKey), NULL)
 
#define SH_EQUAL(tb, a, b)    pgstat_cmp_hash_key(&a, &b, sizeof(PgStat_HashKey), NULL) == 0
 
#define SH_SCOPE   static inline
 
#define SH_DEFINE
 
#define SH_DECLARE
 

Typedefs

typedef struct PgStat_EntryRefHashEntry PgStat_EntryRefHashEntry
 
typedef bool(* ReleaseMatchCB) (PgStat_EntryRefHashEntry *, Datum data)
 

Functions

static void pgstat_drop_database_and_contents (Oid dboid)
 
static void pgstat_free_entry (PgStatShared_HashEntry *shent, dshash_seq_status *hstat)
 
static void pgstat_release_entry_ref (PgStat_HashKey key, PgStat_EntryRef *entry_ref, bool discard_pending)
 
static bool pgstat_need_entry_refs_gc (void)
 
static void pgstat_gc_entry_refs (void)
 
static void pgstat_release_all_entry_refs (bool discard_pending)
 
static void pgstat_release_matching_entry_refs (bool discard_pending, ReleaseMatchCB match, Datum match_data)
 
static void pgstat_setup_memcxt (void)
 
static Size pgstat_dsa_init_size (void)
 
Size StatsShmemSize (void)
 
void StatsShmemInit (void)
 
void pgstat_attach_shmem (void)
 
void pgstat_detach_shmem (void)
 
PgStatShared_Commonpgstat_init_entry (PgStat_Kind kind, PgStatShared_HashEntry *shhashent)
 
static PgStatShared_Commonpgstat_reinit_entry (PgStat_Kind kind, PgStatShared_HashEntry *shhashent)
 
static void pgstat_setup_shared_refs (void)
 
static void pgstat_acquire_entry_ref (PgStat_EntryRef *entry_ref, PgStatShared_HashEntry *shhashent, PgStatShared_Common *shheader)
 
static bool pgstat_get_entry_ref_cached (PgStat_HashKey key, PgStat_EntryRef **entry_ref_p)
 
PgStat_EntryRefpgstat_get_entry_ref (PgStat_Kind kind, Oid dboid, Oid objoid, bool create, bool *created_entry)
 
bool pgstat_lock_entry (PgStat_EntryRef *entry_ref, bool nowait)
 
bool pgstat_lock_entry_shared (PgStat_EntryRef *entry_ref, bool nowait)
 
void pgstat_unlock_entry (PgStat_EntryRef *entry_ref)
 
PgStat_EntryRefpgstat_get_entry_ref_locked (PgStat_Kind kind, Oid dboid, Oid objoid, bool nowait)
 
void pgstat_request_entry_refs_gc (void)
 
static bool match_db (PgStat_EntryRefHashEntry *ent, Datum match_data)
 
static void pgstat_release_db_entry_refs (Oid dboid)
 
static bool pgstat_drop_entry_internal (PgStatShared_HashEntry *shent, dshash_seq_status *hstat)
 
bool pgstat_drop_entry (PgStat_Kind kind, Oid dboid, Oid objoid)
 
void pgstat_drop_all_entries (void)
 
static void shared_stat_reset_contents (PgStat_Kind kind, PgStatShared_Common *header, TimestampTz ts)
 
void pgstat_reset_entry (PgStat_Kind kind, Oid dboid, Oid objoid, TimestampTz ts)
 
void pgstat_reset_matching_entries (bool(*do_reset)(PgStatShared_HashEntry *, Datum), Datum match_data, TimestampTz ts)
 
static bool match_kind (PgStatShared_HashEntry *p, Datum match_data)
 
void pgstat_reset_entries_of_kind (PgStat_Kind kind, TimestampTz ts)
 

Variables

static const dshash_parameters dsh_params
 
static pgstat_entry_ref_hash_hash * pgStatEntryRefHash = NULL
 
static int pgStatSharedRefAge = 0
 
static MemoryContext pgStatSharedRefContext = NULL
 
static MemoryContext pgStatEntryRefHashContext = NULL
 

Macro Definition Documentation

◆ PGSTAT_ENTRY_REF_HASH_SIZE

#define PGSTAT_ENTRY_REF_HASH_SIZE   128

Definition at line 21 of file pgstat_shmem.c.

◆ SH_DECLARE

#define SH_DECLARE

Definition at line 43 of file pgstat_shmem.c.

◆ SH_DEFINE

#define SH_DEFINE

Definition at line 42 of file pgstat_shmem.c.

◆ SH_ELEMENT_TYPE

#define SH_ELEMENT_TYPE   PgStat_EntryRefHashEntry

Definition at line 34 of file pgstat_shmem.c.

◆ SH_EQUAL

#define SH_EQUAL (   tb,
  a,
  b 
)     pgstat_cmp_hash_key(&a, &b, sizeof(PgStat_HashKey), NULL) == 0

Definition at line 39 of file pgstat_shmem.c.

◆ SH_HASH_KEY

#define SH_HASH_KEY (   tb,
  key 
)     pgstat_hash_hash_key(&key, sizeof(PgStat_HashKey), NULL)

Definition at line 37 of file pgstat_shmem.c.

◆ SH_KEY

#define SH_KEY   key

Definition at line 36 of file pgstat_shmem.c.

◆ SH_KEY_TYPE

#define SH_KEY_TYPE   PgStat_HashKey

Definition at line 35 of file pgstat_shmem.c.

◆ SH_PREFIX

#define SH_PREFIX   pgstat_entry_ref_hash

Definition at line 33 of file pgstat_shmem.c.

◆ SH_SCOPE

#define SH_SCOPE   static inline

Definition at line 41 of file pgstat_shmem.c.

Typedef Documentation

◆ PgStat_EntryRefHashEntry

◆ ReleaseMatchCB

typedef bool(* ReleaseMatchCB) (PgStat_EntryRefHashEntry *, Datum data)

Definition at line 55 of file pgstat_shmem.c.

Function Documentation

◆ match_db()

static bool match_db ( PgStat_EntryRefHashEntry ent,
Datum  match_data 
)
static

Definition at line 736 of file pgstat_shmem.c.

737 {
738  Oid dboid = DatumGetObjectId(match_data);
739 
740  return ent->key.dboid == dboid;
741 }
static Oid DatumGetObjectId(Datum X)
Definition: postgres.h:242
unsigned int Oid
Definition: postgres_ext.h:31

References DatumGetObjectId(), PgStat_HashKey::dboid, and PgStat_EntryRefHashEntry::key.

Referenced by pgstat_release_db_entry_refs().

◆ match_kind()

static bool match_kind ( PgStatShared_HashEntry p,
Datum  match_data 
)
static

Definition at line 1008 of file pgstat_shmem.c.

1009 {
1010  return p->key.kind == DatumGetInt32(match_data);
1011 }
static int32 DatumGetInt32(Datum X)
Definition: postgres.h:202
PgStat_Kind kind

References DatumGetInt32(), PgStatShared_HashEntry::key, and PgStat_HashKey::kind.

Referenced by pgstat_reset_entries_of_kind().

◆ pgstat_acquire_entry_ref()

static void pgstat_acquire_entry_ref ( PgStat_EntryRef entry_ref,
PgStatShared_HashEntry shhashent,
PgStatShared_Common shheader 
)
static

Definition at line 332 of file pgstat_shmem.c.

335 {
336  Assert(shheader->magic == 0xdeadbeef);
337  Assert(pg_atomic_read_u32(&shhashent->refcount) > 0);
338 
339  pg_atomic_fetch_add_u32(&shhashent->refcount, 1);
340 
342 
343  entry_ref->shared_stats = shheader;
344  entry_ref->shared_entry = shhashent;
345 }
static uint32 pg_atomic_fetch_add_u32(volatile pg_atomic_uint32 *ptr, int32 add_)
Definition: atomics.h:359
static uint32 pg_atomic_read_u32(volatile pg_atomic_uint32 *ptr)
Definition: atomics.h:232
#define Assert(condition)
Definition: c.h:858
void dshash_release_lock(dshash_table *hash_table, void *entry)
Definition: dshash.c:558
PgStat_LocalState pgStatLocal
Definition: pgstat.c:201
pg_atomic_uint32 refcount
PgStatShared_Common * shared_stats
PgStatShared_HashEntry * shared_entry
dshash_table * shared_hash

References Assert, dshash_release_lock(), PgStatShared_Common::magic, pg_atomic_fetch_add_u32(), pg_atomic_read_u32(), pgStatLocal, PgStatShared_HashEntry::refcount, PgStat_EntryRef::shared_entry, PgStat_LocalState::shared_hash, and PgStat_EntryRef::shared_stats.

Referenced by pgstat_get_entry_ref().

◆ pgstat_attach_shmem()

void pgstat_attach_shmem ( void  )

Definition at line 219 of file pgstat_shmem.c.

220 {
221  MemoryContext oldcontext;
222 
223  Assert(pgStatLocal.dsa == NULL);
224 
225  /* stats shared memory persists for the backend lifetime */
227 
229  NULL);
231 
234 
235  MemoryContextSwitchTo(oldcontext);
236 }
void dsa_pin_mapping(dsa_area *area)
Definition: dsa.c:635
dsa_area * dsa_attach_in_place(void *place, dsm_segment *segment)
Definition: dsa.c:545
dshash_table * dshash_attach(dsa_area *area, const dshash_parameters *params, dshash_table_handle handle, void *arg)
Definition: dshash.c:270
MemoryContext TopMemoryContext
Definition: mcxt.c:149
static const dshash_parameters dsh_params
Definition: pgstat_shmem.c:62
MemoryContextSwitchTo(old_ctx)
PgStat_ShmemControl * shmem
dshash_table_handle hash_handle

References Assert, PgStat_LocalState::dsa, dsa_attach_in_place(), dsa_pin_mapping(), dsh_params, dshash_attach(), PgStat_ShmemControl::hash_handle, MemoryContextSwitchTo(), pgStatLocal, PgStat_ShmemControl::raw_dsa_area, PgStat_LocalState::shared_hash, PgStat_LocalState::shmem, and TopMemoryContext.

Referenced by pgstat_initialize().

◆ pgstat_detach_shmem()

void pgstat_detach_shmem ( void  )

Definition at line 239 of file pgstat_shmem.c.

240 {
242 
243  /* we shouldn't leave references to shared stats */
245 
247  pgStatLocal.shared_hash = NULL;
248 
250 
251  /*
252  * dsa_detach() does not decrement the DSA reference count as no segment
253  * was provided to dsa_attach_in_place(), causing no cleanup callbacks to
254  * be registered. Hence, release it manually now.
255  */
257 
258  pgStatLocal.dsa = NULL;
259 }
void dsa_release_in_place(void *place)
Definition: dsa.c:605
void dsa_detach(dsa_area *area)
Definition: dsa.c:1952
void dshash_detach(dshash_table *hash_table)
Definition: dshash.c:307
static void pgstat_release_all_entry_refs(bool discard_pending)
Definition: pgstat_shmem.c:724

References Assert, PgStat_LocalState::dsa, dsa_detach(), dsa_release_in_place(), dshash_detach(), pgstat_release_all_entry_refs(), pgStatLocal, PgStat_ShmemControl::raw_dsa_area, PgStat_LocalState::shared_hash, and PgStat_LocalState::shmem.

Referenced by pgstat_shutdown_hook().

◆ pgstat_drop_all_entries()

void pgstat_drop_all_entries ( void  )

Definition at line 920 of file pgstat_shmem.c.

921 {
922  dshash_seq_status hstat;
924  uint64 not_freed_count = 0;
925 
926  dshash_seq_init(&hstat, pgStatLocal.shared_hash, true);
927  while ((ps = dshash_seq_next(&hstat)) != NULL)
928  {
929  if (ps->dropped)
930  continue;
931 
932  if (!pgstat_drop_entry_internal(ps, &hstat))
933  not_freed_count++;
934  }
935  dshash_seq_term(&hstat);
936 
937  if (not_freed_count > 0)
939 }
void dshash_seq_init(dshash_seq_status *status, dshash_table *hash_table, bool exclusive)
Definition: dshash.c:638
void dshash_seq_term(dshash_seq_status *status)
Definition: dshash.c:747
void * dshash_seq_next(dshash_seq_status *status)
Definition: dshash.c:657
struct parser_state ps
void pgstat_request_entry_refs_gc(void)
Definition: pgstat_shmem.c:638
static bool pgstat_drop_entry_internal(PgStatShared_HashEntry *shent, dshash_seq_status *hstat)
Definition: pgstat_shmem.c:783

References dshash_seq_init(), dshash_seq_next(), dshash_seq_term(), pgstat_drop_entry_internal(), pgstat_request_entry_refs_gc(), pgStatLocal, ps, and PgStat_LocalState::shared_hash.

Referenced by pgstat_reset_after_failure().

◆ pgstat_drop_database_and_contents()

static void pgstat_drop_database_and_contents ( Oid  dboid)
static

Definition at line 822 of file pgstat_shmem.c.

823 {
824  dshash_seq_status hstat;
826  uint64 not_freed_count = 0;
827 
828  Assert(OidIsValid(dboid));
829 
830  Assert(pgStatLocal.shared_hash != NULL);
831 
832  /*
833  * This backend might very well be the only backend holding a reference to
834  * about-to-be-dropped entries. Ensure that we're not preventing it from
835  * being cleaned up till later.
836  *
837  * Doing this separately from the dshash iteration below avoids having to
838  * do so while holding a partition lock on the shared hashtable.
839  */
841 
842  /* some of the dshash entries are to be removed, take exclusive lock. */
843  dshash_seq_init(&hstat, pgStatLocal.shared_hash, true);
844  while ((p = dshash_seq_next(&hstat)) != NULL)
845  {
846  if (p->dropped)
847  continue;
848 
849  if (p->key.dboid != dboid)
850  continue;
851 
852  if (!pgstat_drop_entry_internal(p, &hstat))
853  {
854  /*
855  * Even statistics for a dropped database might currently be
856  * accessed (consider e.g. database stats for pg_stat_database).
857  */
858  not_freed_count++;
859  }
860  }
861  dshash_seq_term(&hstat);
862 
863  /*
864  * If some of the stats data could not be freed, signal the reference
865  * holders to run garbage collection of their cached pgStatShmLookupCache.
866  */
867  if (not_freed_count > 0)
869 }
#define OidIsValid(objectId)
Definition: c.h:775
static void pgstat_release_db_entry_refs(Oid dboid)
Definition: pgstat_shmem.c:744

References Assert, PgStat_HashKey::dboid, PgStatShared_HashEntry::dropped, dshash_seq_init(), dshash_seq_next(), dshash_seq_term(), PgStatShared_HashEntry::key, OidIsValid, pgstat_drop_entry_internal(), pgstat_release_db_entry_refs(), pgstat_request_entry_refs_gc(), pgStatLocal, and PgStat_LocalState::shared_hash.

Referenced by pgstat_drop_entry().

◆ pgstat_drop_entry()

bool pgstat_drop_entry ( PgStat_Kind  kind,
Oid  dboid,
Oid  objoid 
)

Definition at line 883 of file pgstat_shmem.c.

884 {
885  PgStat_HashKey key = {.kind = kind,.dboid = dboid,.objoid = objoid};
886  PgStatShared_HashEntry *shent;
887  bool freed = true;
888 
889  /* delete local reference */
890  if (pgStatEntryRefHash)
891  {
892  PgStat_EntryRefHashEntry *lohashent =
893  pgstat_entry_ref_hash_lookup(pgStatEntryRefHash, key);
894 
895  if (lohashent)
896  pgstat_release_entry_ref(lohashent->key, lohashent->entry_ref,
897  true);
898  }
899 
900  /* mark entry in shared hashtable as deleted, drop if possible */
901  shent = dshash_find(pgStatLocal.shared_hash, &key, true);
902  if (shent)
903  {
904  freed = pgstat_drop_entry_internal(shent, NULL);
905 
906  /*
907  * Database stats contain other stats. Drop those as well when
908  * dropping the database. XXX: Perhaps this should be done in a
909  * slightly more principled way? But not obvious what that'd look
910  * like, and so far this is the only case...
911  */
912  if (key.kind == PGSTAT_KIND_DATABASE)
914  }
915 
916  return freed;
917 }
void * dshash_find(dshash_table *hash_table, const void *key, bool exclusive)
Definition: dshash.c:390
@ PGSTAT_KIND_DATABASE
Definition: pgstat.h:41
static void pgstat_release_entry_ref(PgStat_HashKey key, PgStat_EntryRef *entry_ref, bool discard_pending)
Definition: pgstat_shmem.c:531
static void pgstat_drop_database_and_contents(Oid dboid)
Definition: pgstat_shmem.c:822
static pgstat_entry_ref_hash_hash * pgStatEntryRefHash
Definition: pgstat_shmem.c:82
PgStat_EntryRef * entry_ref
Definition: pgstat_shmem.c:28

References dshash_find(), PgStat_EntryRefHashEntry::entry_ref, PgStat_EntryRefHashEntry::key, sort-test::key, pgstat_drop_database_and_contents(), pgstat_drop_entry_internal(), PGSTAT_KIND_DATABASE, pgstat_release_entry_ref(), pgStatEntryRefHash, pgStatLocal, and PgStat_LocalState::shared_hash.

Referenced by AtEOSubXact_PgStat_DroppedStats(), AtEOXact_PgStat_DroppedStats(), pgstat_drop_replslot(), pgstat_execute_transactional_drops(), and pgstat_init_function_usage().

◆ pgstat_drop_entry_internal()

static bool pgstat_drop_entry_internal ( PgStatShared_HashEntry shent,
dshash_seq_status hstat 
)
static

Definition at line 783 of file pgstat_shmem.c.

785 {
786  Assert(shent->body != InvalidDsaPointer);
787 
788  /* should already have released local reference */
789  if (pgStatEntryRefHash)
790  Assert(!pgstat_entry_ref_hash_lookup(pgStatEntryRefHash, shent->key));
791 
792  /*
793  * Signal that the entry is dropped - this will eventually cause other
794  * backends to release their references.
795  */
796  if (shent->dropped)
797  elog(ERROR,
798  "trying to drop stats entry already dropped: kind=%s dboid=%u objoid=%u refcount=%u",
800  shent->key.dboid, shent->key.objoid,
801  pg_atomic_read_u32(&shent->refcount));
802  shent->dropped = true;
803 
804  /* release refcount marking entry as not dropped */
805  if (pg_atomic_sub_fetch_u32(&shent->refcount, 1) == 0)
806  {
807  pgstat_free_entry(shent, hstat);
808  return true;
809  }
810  else
811  {
812  if (!hstat)
814  return false;
815  }
816 }
static uint32 pg_atomic_sub_fetch_u32(volatile pg_atomic_uint32 *ptr, int32 sub_)
Definition: atomics.h:432
#define InvalidDsaPointer
Definition: dsa.h:78
#define ERROR
Definition: elog.h:39
#define elog(elevel,...)
Definition: elog.h:224
const PgStat_KindInfo * pgstat_get_kind_info(PgStat_Kind kind)
Definition: pgstat.c:1306
static void pgstat_free_entry(PgStatShared_HashEntry *shent, dshash_seq_status *hstat)
Definition: pgstat_shmem.c:758
const char *const name

References Assert, PgStatShared_HashEntry::body, PgStat_HashKey::dboid, PgStatShared_HashEntry::dropped, dshash_release_lock(), elog, ERROR, InvalidDsaPointer, PgStatShared_HashEntry::key, PgStat_HashKey::kind, PgStat_KindInfo::name, PgStat_HashKey::objoid, pg_atomic_read_u32(), pg_atomic_sub_fetch_u32(), pgstat_free_entry(), pgstat_get_kind_info(), pgStatEntryRefHash, pgStatLocal, PgStatShared_HashEntry::refcount, and PgStat_LocalState::shared_hash.

Referenced by pgstat_drop_all_entries(), pgstat_drop_database_and_contents(), and pgstat_drop_entry().

◆ pgstat_dsa_init_size()

static Size pgstat_dsa_init_size ( void  )
static

Definition at line 106 of file pgstat_shmem.c.

107 {
108  Size sz;
109 
110  /*
111  * The dshash header / initial buckets array needs to fit into "plain"
112  * shared memory, but it's beneficial to not need dsm segments
113  * immediately. A size of 256kB seems works well and is not
114  * disproportional compared to other constant sized shared memory
115  * allocations. NB: To avoid DSMs further, the user can configure
116  * min_dynamic_shared_memory.
117  */
118  sz = 256 * 1024;
119  Assert(dsa_minimum_size() <= sz);
120  return MAXALIGN(sz);
121 }
#define MAXALIGN(LEN)
Definition: c.h:811
size_t Size
Definition: c.h:605
size_t dsa_minimum_size(void)
Definition: dsa.c:1196

References Assert, dsa_minimum_size(), and MAXALIGN.

Referenced by StatsShmemInit(), and StatsShmemSize().

◆ pgstat_free_entry()

static void pgstat_free_entry ( PgStatShared_HashEntry shent,
dshash_seq_status hstat 
)
static

Definition at line 758 of file pgstat_shmem.c.

759 {
760  dsa_pointer pdsa;
761 
762  /*
763  * Fetch dsa pointer before deleting entry - that way we can free the
764  * memory after releasing the lock.
765  */
766  pdsa = shent->body;
767 
768  if (!hstat)
770  else
771  dshash_delete_current(hstat);
772 
773  dsa_free(pgStatLocal.dsa, pdsa);
774 }
void dsa_free(dsa_area *area, dsa_pointer dp)
Definition: dsa.c:826
uint64 dsa_pointer
Definition: dsa.h:62
void dshash_delete_entry(dshash_table *hash_table, void *entry)
Definition: dshash.c:541
void dshash_delete_current(dshash_seq_status *status)
Definition: dshash.c:757

References PgStatShared_HashEntry::body, PgStat_LocalState::dsa, dsa_free(), dshash_delete_current(), dshash_delete_entry(), pgStatLocal, and PgStat_LocalState::shared_hash.

Referenced by pgstat_drop_entry_internal(), and pgstat_release_entry_ref().

◆ pgstat_gc_entry_refs()

static void pgstat_gc_entry_refs ( void  )
static

Definition at line 660 of file pgstat_shmem.c.

661 {
662  pgstat_entry_ref_hash_iterator i;
664  uint64 curage;
665 
667  Assert(curage != 0);
668 
669  /*
670  * Some entries have been dropped. Invalidate cache pointer to them.
671  */
672  pgstat_entry_ref_hash_start_iterate(pgStatEntryRefHash, &i);
673  while ((ent = pgstat_entry_ref_hash_iterate(pgStatEntryRefHash, &i)) != NULL)
674  {
675  PgStat_EntryRef *entry_ref = ent->entry_ref;
676 
677  Assert(!entry_ref->shared_stats ||
678  entry_ref->shared_stats->magic == 0xdeadbeef);
679 
680  if (!entry_ref->shared_entry->dropped)
681  continue;
682 
683  /* cannot gc shared ref that has pending data */
684  if (entry_ref->pending != NULL)
685  continue;
686 
687  pgstat_release_entry_ref(ent->key, entry_ref, false);
688  }
689 
690  pgStatSharedRefAge = curage;
691 }
static uint64 pg_atomic_read_u64(volatile pg_atomic_uint64 *ptr)
Definition: atomics.h:460
int i
Definition: isn.c:73
static int pgStatSharedRefAge
Definition: pgstat_shmem.c:83
pg_atomic_uint64 gc_request_count

References Assert, PgStatShared_HashEntry::dropped, PgStat_EntryRefHashEntry::entry_ref, PgStat_ShmemControl::gc_request_count, i, PgStat_EntryRefHashEntry::key, PgStatShared_Common::magic, PgStat_EntryRef::pending, pg_atomic_read_u64(), pgstat_release_entry_ref(), pgStatEntryRefHash, pgStatLocal, pgStatSharedRefAge, PgStat_EntryRef::shared_entry, PgStat_EntryRef::shared_stats, and PgStat_LocalState::shmem.

Referenced by pgstat_get_entry_ref().

◆ pgstat_get_entry_ref()

PgStat_EntryRef* pgstat_get_entry_ref ( PgStat_Kind  kind,
Oid  dboid,
Oid  objoid,
bool  create,
bool created_entry 
)

Definition at line 407 of file pgstat_shmem.c.

409 {
410  PgStat_HashKey key = {.kind = kind,.dboid = dboid,.objoid = objoid};
411  PgStatShared_HashEntry *shhashent;
412  PgStatShared_Common *shheader = NULL;
413  PgStat_EntryRef *entry_ref;
414 
415  /*
416  * passing in created_entry only makes sense if we possibly could create
417  * entry.
418  */
419  Assert(create || created_entry == NULL);
421  Assert(pgStatLocal.shared_hash != NULL);
423 
426 
427  if (created_entry != NULL)
428  *created_entry = false;
429 
430  /*
431  * Check if other backends dropped stats that could not be deleted because
432  * somebody held references to it. If so, check this backend's references.
433  * This is not expected to happen often. The location of the check is a
434  * bit random, but this is a relatively frequently called path, so better
435  * than most.
436  */
439 
440  /*
441  * First check the lookup cache hashtable in local memory. If we find a
442  * match here we can avoid taking locks / causing contention.
443  */
444  if (pgstat_get_entry_ref_cached(key, &entry_ref))
445  return entry_ref;
446 
447  Assert(entry_ref != NULL);
448 
449  /*
450  * Do a lookup in the hash table first - it's quite likely that the entry
451  * already exists, and that way we only need a shared lock.
452  */
453  shhashent = dshash_find(pgStatLocal.shared_hash, &key, false);
454 
455  if (create && !shhashent)
456  {
457  bool shfound;
458 
459  /*
460  * It's possible that somebody created the entry since the above
461  * lookup. If so, fall through to the same path as if we'd have if it
462  * already had been created before the dshash_find() calls.
463  */
464  shhashent = dshash_find_or_insert(pgStatLocal.shared_hash, &key, &shfound);
465  if (!shfound)
466  {
467  shheader = pgstat_init_entry(kind, shhashent);
468  pgstat_acquire_entry_ref(entry_ref, shhashent, shheader);
469 
470  if (created_entry != NULL)
471  *created_entry = true;
472 
473  return entry_ref;
474  }
475  }
476 
477  if (!shhashent)
478  {
479  /*
480  * If we're not creating, delete the reference again. In all
481  * likelihood it's just a stats lookup - no point wasting memory for a
482  * shared ref to nothing...
483  */
484  pgstat_release_entry_ref(key, entry_ref, false);
485 
486  return NULL;
487  }
488  else
489  {
490  /*
491  * Can get here either because dshash_find() found a match, or if
492  * dshash_find_or_insert() found a concurrently inserted entry.
493  */
494 
495  if (shhashent->dropped && create)
496  {
497  /*
498  * There are legitimate cases where the old stats entry might not
499  * yet have been dropped by the time it's reused. The most obvious
500  * case are replication slot stats, where a new slot can be
501  * created with the same index just after dropping. But oid
502  * wraparound can lead to other cases as well. We just reset the
503  * stats to their plain state.
504  */
505  shheader = pgstat_reinit_entry(kind, shhashent);
506  pgstat_acquire_entry_ref(entry_ref, shhashent, shheader);
507 
508  if (created_entry != NULL)
509  *created_entry = true;
510 
511  return entry_ref;
512  }
513  else if (shhashent->dropped)
514  {
516  pgstat_release_entry_ref(key, entry_ref, false);
517 
518  return NULL;
519  }
520  else
521  {
522  shheader = dsa_get_address(pgStatLocal.dsa, shhashent->body);
523  pgstat_acquire_entry_ref(entry_ref, shhashent, shheader);
524 
525  return entry_ref;
526  }
527  }
528 }
void * dsa_get_address(dsa_area *area, dsa_pointer dp)
Definition: dsa.c:942
void * dshash_find_or_insert(dshash_table *hash_table, const void *key, bool *found)
Definition: dshash.c:433
#define pgstat_assert_is_up()
static void pgstat_setup_memcxt(void)
PgStatShared_Common * pgstat_init_entry(PgStat_Kind kind, PgStatShared_HashEntry *shhashent)
Definition: pgstat_shmem.c:268
static void pgstat_setup_shared_refs(void)
Definition: pgstat_shmem.c:316
static PgStatShared_Common * pgstat_reinit_entry(PgStat_Kind kind, PgStatShared_HashEntry *shhashent)
Definition: pgstat_shmem.c:297
static void pgstat_acquire_entry_ref(PgStat_EntryRef *entry_ref, PgStatShared_HashEntry *shhashent, PgStatShared_Common *shheader)
Definition: pgstat_shmem.c:332
static bool pgstat_get_entry_ref_cached(PgStat_HashKey key, PgStat_EntryRef **entry_ref_p)
Definition: pgstat_shmem.c:351
static void pgstat_gc_entry_refs(void)
Definition: pgstat_shmem.c:660
static bool pgstat_need_entry_refs_gc(void)
Definition: pgstat_shmem.c:644

References Assert, PgStatShared_HashEntry::body, PgStatShared_HashEntry::dropped, PgStat_LocalState::dsa, dsa_get_address(), dshash_find(), dshash_find_or_insert(), dshash_release_lock(), PgStat_ShmemControl::is_shutdown, sort-test::key, pgstat_acquire_entry_ref(), pgstat_assert_is_up, pgstat_gc_entry_refs(), pgstat_get_entry_ref_cached(), pgstat_init_entry(), pgstat_need_entry_refs_gc(), pgstat_reinit_entry(), pgstat_release_entry_ref(), pgstat_setup_memcxt(), pgstat_setup_shared_refs(), pgStatLocal, PgStat_LocalState::shared_hash, and PgStat_LocalState::shmem.

Referenced by pgstat_acquire_replslot(), pgstat_create_subscription(), pgstat_create_transactional(), pgstat_fetch_entry(), pgstat_fetch_pending_entry(), pgstat_get_entry_ref_locked(), pgstat_have_entry(), pgstat_prep_pending_entry(), and pgstat_reset_entry().

◆ pgstat_get_entry_ref_cached()

static bool pgstat_get_entry_ref_cached ( PgStat_HashKey  key,
PgStat_EntryRef **  entry_ref_p 
)
static

Definition at line 351 of file pgstat_shmem.c.

352 {
353  bool found;
354  PgStat_EntryRefHashEntry *cache_entry;
355 
356  /*
357  * We immediately insert a cache entry, because it avoids 1) multiple
358  * hashtable lookups in case of a cache miss 2) having to deal with
359  * out-of-memory errors after incrementing PgStatShared_Common->refcount.
360  */
361 
362  cache_entry = pgstat_entry_ref_hash_insert(pgStatEntryRefHash, key, &found);
363 
364  if (!found || !cache_entry->entry_ref)
365  {
366  PgStat_EntryRef *entry_ref;
367 
368  cache_entry->entry_ref = entry_ref =
370  sizeof(PgStat_EntryRef));
371  entry_ref->shared_stats = NULL;
372  entry_ref->shared_entry = NULL;
373  entry_ref->pending = NULL;
374 
375  found = false;
376  }
377  else if (cache_entry->entry_ref->shared_stats == NULL)
378  {
379  Assert(cache_entry->entry_ref->pending == NULL);
380  found = false;
381  }
382  else
383  {
385 
386  entry_ref = cache_entry->entry_ref;
387  Assert(entry_ref->shared_entry != NULL);
388  Assert(entry_ref->shared_stats != NULL);
389 
390  Assert(entry_ref->shared_stats->magic == 0xdeadbeef);
391  /* should have at least our reference */
392  Assert(pg_atomic_read_u32(&entry_ref->shared_entry->refcount) > 0);
393  }
394 
395  *entry_ref_p = cache_entry->entry_ref;
396  return found;
397 }
#define PG_USED_FOR_ASSERTS_ONLY
Definition: c.h:182
void * MemoryContextAlloc(MemoryContext context, Size size)
Definition: mcxt.c:1181
static MemoryContext pgStatSharedRefContext
Definition: pgstat_shmem.c:90

References Assert, PgStat_EntryRefHashEntry::entry_ref, sort-test::key, MemoryContextAlloc(), PgStat_EntryRef::pending, pg_atomic_read_u32(), PG_USED_FOR_ASSERTS_ONLY, pgStatEntryRefHash, pgStatSharedRefContext, PgStat_EntryRef::shared_entry, and PgStat_EntryRef::shared_stats.

Referenced by pgstat_get_entry_ref().

◆ pgstat_get_entry_ref_locked()

PgStat_EntryRef* pgstat_get_entry_ref_locked ( PgStat_Kind  kind,
Oid  dboid,
Oid  objoid,
bool  nowait 
)

Definition at line 622 of file pgstat_shmem.c.

624 {
625  PgStat_EntryRef *entry_ref;
626 
627  /* find shared table stats entry corresponding to the local entry */
628  entry_ref = pgstat_get_entry_ref(kind, dboid, objoid, true, NULL);
629 
630  /* lock the shared entry to protect the content, skip if failed */
631  if (!pgstat_lock_entry(entry_ref, nowait))
632  return NULL;
633 
634  return entry_ref;
635 }
PgStat_EntryRef * pgstat_get_entry_ref(PgStat_Kind kind, Oid dboid, Oid objoid, bool create, bool *created_entry)
Definition: pgstat_shmem.c:407
bool pgstat_lock_entry(PgStat_EntryRef *entry_ref, bool nowait)
Definition: pgstat_shmem.c:585

References pgstat_get_entry_ref(), and pgstat_lock_entry().

Referenced by pgstat_copy_relation_stats(), pgstat_create_replslot(), pgstat_report_analyze(), pgstat_report_autovac(), pgstat_report_checksum_failures_in_db(), pgstat_report_replslot(), pgstat_report_vacuum(), and pgstat_reset_database_timestamp().

◆ pgstat_init_entry()

PgStatShared_Common* pgstat_init_entry ( PgStat_Kind  kind,
PgStatShared_HashEntry shhashent 
)

Definition at line 268 of file pgstat_shmem.c.

270 {
271  /* Create new stats entry. */
273  PgStatShared_Common *shheader;
274 
275  /*
276  * Initialize refcount to 1, marking it as valid / not dropped. The entry
277  * can't be freed before the initialization because it can't be found as
278  * long as we hold the dshash partition lock. Caller needs to increase
279  * further if a longer lived reference is needed.
280  */
281  pg_atomic_init_u32(&shhashent->refcount, 1);
282  shhashent->dropped = false;
283 
284  chunk = dsa_allocate0(pgStatLocal.dsa, pgstat_get_kind_info(kind)->shared_size);
285  shheader = dsa_get_address(pgStatLocal.dsa, chunk);
286  shheader->magic = 0xdeadbeef;
287 
288  /* Link the new entry from the hash entry. */
289  shhashent->body = chunk;
290 
292 
293  return shheader;
294 }
static void pg_atomic_init_u32(volatile pg_atomic_uint32 *ptr, uint32 val)
Definition: atomics.h:214
#define dsa_allocate0(area, size)
Definition: dsa.h:113
uint64 chunk
void LWLockInitialize(LWLock *lock, int tranche_id)
Definition: lwlock.c:707
@ LWTRANCHE_PGSTATS_DATA
Definition: lwlock.h:205

References PgStatShared_HashEntry::body, chunk, PgStatShared_HashEntry::dropped, PgStat_LocalState::dsa, dsa_allocate0, dsa_get_address(), PgStatShared_Common::lock, LWLockInitialize(), LWTRANCHE_PGSTATS_DATA, PgStatShared_Common::magic, pg_atomic_init_u32(), pgstat_get_kind_info(), pgStatLocal, and PgStatShared_HashEntry::refcount.

Referenced by pgstat_get_entry_ref(), and pgstat_read_statsfile().

◆ pgstat_lock_entry()

bool pgstat_lock_entry ( PgStat_EntryRef entry_ref,
bool  nowait 
)

Definition at line 585 of file pgstat_shmem.c.

586 {
587  LWLock *lock = &entry_ref->shared_stats->lock;
588 
589  if (nowait)
591 
593  return true;
594 }
bool LWLockAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1168
bool LWLockConditionalAcquire(LWLock *lock, LWLockMode mode)
Definition: lwlock.c:1339
@ LW_EXCLUSIVE
Definition: lwlock.h:114
Definition: lwlock.h:42

References PgStatShared_Common::lock, LW_EXCLUSIVE, LWLockAcquire(), LWLockConditionalAcquire(), and PgStat_EntryRef::shared_stats.

Referenced by pgstat_database_flush_cb(), pgstat_function_flush_cb(), pgstat_get_entry_ref_locked(), pgstat_relation_flush_cb(), pgstat_reset_entry(), and pgstat_subscription_flush_cb().

◆ pgstat_lock_entry_shared()

bool pgstat_lock_entry_shared ( PgStat_EntryRef entry_ref,
bool  nowait 
)

Definition at line 601 of file pgstat_shmem.c.

602 {
603  LWLock *lock = &entry_ref->shared_stats->lock;
604 
605  if (nowait)
606  return LWLockConditionalAcquire(lock, LW_SHARED);
607 
608  LWLockAcquire(lock, LW_SHARED);
609  return true;
610 }
@ LW_SHARED
Definition: lwlock.h:115

References PgStatShared_Common::lock, LW_SHARED, LWLockAcquire(), LWLockConditionalAcquire(), and PgStat_EntryRef::shared_stats.

Referenced by pgstat_fetch_entry().

◆ pgstat_need_entry_refs_gc()

static bool pgstat_need_entry_refs_gc ( void  )
static

Definition at line 644 of file pgstat_shmem.c.

645 {
646  uint64 curage;
647 
648  if (!pgStatEntryRefHash)
649  return false;
650 
651  /* should have been initialized when creating pgStatEntryRefHash */
653 
655 
656  return pgStatSharedRefAge != curage;
657 }

References Assert, PgStat_ShmemControl::gc_request_count, pg_atomic_read_u64(), pgStatEntryRefHash, pgStatLocal, pgStatSharedRefAge, and PgStat_LocalState::shmem.

Referenced by pgstat_get_entry_ref().

◆ pgstat_reinit_entry()

static PgStatShared_Common* pgstat_reinit_entry ( PgStat_Kind  kind,
PgStatShared_HashEntry shhashent 
)
static

Definition at line 297 of file pgstat_shmem.c.

298 {
299  PgStatShared_Common *shheader;
300 
301  shheader = dsa_get_address(pgStatLocal.dsa, shhashent->body);
302 
303  /* mark as not dropped anymore */
304  pg_atomic_fetch_add_u32(&shhashent->refcount, 1);
305  shhashent->dropped = false;
306 
307  /* reinitialize content */
308  Assert(shheader->magic == 0xdeadbeef);
309  memset(pgstat_get_entry_data(kind, shheader), 0,
310  pgstat_get_entry_len(kind));
311 
312  return shheader;
313 }
static void * pgstat_get_entry_data(PgStat_Kind kind, PgStatShared_Common *entry)
static size_t pgstat_get_entry_len(PgStat_Kind kind)

References Assert, PgStatShared_HashEntry::body, PgStatShared_HashEntry::dropped, PgStat_LocalState::dsa, dsa_get_address(), PgStatShared_Common::magic, pg_atomic_fetch_add_u32(), pgstat_get_entry_data(), pgstat_get_entry_len(), pgStatLocal, and PgStatShared_HashEntry::refcount.

Referenced by pgstat_get_entry_ref().

◆ pgstat_release_all_entry_refs()

static void pgstat_release_all_entry_refs ( bool  discard_pending)
static

Definition at line 724 of file pgstat_shmem.c.

725 {
726  if (pgStatEntryRefHash == NULL)
727  return;
728 
729  pgstat_release_matching_entry_refs(discard_pending, NULL, 0);
730  Assert(pgStatEntryRefHash->members == 0);
731  pgstat_entry_ref_hash_destroy(pgStatEntryRefHash);
732  pgStatEntryRefHash = NULL;
733 }
static void pgstat_release_matching_entry_refs(bool discard_pending, ReleaseMatchCB match, Datum match_data)
Definition: pgstat_shmem.c:694

References Assert, pgstat_release_matching_entry_refs(), and pgStatEntryRefHash.

Referenced by pgstat_detach_shmem().

◆ pgstat_release_db_entry_refs()

static void pgstat_release_db_entry_refs ( Oid  dboid)
static

Definition at line 744 of file pgstat_shmem.c.

745 {
746  pgstat_release_matching_entry_refs( /* discard pending = */ true,
747  match_db,
748  ObjectIdGetDatum(dboid));
749 }
static bool match_db(PgStat_EntryRefHashEntry *ent, Datum match_data)
Definition: pgstat_shmem.c:736
static Datum ObjectIdGetDatum(Oid X)
Definition: postgres.h:252

References match_db(), ObjectIdGetDatum(), and pgstat_release_matching_entry_refs().

Referenced by pgstat_drop_database_and_contents().

◆ pgstat_release_entry_ref()

static void pgstat_release_entry_ref ( PgStat_HashKey  key,
PgStat_EntryRef entry_ref,
bool  discard_pending 
)
static

Definition at line 531 of file pgstat_shmem.c.

533 {
534  if (entry_ref && entry_ref->pending)
535  {
536  if (discard_pending)
537  pgstat_delete_pending_entry(entry_ref);
538  else
539  elog(ERROR, "releasing ref with pending data");
540  }
541 
542  if (entry_ref && entry_ref->shared_stats)
543  {
544  Assert(entry_ref->shared_stats->magic == 0xdeadbeef);
545  Assert(entry_ref->pending == NULL);
546 
547  /*
548  * This can't race with another backend looking up the stats entry and
549  * increasing the refcount because it is not "legal" to create
550  * additional references to dropped entries.
551  */
552  if (pg_atomic_fetch_sub_u32(&entry_ref->shared_entry->refcount, 1) == 1)
553  {
554  PgStatShared_HashEntry *shent;
555 
556  /*
557  * We're the last referrer to this entry, try to drop the shared
558  * entry.
559  */
560 
561  /* only dropped entries can reach a 0 refcount */
562  Assert(entry_ref->shared_entry->dropped);
563 
565  &entry_ref->shared_entry->key,
566  true);
567  if (!shent)
568  elog(ERROR, "could not find just referenced shared stats entry");
569 
570  Assert(pg_atomic_read_u32(&entry_ref->shared_entry->refcount) == 0);
571  Assert(entry_ref->shared_entry == shent);
572 
573  pgstat_free_entry(shent, NULL);
574  }
575  }
576 
577  if (!pgstat_entry_ref_hash_delete(pgStatEntryRefHash, key))
578  elog(ERROR, "entry ref vanished before deletion");
579 
580  if (entry_ref)
581  pfree(entry_ref);
582 }
static uint32 pg_atomic_fetch_sub_u32(volatile pg_atomic_uint32 *ptr, int32 sub_)
Definition: atomics.h:374
void pfree(void *pointer)
Definition: mcxt.c:1521
void pgstat_delete_pending_entry(PgStat_EntryRef *entry_ref)
Definition: pgstat.c:1199

References Assert, PgStatShared_HashEntry::dropped, dshash_find(), elog, ERROR, PgStatShared_HashEntry::key, sort-test::key, PgStatShared_Common::magic, PgStat_EntryRef::pending, pfree(), pg_atomic_fetch_sub_u32(), pg_atomic_read_u32(), pgstat_delete_pending_entry(), pgstat_free_entry(), pgStatEntryRefHash, pgStatLocal, PgStatShared_HashEntry::refcount, PgStat_EntryRef::shared_entry, PgStat_LocalState::shared_hash, and PgStat_EntryRef::shared_stats.

Referenced by pgstat_drop_entry(), pgstat_gc_entry_refs(), pgstat_get_entry_ref(), and pgstat_release_matching_entry_refs().

◆ pgstat_release_matching_entry_refs()

static void pgstat_release_matching_entry_refs ( bool  discard_pending,
ReleaseMatchCB  match,
Datum  match_data 
)
static

Definition at line 694 of file pgstat_shmem.c.

696 {
697  pgstat_entry_ref_hash_iterator i;
699 
700  if (pgStatEntryRefHash == NULL)
701  return;
702 
703  pgstat_entry_ref_hash_start_iterate(pgStatEntryRefHash, &i);
704 
705  while ((ent = pgstat_entry_ref_hash_iterate(pgStatEntryRefHash, &i))
706  != NULL)
707  {
708  Assert(ent->entry_ref != NULL);
709 
710  if (match && !match(ent, match_data))
711  continue;
712 
713  pgstat_release_entry_ref(ent->key, ent->entry_ref, discard_pending);
714  }
715 }

References Assert, PgStat_EntryRefHashEntry::entry_ref, i, PgStat_EntryRefHashEntry::key, pgstat_release_entry_ref(), and pgStatEntryRefHash.

Referenced by pgstat_release_all_entry_refs(), and pgstat_release_db_entry_refs().

◆ pgstat_request_entry_refs_gc()

◆ pgstat_reset_entries_of_kind()

void pgstat_reset_entries_of_kind ( PgStat_Kind  kind,
TimestampTz  ts 
)

Definition at line 1014 of file pgstat_shmem.c.

1015 {
1017 }
static bool match_kind(PgStatShared_HashEntry *p, Datum match_data)
void pgstat_reset_matching_entries(bool(*do_reset)(PgStatShared_HashEntry *, Datum), Datum match_data, TimestampTz ts)
Definition: pgstat_shmem.c:978
static Datum Int32GetDatum(int32 X)
Definition: postgres.h:212

References Int32GetDatum(), match_kind(), and pgstat_reset_matching_entries().

Referenced by pgstat_reset_of_kind().

◆ pgstat_reset_entry()

void pgstat_reset_entry ( PgStat_Kind  kind,
Oid  dboid,
Oid  objoid,
TimestampTz  ts 
)

Definition at line 958 of file pgstat_shmem.c.

959 {
960  PgStat_EntryRef *entry_ref;
961 
962  Assert(!pgstat_get_kind_info(kind)->fixed_amount);
963 
964  entry_ref = pgstat_get_entry_ref(kind, dboid, objoid, false, NULL);
965  if (!entry_ref || entry_ref->shared_entry->dropped)
966  return;
967 
968  (void) pgstat_lock_entry(entry_ref, false);
969  shared_stat_reset_contents(kind, entry_ref->shared_stats, ts);
970  pgstat_unlock_entry(entry_ref);
971 }
static void shared_stat_reset_contents(PgStat_Kind kind, PgStatShared_Common *header, TimestampTz ts)
Definition: pgstat_shmem.c:942
void pgstat_unlock_entry(PgStat_EntryRef *entry_ref)
Definition: pgstat_shmem.c:613

References Assert, PgStatShared_HashEntry::dropped, pgstat_get_entry_ref(), pgstat_get_kind_info(), pgstat_lock_entry(), pgstat_unlock_entry(), PgStat_EntryRef::shared_entry, shared_stat_reset_contents(), and PgStat_EntryRef::shared_stats.

Referenced by pgstat_create_subscription(), and pgstat_reset().

◆ pgstat_reset_matching_entries()

void pgstat_reset_matching_entries ( bool(*)(PgStatShared_HashEntry *, Datum do_reset,
Datum  match_data,
TimestampTz  ts 
)

Definition at line 978 of file pgstat_shmem.c.

980 {
981  dshash_seq_status hstat;
983 
984  /* dshash entry is not modified, take shared lock */
985  dshash_seq_init(&hstat, pgStatLocal.shared_hash, false);
986  while ((p = dshash_seq_next(&hstat)) != NULL)
987  {
988  PgStatShared_Common *header;
989 
990  if (p->dropped)
991  continue;
992 
993  if (!do_reset(p, match_data))
994  continue;
995 
996  header = dsa_get_address(pgStatLocal.dsa, p->body);
997 
998  LWLockAcquire(&header->lock, LW_EXCLUSIVE);
999 
1000  shared_stat_reset_contents(p->key.kind, header, ts);
1001 
1002  LWLockRelease(&header->lock);
1003  }
1004  dshash_seq_term(&hstat);
1005 }
void LWLockRelease(LWLock *lock)
Definition: lwlock.c:1781

References PgStatShared_HashEntry::body, PgStatShared_HashEntry::dropped, PgStat_LocalState::dsa, dsa_get_address(), dshash_seq_init(), dshash_seq_next(), dshash_seq_term(), PgStatShared_HashEntry::key, PgStat_HashKey::kind, PgStatShared_Common::lock, LW_EXCLUSIVE, LWLockAcquire(), LWLockRelease(), pgStatLocal, PgStat_LocalState::shared_hash, and shared_stat_reset_contents().

Referenced by pgstat_reset_counters(), and pgstat_reset_entries_of_kind().

◆ pgstat_setup_memcxt()

static void pgstat_setup_memcxt ( void  )
static

Definition at line 1020 of file pgstat_shmem.c.

1021 {
1025  "PgStat Shared Ref",
1030  "PgStat Shared Ref Hash",
1032 }
#define unlikely(x)
Definition: c.h:311
#define AllocSetContextCreate
Definition: memutils.h:129
#define ALLOCSET_SMALL_SIZES
Definition: memutils.h:170
static MemoryContext pgStatEntryRefHashContext
Definition: pgstat_shmem.c:91

References ALLOCSET_SMALL_SIZES, AllocSetContextCreate, pgStatEntryRefHashContext, pgStatSharedRefContext, TopMemoryContext, and unlikely.

Referenced by pgstat_get_entry_ref().

◆ pgstat_setup_shared_refs()

static void pgstat_setup_shared_refs ( void  )
static

◆ pgstat_unlock_entry()

◆ shared_stat_reset_contents()

static void shared_stat_reset_contents ( PgStat_Kind  kind,
PgStatShared_Common header,
TimestampTz  ts 
)
static

Definition at line 942 of file pgstat_shmem.c.

944 {
945  const PgStat_KindInfo *kind_info = pgstat_get_kind_info(kind);
946 
947  memset(pgstat_get_entry_data(kind, header), 0,
948  pgstat_get_entry_len(kind));
949 
950  if (kind_info->reset_timestamp_cb)
951  kind_info->reset_timestamp_cb(header, ts);
952 }
void(* reset_timestamp_cb)(PgStatShared_Common *header, TimestampTz ts)

References pgstat_get_entry_data(), pgstat_get_entry_len(), pgstat_get_kind_info(), and PgStat_KindInfo::reset_timestamp_cb.

Referenced by pgstat_reset_entry(), and pgstat_reset_matching_entries().

◆ StatsShmemInit()

void StatsShmemInit ( void  )

Definition at line 141 of file pgstat_shmem.c.

142 {
143  bool found;
144  Size sz;
145 
146  sz = StatsShmemSize();
148  ShmemInitStruct("Shared Memory Stats", sz, &found);
149 
150  if (!IsUnderPostmaster)
151  {
152  dsa_area *dsa;
153  dshash_table *dsh;
155  char *p = (char *) ctl;
156 
157  Assert(!found);
158 
159  /* the allocation of pgStatLocal.shmem itself */
160  p += MAXALIGN(sizeof(PgStat_ShmemControl));
161 
162  /*
163  * Create a small dsa allocation in plain shared memory. This is
164  * required because postmaster cannot use dsm segments. It also
165  * provides a small efficiency win.
166  */
167  ctl->raw_dsa_area = p;
169  dsa = dsa_create_in_place(ctl->raw_dsa_area,
172  dsa_pin(dsa);
173 
174  /*
175  * To ensure dshash is created in "plain" shared memory, temporarily
176  * limit size of dsa to the initial size of the dsa.
177  */
179 
180  /*
181  * With the limit in place, create the dshash table. XXX: It'd be nice
182  * if there were dshash_create_in_place().
183  */
184  dsh = dshash_create(dsa, &dsh_params, NULL);
185  ctl->hash_handle = dshash_get_hash_table_handle(dsh);
186 
187  /* lift limit set above */
188  dsa_set_size_limit(dsa, -1);
189 
190  /*
191  * Postmaster will never access these again, thus free the local
192  * dsa/dshash references.
193  */
194  dshash_detach(dsh);
195  dsa_detach(dsa);
196 
197  pg_atomic_init_u64(&ctl->gc_request_count, 1);
198 
199  /* initialize fixed-numbered stats */
200  for (int kind = PGSTAT_KIND_FIRST_VALID; kind <= PGSTAT_KIND_LAST; kind++)
201  {
202  const PgStat_KindInfo *kind_info = pgstat_get_kind_info(kind);
203  char *ptr;
204 
205  if (!kind_info->fixed_amount)
206  continue;
207 
208  ptr = ((char *) ctl) + kind_info->shared_ctl_off;
209  kind_info->init_shmem_cb(ptr);
210  }
211  }
212  else
213  {
214  Assert(found);
215  }
216 }
static void pg_atomic_init_u64(volatile pg_atomic_uint64 *ptr, uint64 val)
Definition: atomics.h:446
void dsa_set_size_limit(dsa_area *area, size_t limit)
Definition: dsa.c:1018
void dsa_pin(dsa_area *area)
Definition: dsa.c:975
#define dsa_create_in_place(place, size, tranch_id, segment)
Definition: dsa.h:122
dshash_table_handle dshash_get_hash_table_handle(dshash_table *hash_table)
Definition: dshash.c:367
dshash_table * dshash_create(dsa_area *area, const dshash_parameters *params, void *arg)
Definition: dshash.c:206
bool IsUnderPostmaster
Definition: globals.c:118
@ LWTRANCHE_PGSTATS_DSA
Definition: lwlock.h:203
#define PGSTAT_KIND_LAST
Definition: pgstat.h:57
#define PGSTAT_KIND_FIRST_VALID
Definition: pgstat.h:56
static Size pgstat_dsa_init_size(void)
Definition: pgstat_shmem.c:106
Size StatsShmemSize(void)
Definition: pgstat_shmem.c:127
tree ctl
Definition: radixtree.h:1853
void * ShmemInitStruct(const char *name, Size size, bool *foundPtr)
Definition: shmem.c:387
void(* init_shmem_cb)(void *stats)
Definition: dsa.c:348

References Assert, ctl, dsa_create_in_place, dsa_detach(), dsa_pin(), dsa_set_size_limit(), dsh_params, dshash_create(), dshash_detach(), dshash_get_hash_table_handle(), PgStat_KindInfo::fixed_amount, PgStat_KindInfo::init_shmem_cb, IsUnderPostmaster, LWTRANCHE_PGSTATS_DSA, MAXALIGN, pg_atomic_init_u64(), pgstat_dsa_init_size(), pgstat_get_kind_info(), PGSTAT_KIND_FIRST_VALID, PGSTAT_KIND_LAST, pgStatLocal, PgStat_KindInfo::shared_ctl_off, PgStat_LocalState::shmem, ShmemInitStruct(), and StatsShmemSize().

Referenced by CreateOrAttachShmemStructs().

◆ StatsShmemSize()

Size StatsShmemSize ( void  )

Definition at line 127 of file pgstat_shmem.c.

128 {
129  Size sz;
130 
131  sz = MAXALIGN(sizeof(PgStat_ShmemControl));
132  sz = add_size(sz, pgstat_dsa_init_size());
133 
134  return sz;
135 }
Size add_size(Size s1, Size s2)
Definition: shmem.c:493

References add_size(), MAXALIGN, and pgstat_dsa_init_size().

Referenced by CalculateShmemSize(), and StatsShmemInit().

Variable Documentation

◆ dsh_params

const dshash_parameters dsh_params
static
Initial value:
= {
sizeof(PgStat_HashKey),
}
void dshash_memcpy(void *dest, const void *src, size_t size, void *arg)
Definition: dshash.c:590
@ LWTRANCHE_PGSTATS_HASH
Definition: lwlock.h:204
static uint32 pgstat_hash_hash_key(const void *d, size_t size, void *arg)
static int pgstat_cmp_hash_key(const void *a, const void *b, size_t size, void *arg)
struct PgStat_HashKey PgStat_HashKey

Definition at line 62 of file pgstat_shmem.c.

Referenced by pgstat_attach_shmem(), and StatsShmemInit().

◆ pgStatEntryRefHash

◆ pgStatEntryRefHashContext

MemoryContext pgStatEntryRefHashContext = NULL
static

Definition at line 91 of file pgstat_shmem.c.

Referenced by pgstat_setup_memcxt(), and pgstat_setup_shared_refs().

◆ pgStatSharedRefAge

int pgStatSharedRefAge = 0
static

◆ pgStatSharedRefContext

MemoryContext pgStatSharedRefContext = NULL
static

Definition at line 90 of file pgstat_shmem.c.

Referenced by pgstat_get_entry_ref_cached(), and pgstat_setup_memcxt().