mirror of
https://github.com/timescale/timescaledb.git
synced 2025-05-15 18:13:18 +08:00
7b94c573ba
- Directory structure now matches common practices - Regression tests now run with pg_regress via the PGXS infrastructure. - Unit tests do not integrate well with pg_regress and have to be run separately. - Docker functionality is separate from main Makefile. Run with `make -f docker.mk` to build and `make -f docker.mk run` to run the database in a container.
108 lines
3.7 KiB
PL/PgSQL
108 lines
3.7 KiB
PL/PgSQL
-- This file contains triggers that act on the main 'hypertable' table as
|
|
-- well as triggers for newly created hypertables.
|
|
|
|
-- Trigger to prevent unsupported operations on the main table.
|
|
-- Our C code should rewrite a DELETE or UPDATE to apply to the replica and not main table.
|
|
-- The only exception is if we use ONLY, which should be an error.
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.on_unsupported_main_table()
|
|
RETURNS TRIGGER LANGUAGE PLPGSQL AS
|
|
$BODY$
|
|
BEGIN
|
|
IF TG_OP = 'UPDATE' THEN
|
|
RAISE EXCEPTION 'UPDATE ONLY not supported on hypertables'
|
|
USING ERRCODE = 'IO101';
|
|
ELSIF TG_OP = 'DELETE' AND current_setting('io.ignore_delete_in_trigger', true) <> 'true' THEN
|
|
RAISE EXCEPTION 'DELETE ONLY not currently supported on hypertables'
|
|
USING ERRCODE = 'IO101';
|
|
END IF;
|
|
RETURN NEW;
|
|
END
|
|
$BODY$;
|
|
|
|
-- Trigger function to move INSERT'd data on the main table to child tables.
|
|
-- After data is inserted on the main table, it is placed in the correct
|
|
-- partition tables based on its partition key and time.
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.on_modify_main_table()
|
|
RETURNS TRIGGER LANGUAGE PLPGSQL AS
|
|
$BODY$
|
|
BEGIN
|
|
EXECUTE format(
|
|
$$
|
|
SELECT _iobeamdb_internal.insert_data(
|
|
(SELECT id FROM _iobeamdb_catalog.hypertable h
|
|
WHERE h.schema_name = %1$L AND h.table_name = %2$L)
|
|
, %3$L)
|
|
$$, TG_TABLE_SCHEMA, TG_TABLE_NAME, TG_RELID);
|
|
RETURN NEW;
|
|
END
|
|
$BODY$;
|
|
|
|
-- Trigger for handling the addition of new hypertables.
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.on_change_hypertable()
|
|
RETURNS TRIGGER LANGUAGE PLPGSQL AS
|
|
$BODY$
|
|
DECLARE
|
|
remote_node _iobeamdb_catalog.node;
|
|
BEGIN
|
|
|
|
IF TG_OP = 'INSERT' THEN
|
|
|
|
EXECUTE format(
|
|
$$
|
|
CREATE SCHEMA IF NOT EXISTS %I
|
|
$$, NEW.associated_schema_name);
|
|
|
|
PERFORM _iobeamdb_internal.create_table(NEW.root_schema_name, NEW.root_table_name);
|
|
|
|
IF NEW.created_on <> current_database() THEN
|
|
PERFORM _iobeamdb_internal.create_schema(NEW.schema_name);
|
|
PERFORM _iobeamdb_internal.create_table(NEW.schema_name, NEW.table_name);
|
|
END IF;
|
|
|
|
-- UPDATE not supported, so do them before action
|
|
EXECUTE format(
|
|
$$
|
|
CREATE TRIGGER modify_trigger BEFORE UPDATE OR DELETE ON %I.%I
|
|
FOR EACH STATEMENT EXECUTE PROCEDURE _iobeamdb_internal.on_unsupported_main_table();
|
|
$$, NEW.schema_name, NEW.table_name);
|
|
EXECUTE format(
|
|
$$
|
|
CREATE TRIGGER insert_trigger AFTER INSERT ON %I.%I
|
|
FOR EACH STATEMENT EXECUTE PROCEDURE _iobeamdb_internal.on_modify_main_table();
|
|
$$, NEW.schema_name, NEW.table_name);
|
|
|
|
RETURN NEW;
|
|
END IF;
|
|
|
|
PERFORM _iobeamdb_internal.on_trigger_error(TG_OP, TG_TABLE_SCHEMA, TG_TABLE_NAME);
|
|
END
|
|
$BODY$
|
|
SET client_min_messages = WARNING; --suppress NOTICE on IF EXISTS schema
|
|
|
|
|
|
/*
|
|
Drops public hypertable when hypertable rows deleted (row created in deleted_hypertable table).
|
|
*/
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.on_change_deleted_hypertable()
|
|
RETURNS TRIGGER LANGUAGE PLPGSQL AS
|
|
$BODY$
|
|
DECLARE
|
|
hypertable_row _iobeamdb_catalog.hypertable;
|
|
BEGIN
|
|
IF TG_OP <> 'INSERT' THEN
|
|
PERFORM _iobeamdb_internal.on_trigger_error(TG_OP, TG_TABLE_SCHEMA, TG_TABLE_NAME);
|
|
END IF;
|
|
--drop index on all chunks
|
|
|
|
PERFORM _iobeamdb_internal.drop_root_table(NEW.root_schema_name, NEW.root_table_name);
|
|
|
|
IF new.deleted_on <> current_database() THEN
|
|
PERFORM set_config('io.ignore_ddl_in_trigger', 'true', true);
|
|
EXECUTE format('DROP TABLE %I.%I', NEW.schema_name, NEW.table_name);
|
|
END IF;
|
|
|
|
RETURN NEW;
|
|
END
|
|
$BODY$
|
|
;
|