mirror of https://github.com/postgres/postgres
real small function to revoke update on a column. The function > > doesn't do anything > > fancy like checking user ids. > > > > I copied most of it from the refint.c in the contrib directory. > > > > Should I post this somewhere? It really isn't very big. > > Here it is... -- | Email - rick@rpacorp.com Rick Poleshuck | Voice - (908) 653-1070 Fax - (908) 653-0265 | Mail - RPA Corporation | - 308 Elizabeth Avenue, Cranford, New Jersey 07016pull/50/head
parent
6e771e8dc2
commit
fd262dac8c
@ -0,0 +1,111 @@ |
|||||||
|
/*
|
||||||
|
* noup.c -- functions to remove update permission from a column |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "executor/spi.h" /* this is what you need to work with SPI */ |
||||||
|
#include "commands/trigger.h" /* -"- and triggers */ |
||||||
|
#include <ctype.h> /* tolower () */ |
||||||
|
|
||||||
|
HeapTuple noup(void); |
||||||
|
|
||||||
|
/*
|
||||||
|
* noup () -- revoke permission on column |
||||||
|
* |
||||||
|
* Though it's called without args You have to specify referenced |
||||||
|
* table/column while creating trigger:
|
||||||
|
* EXECUTE PROCEDURE noup ('col'). |
||||||
|
*/ |
||||||
|
|
||||||
|
HeapTuple /* have to return HeapTuple to Executor */ |
||||||
|
noup() |
||||||
|
{ |
||||||
|
Trigger *trigger; /* to get trigger name */ |
||||||
|
int nargs; /* # of args specified in CREATE TRIGGER */ |
||||||
|
char **args; /* arguments: column names and table name */ |
||||||
|
int nkeys; /* # of key columns (= nargs / 2) */ |
||||||
|
Datum *kvals; /* key values */ |
||||||
|
Relation rel; /* triggered relation */ |
||||||
|
HeapTuple tuple = NULL; /* tuple to return */ |
||||||
|
TupleDesc tupdesc; /* tuple description */ |
||||||
|
bool isnull; /* to know is some column NULL or not */ |
||||||
|
int ret; |
||||||
|
int i; |
||||||
|
|
||||||
|
/*
|
||||||
|
* Some checks first... |
||||||
|
*/ |
||||||
|
|
||||||
|
/* Called by trigger manager ? */ |
||||||
|
if (!CurrentTriggerData) |
||||||
|
elog(WARN, "noup: triggers are not initialized"); |
||||||
|
|
||||||
|
/* Should be called for ROW trigger */ |
||||||
|
if (TRIGGER_FIRED_FOR_STATEMENT(CurrentTriggerData->tg_event)) |
||||||
|
elog(WARN, "noup: can't process STATEMENT events"); |
||||||
|
|
||||||
|
/* Not should be called for INSERT */ |
||||||
|
if (TRIGGER_FIRED_BY_INSERT(CurrentTriggerData->tg_event)) |
||||||
|
elog(WARN, "noup: can't process INSERT events"); |
||||||
|
|
||||||
|
/* Not should be called for DELETE */ |
||||||
|
else if (TRIGGER_FIRED_BY_DELETE(CurrentTriggerData->tg_event)) |
||||||
|
elog(WARN, "noup: can't process DELETE events"); |
||||||
|
|
||||||
|
/* check new Tuple */ |
||||||
|
tuple = CurrentTriggerData->tg_newtuple; |
||||||
|
|
||||||
|
trigger = CurrentTriggerData->tg_trigger; |
||||||
|
nargs = trigger->tgnargs; |
||||||
|
args = trigger->tgargs; |
||||||
|
|
||||||
|
nkeys = nargs; |
||||||
|
rel = CurrentTriggerData->tg_relation; |
||||||
|
tupdesc = rel->rd_att; |
||||||
|
|
||||||
|
/*
|
||||||
|
* Setting CurrentTriggerData to NULL prevents direct calls to trigger |
||||||
|
* functions in queries. Normally, trigger functions have to be called |
||||||
|
* by trigger manager code only. |
||||||
|
*/ |
||||||
|
CurrentTriggerData = NULL; |
||||||
|
|
||||||
|
/* Connect to SPI manager */ |
||||||
|
if ((ret = SPI_connect()) < 0) |
||||||
|
elog(WARN, "noup: SPI_connect returned %d", ret); |
||||||
|
|
||||||
|
/*
|
||||||
|
* We use SPI plan preparation feature, so allocate space to place key |
||||||
|
* values. |
||||||
|
*/ |
||||||
|
kvals = (Datum *) palloc(nkeys * sizeof(Datum)); |
||||||
|
|
||||||
|
/* For each column in key ... */ |
||||||
|
for (i = 0; i < nkeys; i++) |
||||||
|
{ |
||||||
|
/* get index of column in tuple */ |
||||||
|
int fnumber = SPI_fnumber(tupdesc, args[i]); |
||||||
|
|
||||||
|
/* Bad guys may give us un-existing column in CREATE TRIGGER */ |
||||||
|
if (fnumber < 0) |
||||||
|
elog(WARN, "noup: there is no attribute %s in relation %s", |
||||||
|
args[i], SPI_getrelname(rel)); |
||||||
|
|
||||||
|
/* Well, get binary (in internal format) value of column */ |
||||||
|
kvals[i] = SPI_getbinval(tuple, tupdesc, fnumber, &isnull); |
||||||
|
|
||||||
|
/*
|
||||||
|
* If it's NOT NULL then cancel update |
||||||
|
*/ |
||||||
|
if (!isnull) |
||||||
|
{ |
||||||
|
|
||||||
|
elog(WARN, "%s: update not allowed", args[i] ); |
||||||
|
SPI_finish(); |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
SPI_finish(); |
||||||
|
return (tuple); |
||||||
|
} |
@ -0,0 +1,9 @@ |
|||||||
|
CREATE TABLE TEST ( COL1 INT, COL2 INT, COL3 INT ); |
||||||
|
|
||||||
|
CREATE TRIGGER BT BEFORE UPDATE ON TEST FOR EACH ROW |
||||||
|
EXECUTE PROCEDURE |
||||||
|
noup ('COL1'); |
||||||
|
|
||||||
|
-- Now Try |
||||||
|
INSERT INTO TEST VALUES (10,20,30); |
||||||
|
UPDATE TEST SET COL1 = 5; |
@ -0,0 +1,7 @@ |
|||||||
|
DROP FUNCTION noup (); |
||||||
|
|
||||||
|
CREATE FUNCTION noup () |
||||||
|
RETURNS opaque |
||||||
|
AS '_OBJWD_/noup_DLSUFFIX_' |
||||||
|
LANGUAGE 'c' |
||||||
|
; |
Loading…
Reference in new issue