PostgreSQL Source Code  git master
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros
evtcache.c
Go to the documentation of this file.
1 /*-------------------------------------------------------------------------
2  *
3  * evtcache.c
4  * Special-purpose cache for event trigger data.
5  *
6  * Portions Copyright (c) 1996-2017, PostgreSQL Global Development Group
7  * Portions Copyright (c) 1994, Regents of the University of California
8  *
9  * IDENTIFICATION
10  * src/backend/utils/cache/evtcache.c
11  *
12  *-------------------------------------------------------------------------
13  */
14 #include "postgres.h"
15 
16 #include "access/genam.h"
17 #include "access/heapam.h"
18 #include "access/htup_details.h"
20 #include "catalog/indexing.h"
21 #include "catalog/pg_type.h"
22 #include "commands/trigger.h"
23 #include "utils/array.h"
24 #include "utils/builtins.h"
25 #include "utils/catcache.h"
26 #include "utils/evtcache.h"
27 #include "utils/inval.h"
28 #include "utils/memutils.h"
29 #include "utils/hsearch.h"
30 #include "utils/rel.h"
31 #include "utils/snapmgr.h"
32 #include "utils/syscache.h"
33 
34 typedef enum
35 {
40 
41 typedef struct
42 {
46 
50 
51 static void BuildEventTriggerCache(void);
53  int cacheid, uint32 hashvalue);
54 static int DecodeTextArrayToCString(Datum array, char ***cstringp);
55 
56 /*
57  * Search the event cache by trigger event.
58  *
59  * Note that the caller had better copy any data it wants to keep around
60  * across any operation that might touch a system catalog into some other
61  * memory context, since a cache reset could blow the return value away.
62  */
63 List *
65 {
67 
70  entry = hash_search(EventTriggerCache, &event, HASH_FIND, NULL);
71  return entry != NULL ? entry->triggerlist : NIL;
72 }
73 
74 /*
75  * Rebuild the event trigger cache.
76  */
77 static void
79 {
80  HASHCTL ctl;
81  HTAB *cache;
82  MemoryContext oldcontext;
83  Relation rel;
84  Relation irel;
85  SysScanDesc scan;
86 
87  if (EventTriggerCacheContext != NULL)
88  {
89  /*
90  * Free up any memory already allocated in EventTriggerCacheContext.
91  * This can happen either because a previous rebuild failed, or
92  * because an invalidation happened before the rebuild was complete.
93  */
94  MemoryContextResetAndDeleteChildren(EventTriggerCacheContext);
95  }
96  else
97  {
98  /*
99  * This is our first time attempting to build the cache, so we need to
100  * set up the memory context and register a syscache callback to
101  * capture future invalidation events.
102  */
103  if (CacheMemoryContext == NULL)
105  EventTriggerCacheContext =
107  "EventTriggerCache",
111  (Datum) 0);
112  }
113 
114  /* Switch to correct memory context. */
115  oldcontext = MemoryContextSwitchTo(EventTriggerCacheContext);
116 
117  /* Prevent the memory context from being nuked while we're rebuilding. */
119 
120  /* Create new hash table. */
121  MemSet(&ctl, 0, sizeof(ctl));
122  ctl.keysize = sizeof(EventTriggerEvent);
123  ctl.entrysize = sizeof(EventTriggerCacheEntry);
125  cache = hash_create("Event Trigger Cache", 32, &ctl,
127 
128  /*
129  * Prepare to scan pg_event_trigger in name order.
130  */
133  scan = systable_beginscan_ordered(rel, irel, NULL, 0, NULL);
134 
135  /*
136  * Build a cache item for each pg_event_trigger tuple, and append each one
137  * to the appropriate cache entry.
138  */
139  for (;;)
140  {
141  HeapTuple tup;
143  char *evtevent;
144  EventTriggerEvent event;
145  EventTriggerCacheItem *item;
146  Datum evttags;
147  bool evttags_isnull;
148  EventTriggerCacheEntry *entry;
149  bool found;
150 
151  /* Get next tuple. */
153  if (!HeapTupleIsValid(tup))
154  break;
155 
156  /* Skip trigger if disabled. */
157  form = (Form_pg_event_trigger) GETSTRUCT(tup);
158  if (form->evtenabled == TRIGGER_DISABLED)
159  continue;
160 
161  /* Decode event name. */
162  evtevent = NameStr(form->evtevent);
163  if (strcmp(evtevent, "ddl_command_start") == 0)
164  event = EVT_DDLCommandStart;
165  else if (strcmp(evtevent, "ddl_command_end") == 0)
166  event = EVT_DDLCommandEnd;
167  else if (strcmp(evtevent, "sql_drop") == 0)
168  event = EVT_SQLDrop;
169  else if (strcmp(evtevent, "table_rewrite") == 0)
170  event = EVT_TableRewrite;
171  else
172  continue;
173 
174  /* Allocate new cache item. */
175  item = palloc0(sizeof(EventTriggerCacheItem));
176  item->fnoid = form->evtfoid;
177  item->enabled = form->evtenabled;
178 
179  /* Decode and sort tags array. */
181  RelationGetDescr(rel), &evttags_isnull);
182  if (!evttags_isnull)
183  {
184  item->ntags = DecodeTextArrayToCString(evttags, &item->tag);
185  qsort(item->tag, item->ntags, sizeof(char *), pg_qsort_strcmp);
186  }
187 
188  /* Add to cache entry. */
189  entry = hash_search(cache, &event, HASH_ENTER, &found);
190  if (found)
191  entry->triggerlist = lappend(entry->triggerlist, item);
192  else
193  entry->triggerlist = list_make1(item);
194  }
195 
196  /* Done with pg_event_trigger scan. */
200 
201  /* Restore previous memory context. */
202  MemoryContextSwitchTo(oldcontext);
203 
204  /* Install new cache. */
205  EventTriggerCache = cache;
206 
207  /*
208  * If the cache has been invalidated since we entered this routine, we
209  * still use and return the cache we just finished constructing, to avoid
210  * infinite loops, but we leave the cache marked stale so that we'll
211  * rebuild it again on next access. Otherwise, we mark the cache valid.
212  */
215 }
216 
217 /*
218  * Decode text[] to an array of C strings.
219  *
220  * We could avoid a bit of overhead here if we were willing to duplicate some
221  * of the logic from deconstruct_array, but it doesn't seem worth the code
222  * complexity.
223  */
224 static int
225 DecodeTextArrayToCString(Datum array, char ***cstringp)
226 {
227  ArrayType *arr = DatumGetArrayTypeP(array);
228  Datum *elems;
229  char **cstring;
230  int i;
231  int nelems;
232 
233  if (ARR_NDIM(arr) != 1 || ARR_HASNULL(arr) || ARR_ELEMTYPE(arr) != TEXTOID)
234  elog(ERROR, "expected 1-D text array");
235  deconstruct_array(arr, TEXTOID, -1, false, 'i', &elems, NULL, &nelems);
236 
237  cstring = palloc(nelems * sizeof(char *));
238  for (i = 0; i < nelems; ++i)
239  cstring[i] = TextDatumGetCString(elems[i]);
240 
241  pfree(elems);
242  *cstringp = cstring;
243  return nelems;
244 }
245 
246 /*
247  * Flush all cache entries when pg_event_trigger is updated.
248  *
249  * This should be rare enough that we don't need to be very granular about
250  * it, so we just blow away everything, which also avoids the possibility of
251  * memory leaks.
252  */
253 static void
255 {
256  /*
257  * If the cache isn't valid, then there might be a rebuild in progress, so
258  * we can't immediately blow it away. But it's advantageous to do this
259  * when possible, so as to immediately free memory.
260  */
262  {
263  MemoryContextResetAndDeleteChildren(EventTriggerCacheContext);
264  EventTriggerCache = NULL;
265  }
266 
267  /* Mark cache for rebuild. */
269 }
#define NIL
Definition: pg_list.h:69
static void InvalidateEventCacheCallback(Datum arg, int cacheid, uint32 hashvalue)
Definition: evtcache.c:254
static HTAB * EventTriggerCache
Definition: evtcache.c:47
#define GETSTRUCT(TUP)
Definition: htup_details.h:656
#define HASH_CONTEXT
Definition: hsearch.h:93
#define HASH_ELEM
Definition: hsearch.h:87
MemoryContext hcxt
Definition: hsearch.h:78
#define RelationGetDescr(relation)
Definition: rel.h:428
#define TEXTOID
Definition: pg_type.h:324
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
Definition: palloc.h:109
static int DecodeTextArrayToCString(Datum array, char ***cstringp)
Definition: evtcache.c:225
#define AccessShareLock
Definition: lockdefs.h:36
Size entrysize
Definition: hsearch.h:73
static MemoryContext EventTriggerCacheContext
Definition: evtcache.c:48
void relation_close(Relation relation, LOCKMODE lockmode)
Definition: heapam.c:1266
#define MemSet(start, val, len)
Definition: c.h:863
void * hash_search(HTAB *hashp, const void *keyPtr, HASHACTION action, bool *foundPtr)
Definition: dynahash.c:902
HeapTuple systable_getnext_ordered(SysScanDesc sysscan, ScanDirection direction)
Definition: genam.c:597
EventTriggerEvent
Definition: evtcache.h:19
#define list_make1(x1)
Definition: pg_list.h:139
Definition: dynahash.c:208
void pfree(void *pointer)
Definition: mcxt.c:949
#define ERROR
Definition: elog.h:43
#define ALLOCSET_DEFAULT_SIZES
Definition: memutils.h:165
static void BuildEventTriggerCache(void)
Definition: evtcache.c:78
unsigned int uint32
Definition: c.h:258
#define ARR_HASNULL(a)
Definition: array.h:276
static EventTriggerCacheStateType EventTriggerCacheState
Definition: evtcache.c:49
List * lappend(List *list, void *datum)
Definition: list.c:128
#define heap_getattr(tup, attnum, tupleDesc, isnull)
Definition: htup_details.h:769
#define MemoryContextResetAndDeleteChildren(ctx)
Definition: memutils.h:67
#define HASH_BLOBS
Definition: hsearch.h:88
#define TextDatumGetCString(d)
Definition: builtins.h:92
MemoryContext AllocSetContextCreate(MemoryContext parent, const char *name, Size minContextSize, Size initBlockSize, Size maxBlockSize)
Definition: aset.c:322
void CacheRegisterSyscacheCallback(int cacheid, SyscacheCallbackFunction func, Datum arg)
Definition: inval.c:1389
void * palloc0(Size size)
Definition: mcxt.c:877
#define TRIGGER_DISABLED
Definition: trigger.h:158
HTAB * hash_create(const char *tabname, long nelem, HASHCTL *info, int flags)
Definition: dynahash.c:316
uintptr_t Datum
Definition: postgres.h:372
EventTriggerCacheStateType
Definition: evtcache.c:34
void systable_endscan_ordered(SysScanDesc sysscan)
Definition: genam.c:614
Size keysize
Definition: hsearch.h:72
EventTriggerEvent event
Definition: evtcache.c:43
FormData_pg_event_trigger * Form_pg_event_trigger
#define HeapTupleIsValid(tuple)
Definition: htup.h:77
#define EventTriggerRelationId
#define Anum_pg_event_trigger_evttags
void CreateCacheMemoryContext(void)
Definition: catcache.c:636
int pg_qsort_strcmp(const void *a, const void *b)
Definition: qsort.c:232
#define ARR_NDIM(a)
Definition: array.h:275
#define EventTriggerNameIndexId
Definition: indexing.h:254
void index_close(Relation relation, LOCKMODE lockmode)
Definition: indexam.c:176
void deconstruct_array(ArrayType *array, Oid elmtype, int elmlen, bool elmbyval, char elmalign, Datum **elemsp, bool **nullsp, int *nelemsp)
Definition: arrayfuncs.c:3449
void * palloc(Size size)
Definition: mcxt.c:848
SysScanDesc systable_beginscan_ordered(Relation heapRelation, Relation indexRelation, Snapshot snapshot, int nkeys, ScanKey key)
Definition: genam.c:533
int i
List * EventCacheLookup(EventTriggerEvent event)
Definition: evtcache.c:64
#define NameStr(name)
Definition: c.h:493
void * arg
Relation relation_open(Oid relationId, LOCKMODE lockmode)
Definition: heapam.c:1120
#define elog
Definition: elog.h:219
#define qsort(a, b, c, d)
Definition: port.h:447
Definition: pg_list.h:45
#define ARR_ELEMTYPE(a)
Definition: array.h:277
Relation index_open(Oid relationId, LOCKMODE lockmode)
Definition: indexam.c:151
MemoryContext CacheMemoryContext
Definition: mcxt.c:46
#define DatumGetArrayTypeP(X)
Definition: array.h:246