diff --git a/contrib/Makefile b/contrib/Makefile index 25263c0..4bd456f 100644 --- a/contrib/Makefile +++ b/contrib/Makefile @@ -40,6 +40,7 @@ SUBDIRS = \ pgstattuple \ pg_visibility \ postgres_fdw \ + require_where \ seg \ spi \ tablefunc \ diff --git a/contrib/require_where/Makefile b/contrib/require_where/Makefile new file mode 100644 index 0000000..9c41691 --- /dev/null +++ b/contrib/require_where/Makefile @@ -0,0 +1,19 @@ +# contrib/require_where/Makefile + +MODULE_big = require_where +OBJS = require_where.o + +PGFILEDESC = 'require_where - require DELETE and/or UPDATE to have a WHERE clause' + +REGRESS = require_where + +ifdef USE_PGXS +PG_CONFIG = pg_config +PGXS = $(shell $(PG_CONFIG) --pgxs) +include $(PGXS) +else +subdir = contrib/require_where +top_builddir = ../.. +include $(top_builddir)/src/Makefile.global +include $(top_builddir)/contrib/contrib-global.mk +endif diff --git a/contrib/require_where/data/test_require_where.data b/contrib/require_where/data/test_require_where.data new file mode 100644 index 0000000..d4a29d8 --- /dev/null +++ b/contrib/require_where/data/test_require_where.data @@ -0,0 +1,16 @@ +Four +score +and +seven +years +ago +our +fathers +brought +forth +on +this +continent +a +new +nation diff --git a/contrib/require_where/expected/require_where.out b/contrib/require_where/expected/require_where.out new file mode 100644 index 0000000..adfd358 --- /dev/null +++ b/contrib/require_where/expected/require_where.out @@ -0,0 +1,22 @@ +-- +-- Test require_where +-- +\set echo all +LOAD 'require_where'; +CREATE TABLE test_require_where(t TEXT); +\copy test_require_where from 'data/test_require_where.data' +UPDATE test_require_where SET t=t; -- succeeds +SET require_where.update TO ON; +UPDATE test_require_where SET t=t; -- fails +ERROR: UPDATE requires a WHERE clause when require_where.delete is set to on +HINT: To update all rows, use "WHERE true" or similar. +UPDATE test_require_where SET t=t WHERE true; -- succeeds +SET require_where.update TO OFF; +UPDATE test_require_where SET t=t; -- succeeds +SET require_where.delete TO ON; +DELETE FROM test_require_where; -- fails +ERROR: DELETE requires a WHERE clause when require_where.delete is set to on +HINT: To delete all rows, use "WHERE true" or similar. +DELETE FROM test_require_where WHERE true; -- succeeds +SET require_where.delete TO OFF; +DELETE FROM test_require_where; -- succeeds diff --git a/contrib/require_where/require_where.c b/contrib/require_where/require_where.c new file mode 100644 index 0000000..27cbc25 --- /dev/null +++ b/contrib/require_where/require_where.c @@ -0,0 +1,92 @@ +/* + * -------------------------------------------------------------------------- + * + * require_where.c + * + * Copyright (C) 2016, PostgreSQL Global Development Group + * + * IDENTIFICATION + * contrib/require_where/require_where.c + * + * -------------------------------------------------------------------------- + */ +#include "postgres.h" + +#include "fmgr.h" + +#include "parser/analyze.h" + +#include "utils/elog.h" +#include "utils/guc.h" + +PG_MODULE_MAGIC; + +void _PG_init(void); +void _PG_fini(void); + +static post_parse_analyze_hook_type original_post_parse_analyze_hook = NULL; +static bool require_where_delete = false; +static bool require_where_update = false; + +static void +require_where_check(ParseState *pstate, Query *query) +{ + + if (require_where_delete && query->commandType == CMD_DELETE) + { + Assert(query->jointree != NULL); + if (query->jointree->quals == NULL) + ereport(ERROR, + (errcode(ERRCODE_SYNTAX_ERROR), + errmsg("DELETE requires a WHERE clause when require_where.delete is set to on"), + errhint("To delete all rows, use \"WHERE true\" or similar."))); + } + + if (require_where_update && query->commandType == CMD_UPDATE) + { + Assert(query->jointree != NULL); + if (query->jointree->quals == NULL) + ereport(ERROR, + (errcode(ERRCODE_SYNTAX_ERROR), + errmsg("UPDATE requires a WHERE clause when require_where.delete is set to on"), + errhint("To update all rows, use \"WHERE true\" or similar."))); + } + + if (original_post_parse_analyze_hook != NULL) + (*original_post_parse_analyze_hook) (pstate, query); +} + +void +_PG_init(void) +{ + DefineCustomBoolVariable("require_where.delete", + "Require every DELETE statement to have a WHERE clause.", + NULL, + &require_where_delete, + false, + PGC_USERSET, + 0, + NULL, + NULL, + NULL); + + DefineCustomBoolVariable("require_where.update", + "Require every UPDATE statement to have a WHERE clause.", + NULL, + &require_where_update, + false, + PGC_USERSET, + 0, + NULL, + NULL, + NULL); + + original_post_parse_analyze_hook = post_parse_analyze_hook; + post_parse_analyze_hook = require_where_check; +} + +void +_PG_fini(void) +{ + post_parse_analyze_hook = original_post_parse_analyze_hook; +} diff --git a/contrib/require_where/sql/require_where.sql b/contrib/require_where/sql/require_where.sql new file mode 100644 index 0000000..4d20b17 --- /dev/null +++ b/contrib/require_where/sql/require_where.sql @@ -0,0 +1,33 @@ +-- +-- Test require_where +-- + +\set echo all + +LOAD 'require_where'; + +CREATE TABLE test_require_where(t TEXT); + +\copy test_require_where from 'data/test_require_where.data' + +UPDATE test_require_where SET t=t; -- succeeds + +SET require_where.update TO ON; + +UPDATE test_require_where SET t=t; -- fails + +UPDATE test_require_where SET t=t WHERE true; -- succeeds + +SET require_where.update TO OFF; + +UPDATE test_require_where SET t=t; -- succeeds + +SET require_where.delete TO ON; + +DELETE FROM test_require_where; -- fails + +DELETE FROM test_require_where WHERE true; -- succeeds + +SET require_where.delete TO OFF; + +DELETE FROM test_require_where; -- succeeds diff --git a/doc/src/sgml/contrib.sgml b/doc/src/sgml/contrib.sgml index c8708ec..48ca717 100644 --- a/doc/src/sgml/contrib.sgml +++ b/doc/src/sgml/contrib.sgml @@ -135,6 +135,7 @@ CREATE EXTENSION module_name FROM unpackaged; &pgtrgm; &pgvisibility; &postgres-fdw; + &require-where; &seg; &sepgsql; &contrib-spi; diff --git a/doc/src/sgml/filelist.sgml b/doc/src/sgml/filelist.sgml index 69649a7..4552273 100644 --- a/doc/src/sgml/filelist.sgml +++ b/doc/src/sgml/filelist.sgml @@ -141,6 +141,7 @@ + diff --git a/doc/src/sgml/require_where.sgml b/doc/src/sgml/require_where.sgml new file mode 100644 index 0000000..d9f59c6 --- /dev/null +++ b/doc/src/sgml/require_where.sgml @@ -0,0 +1,98 @@ + + + + require_where + + + require_where + + + + This module allows a user to require that WHERE clauses on either DELETE or + UPDATE not be empty using a custom GUC to control each. + + + + To use this module, you need to include require_where in + the either the parameter, or + LOAD the library directly. + + + + Here is an example showing how to set up a database cluster with + require_where. + +$ psql -U postgres +# SHOW shared_preload_libraries; /* Make sure not to clobber something by accident */ + +If you found something, +# ALTER SYSTEM SET shared_preload_libraries='the,stuff,you,found,require_where'; + +Otherwise, +# ALTER SYSTEM SET shared_preload_libraries='require_where'; + +Then restart PostgreSQL + + + + + Here is an example using LOAD: + +$ psql -U postgres +# LOAD '$libdir/require_where'; + + + + + GUC Parameters + + + + + require_where.delete (boolean) + + require_where.delete configuration parameter + + + + + This parameter defaults to false. When set + to true, every DELETE requires + some kind of WHERE clause in order to + execute. When DELETE is meant to span the + entire relation, one can issue a WHERE true + or equivalent. + + + + + + require_where.update (boolean) + + require_where.update configuration parameter + + + + + This parameter defaults to false. When set + to true, every UPDATE requires + some kind of WHERE clause in order to + execute. When UPDATE is meant to span the + entire relation, one can issue a WHERE true + or equivalent. + + + + + + + + Authors + + + David Fetter david@fetter.org, + Robert Haas robertmhaas@gmail.com and + Andrew Gierth andrew@tao11.riddles.org.uk. + + +