PostgreSQL Source Code: src/backend/catalog/pg_db_role_setting.c Source File

PostgreSQL Source Code git master
pg_db_role_setting.c
Go to the documentation of this file.
1/*
2 * pg_db_role_setting.c
3 * Routines to support manipulation of the pg_db_role_setting relation
4 *
5 * Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group
6 * Portions Copyright (c) 1994, Regents of the University of California
7 *
8 * IDENTIFICATION
9 * src/backend/catalog/pg_db_role_setting.c
10 */
11#include "postgres.h"
12
13#include "access/genam.h"
14#include "access/heapam.h"
15#include "access/htup_details.h"
16#include "access/tableam.h"
17#include "catalog/indexing.h"
18#include "catalog/objectaccess.h"
19#include "catalog/pg_db_role_setting.h"
20#include "utils/fmgroids.h"
21#include "utils/rel.h"
22
23void
24 AlterSetting(Oid databaseid, Oid roleid, VariableSetStmt *setstmt)
25{
26 char *valuestr;
27 HeapTuple tuple;
28 Relation rel;
29 ScanKeyData scankey[2];
30 SysScanDesc scan;
31
32 valuestr = ExtractSetVariableArgs(setstmt);
33
34 /* Get the old tuple, if any. */
35
36 rel = table_open(DbRoleSettingRelationId, RowExclusiveLock);
37 ScanKeyInit(&scankey[0],
38 Anum_pg_db_role_setting_setdatabase,
39 BTEqualStrategyNumber, F_OIDEQ,
40 ObjectIdGetDatum(databaseid));
41 ScanKeyInit(&scankey[1],
42 Anum_pg_db_role_setting_setrole,
43 BTEqualStrategyNumber, F_OIDEQ,
44 ObjectIdGetDatum(roleid));
45 scan = systable_beginscan(rel, DbRoleSettingDatidRolidIndexId, true,
46 NULL, 2, scankey);
47 tuple = systable_getnext(scan);
48
49 /*
50 * There are three cases:
51 *
52 * - in RESET ALL, request GUC to reset the settings array and update the
53 * catalog if there's anything left, delete it otherwise
54 *
55 * - in other commands, if there's a tuple in pg_db_role_setting, update
56 * it; if it ends up empty, delete it
57 *
58 * - otherwise, insert a new pg_db_role_setting tuple, but only if the
59 * command is not RESET
60 */
61 if (setstmt->kind == VAR_RESET_ALL)
62 {
63 if (HeapTupleIsValid(tuple))
64 {
65 ArrayType *new = NULL;
66 Datum datum;
67 bool isnull;
68
69 datum = heap_getattr(tuple, Anum_pg_db_role_setting_setconfig,
70 RelationGetDescr(rel), &isnull);
71
72 if (!isnull)
73 new = GUCArrayReset(DatumGetArrayTypeP(datum));
74
75 if (new)
76 {
77 Datum repl_val[Natts_pg_db_role_setting];
78 bool repl_null[Natts_pg_db_role_setting];
79 bool repl_repl[Natts_pg_db_role_setting];
80 HeapTuple newtuple;
81
82 memset(repl_repl, false, sizeof(repl_repl));
83
84 repl_val[Anum_pg_db_role_setting_setconfig - 1] =
85 PointerGetDatum(new);
86 repl_repl[Anum_pg_db_role_setting_setconfig - 1] = true;
87 repl_null[Anum_pg_db_role_setting_setconfig - 1] = false;
88
89 newtuple = heap_modify_tuple(tuple, RelationGetDescr(rel),
90 repl_val, repl_null, repl_repl);
91 CatalogTupleUpdate(rel, &tuple->t_self, newtuple);
92 }
93 else
94 CatalogTupleDelete(rel, &tuple->t_self);
95 }
96 }
97 else if (HeapTupleIsValid(tuple))
98 {
99 Datum repl_val[Natts_pg_db_role_setting];
100 bool repl_null[Natts_pg_db_role_setting];
101 bool repl_repl[Natts_pg_db_role_setting];
102 HeapTuple newtuple;
103 Datum datum;
104 bool isnull;
105 ArrayType *a;
106
107 memset(repl_repl, false, sizeof(repl_repl));
108 repl_repl[Anum_pg_db_role_setting_setconfig - 1] = true;
109 repl_null[Anum_pg_db_role_setting_setconfig - 1] = false;
110
111 /* Extract old value of setconfig */
112 datum = heap_getattr(tuple, Anum_pg_db_role_setting_setconfig,
113 RelationGetDescr(rel), &isnull);
114 a = isnull ? NULL : DatumGetArrayTypeP(datum);
115
116 /* Update (valuestr is NULL in RESET cases) */
117 if (valuestr)
118 a = GUCArrayAdd(a, setstmt->name, valuestr);
119 else
120 a = GUCArrayDelete(a, setstmt->name);
121
122 if (a)
123 {
124 repl_val[Anum_pg_db_role_setting_setconfig - 1] =
125 PointerGetDatum(a);
126
127 newtuple = heap_modify_tuple(tuple, RelationGetDescr(rel),
128 repl_val, repl_null, repl_repl);
129 CatalogTupleUpdate(rel, &tuple->t_self, newtuple);
130 }
131 else
132 CatalogTupleDelete(rel, &tuple->t_self);
133 }
134 else if (valuestr)
135 {
136 /* non-null valuestr means it's not RESET, so insert a new tuple */
137 HeapTuple newtuple;
138 Datum values[Natts_pg_db_role_setting];
139 bool nulls[Natts_pg_db_role_setting];
140 ArrayType *a;
141
142 memset(nulls, false, sizeof(nulls));
143
144 a = GUCArrayAdd(NULL, setstmt->name, valuestr);
145
146 values[Anum_pg_db_role_setting_setdatabase - 1] =
147 ObjectIdGetDatum(databaseid);
148 values[Anum_pg_db_role_setting_setrole - 1] = ObjectIdGetDatum(roleid);
149 values[Anum_pg_db_role_setting_setconfig - 1] = PointerGetDatum(a);
150 newtuple = heap_form_tuple(RelationGetDescr(rel), values, nulls);
151
152 CatalogTupleInsert(rel, newtuple);
153 }
154 else
155 {
156 /*
157 * RESET doesn't need to change any state if there's no pre-existing
158 * pg_db_role_setting entry, but for consistency we should still check
159 * that the option is valid and we're allowed to set it.
160 */
161 (void) GUCArrayDelete(NULL, setstmt->name);
162 }
163
164 InvokeObjectPostAlterHookArg(DbRoleSettingRelationId,
165 databaseid, 0, roleid, false);
166
167 systable_endscan(scan);
168
169 /* Close pg_db_role_setting, but keep lock till commit */
170 table_close(rel, NoLock);
171}
172
173/*
174 * Drop some settings from the catalog. These can be for a particular
175 * database, or for a particular role. (It is of course possible to do both
176 * too, but it doesn't make sense for current uses.)
177 */
178void
179 DropSetting(Oid databaseid, Oid roleid)
180{
181 Relation relsetting;
182 TableScanDesc scan;
183 ScanKeyData keys[2];
184 HeapTuple tup;
185 int numkeys = 0;
186
187 relsetting = table_open(DbRoleSettingRelationId, RowExclusiveLock);
188
189 if (OidIsValid(databaseid))
190 {
191 ScanKeyInit(&keys[numkeys],
192 Anum_pg_db_role_setting_setdatabase,
193 BTEqualStrategyNumber,
194 F_OIDEQ,
195 ObjectIdGetDatum(databaseid));
196 numkeys++;
197 }
198 if (OidIsValid(roleid))
199 {
200 ScanKeyInit(&keys[numkeys],
201 Anum_pg_db_role_setting_setrole,
202 BTEqualStrategyNumber,
203 F_OIDEQ,
204 ObjectIdGetDatum(roleid));
205 numkeys++;
206 }
207
208 scan = table_beginscan_catalog(relsetting, numkeys, keys);
209 while (HeapTupleIsValid(tup = heap_getnext(scan, ForwardScanDirection)))
210 {
211 CatalogTupleDelete(relsetting, &tup->t_self);
212 }
213 table_endscan(scan);
214
215 table_close(relsetting, RowExclusiveLock);
216}
217
218/*
219 * Scan pg_db_role_setting looking for applicable settings, and load them on
220 * the current process.
221 *
222 * relsetting is pg_db_role_setting, already opened and locked.
223 *
224 * Note: we only consider setting for the exact databaseid/roleid combination.
225 * This probably needs to be called more than once, with InvalidOid passed as
226 * databaseid/roleid.
227 */
228void
229 ApplySetting(Snapshot snapshot, Oid databaseid, Oid roleid,
230 Relation relsetting, GucSource source)
231{
232 SysScanDesc scan;
233 ScanKeyData keys[2];
234 HeapTuple tup;
235
236 ScanKeyInit(&keys[0],
237 Anum_pg_db_role_setting_setdatabase,
238 BTEqualStrategyNumber,
239 F_OIDEQ,
240 ObjectIdGetDatum(databaseid));
241 ScanKeyInit(&keys[1],
242 Anum_pg_db_role_setting_setrole,
243 BTEqualStrategyNumber,
244 F_OIDEQ,
245 ObjectIdGetDatum(roleid));
246
247 scan = systable_beginscan(relsetting, DbRoleSettingDatidRolidIndexId, true,
248 snapshot, 2, keys);
249 while (HeapTupleIsValid(tup = systable_getnext(scan)))
250 {
251 bool isnull;
252 Datum datum;
253
254 datum = heap_getattr(tup, Anum_pg_db_role_setting_setconfig,
255 RelationGetDescr(relsetting), &isnull);
256 if (!isnull)
257 {
258 ArrayType *a = DatumGetArrayTypeP(datum);
259
260 /*
261 * We process all the options at SUSET level. We assume that the
262 * right to insert an option into pg_db_role_setting was checked
263 * when it was inserted.
264 */
265 ProcessGUCArray(a, PGC_SUSET, source, GUC_ACTION_SET);
266 }
267 }
268
269 systable_endscan(scan);
270}
#define DatumGetArrayTypeP(X)
Definition: array.h:261
static Datum values[MAXATTR]
Definition: bootstrap.c:153
#define OidIsValid(objectId)
Definition: c.h:774
void systable_endscan(SysScanDesc sysscan)
Definition: genam.c:603
HeapTuple systable_getnext(SysScanDesc sysscan)
Definition: genam.c:514
SysScanDesc systable_beginscan(Relation heapRelation, Oid indexId, bool indexOK, Snapshot snapshot, int nkeys, ScanKey key)
Definition: genam.c:388
ArrayType * GUCArrayAdd(ArrayType *array, const char *name, const char *value)
Definition: guc.c:6510
void ProcessGUCArray(ArrayType *array, GucContext context, GucSource source, GucAction action)
Definition: guc.c:6478
ArrayType * GUCArrayReset(ArrayType *array)
Definition: guc.c:6658
ArrayType * GUCArrayDelete(ArrayType *array, const char *name)
Definition: guc.c:6588
@ GUC_ACTION_SET
Definition: guc.h:203
GucSource
Definition: guc.h:112
@ PGC_SUSET
Definition: guc.h:78
char * ExtractSetVariableArgs(VariableSetStmt *stmt)
Definition: guc_funcs.c:167
HeapTuple heap_getnext(TableScanDesc sscan, ScanDirection direction)
Definition: heapam.c:1346
HeapTuple heap_modify_tuple(HeapTuple tuple, TupleDesc tupleDesc, const Datum *replValues, const bool *replIsnull, const bool *doReplace)
Definition: heaptuple.c:1210
HeapTuple heap_form_tuple(TupleDesc tupleDescriptor, const Datum *values, const bool *isnull)
Definition: heaptuple.c:1117
#define HeapTupleIsValid(tuple)
Definition: htup.h:78
static Datum heap_getattr(HeapTuple tup, int attnum, TupleDesc tupleDesc, bool *isnull)
Definition: htup_details.h:904
void CatalogTupleUpdate(Relation heapRel, ItemPointer otid, HeapTuple tup)
Definition: indexing.c:313
void CatalogTupleInsert(Relation heapRel, HeapTuple tup)
Definition: indexing.c:233
void CatalogTupleDelete(Relation heapRel, ItemPointer tid)
Definition: indexing.c:365
a
int a
Definition: isn.c:73
#define NoLock
Definition: lockdefs.h:34
#define RowExclusiveLock
Definition: lockdefs.h:38
#define InvokeObjectPostAlterHookArg(classId, objectId, subId, auxiliaryId, is_internal)
Definition: objectaccess.h:200
@ VAR_RESET_ALL
Definition: parsenodes.h:2699
void DropSetting(Oid databaseid, Oid roleid)
void AlterSetting(Oid databaseid, Oid roleid, VariableSetStmt *setstmt)
void ApplySetting(Snapshot snapshot, Oid databaseid, Oid roleid, Relation relsetting, GucSource source)
static rewind_source * source
Definition: pg_rewind.c:89
static Datum PointerGetDatum(const void *X)
Definition: postgres.h:332
static Datum ObjectIdGetDatum(Oid X)
Definition: postgres.h:262
uint64_t Datum
Definition: postgres.h:70
unsigned int Oid
Definition: postgres_ext.h:32
#define RelationGetDescr(relation)
Definition: rel.h:540
void ScanKeyInit(ScanKey entry, AttrNumber attributeNumber, StrategyNumber strategy, RegProcedure procedure, Datum argument)
Definition: scankey.c:76
@ ForwardScanDirection
Definition: sdir.h:28
#define BTEqualStrategyNumber
Definition: stratnum.h:31
Definition: array.h:93
ItemPointerData t_self
Definition: htup.h:65
Definition: rel.h:56
Definition: skey.h:65
VariableSetKind kind
Definition: parsenodes.h:2707
void table_close(Relation relation, LOCKMODE lockmode)
Definition: table.c:126
Relation table_open(Oid relationId, LOCKMODE lockmode)
Definition: table.c:40
TableScanDesc table_beginscan_catalog(Relation relation, int nkeys, ScanKeyData *key)
Definition: tableam.c:113
static void table_endscan(TableScanDesc scan)
Definition: tableam.h:985

AltStyle によって変換されたページ (->オリジナル) /