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.
261 lines
9.3 KiB
PL/PgSQL
261 lines
9.3 KiB
PL/PgSQL
-- This file has functions that implement changes to hypertable columns as
|
|
-- change to the underlying chunk tables.
|
|
|
|
-- TODO(mat) - Doc this? Not sure I can do it justice.
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.create_partition_constraint_for_column(
|
|
hypertable_id INTEGER,
|
|
column_name NAME
|
|
)
|
|
RETURNS VOID LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
BEGIN
|
|
PERFORM _iobeamdb_internal.add_partition_constraint(pr.schema_name, pr.table_name, p.keyspace_start, p.keyspace_end,
|
|
p.epoch_id)
|
|
FROM _iobeamdb_catalog.partition_epoch AS pe
|
|
INNER JOIN _iobeamdb_catalog.partition AS p ON (p.epoch_id = pe.id)
|
|
INNER JOIN _iobeamdb_catalog.partition_replica AS pr ON (pr.partition_id = p.id)
|
|
WHERE pe.hypertable_id = create_partition_constraint_for_column.hypertable_id
|
|
AND pe.partitioning_column = create_partition_constraint_for_column.column_name;
|
|
END
|
|
$BODY$;
|
|
|
|
-- Adds a column to a table (e.g. main table or root table)
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.create_column_on_table(
|
|
schema_name NAME,
|
|
table_name NAME,
|
|
column_name NAME,
|
|
attnum int2,
|
|
data_type_oid REGTYPE,
|
|
default_value TEXT,
|
|
not_null BOOLEAN
|
|
)
|
|
RETURNS VOID LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
DECLARE
|
|
null_constraint TEXT := 'NOT NULL';
|
|
default_constraint TEXT := '';
|
|
created_columns_att_num INT2;
|
|
BEGIN
|
|
IF NOT not_null THEN
|
|
null_constraint = 'NULL';
|
|
END IF;
|
|
|
|
default_constraint = 'DEFAULT '|| default_value;
|
|
|
|
EXECUTE format(
|
|
$$
|
|
ALTER TABLE %1$I.%2$I ADD COLUMN %3$I %4$s %5$s %6$s
|
|
$$,
|
|
schema_name, table_name, column_name, data_type_oid, default_constraint, null_constraint);
|
|
|
|
SELECT att.attnum INTO STRICT created_columns_att_num
|
|
FROM pg_attribute att
|
|
WHERE att.attrelid = format('%I.%I', schema_name, table_name)::regclass AND att.attname = column_name
|
|
AND NOT attisdropped;
|
|
|
|
IF created_columns_att_num IS DISTINCT FROM attnum THEN
|
|
RAISE EXCEPTION 'Inconsistent state: the attnum of newly created colum does not match (% vs %)', attnum, created_columns_att_num
|
|
USING ERRCODE = 'IO501';
|
|
END IF;
|
|
END
|
|
$BODY$;
|
|
|
|
|
|
-- Removes a column from a table (e.g. main table or root table)
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.drop_column_on_table(
|
|
schema_name NAME,
|
|
table_name NAME,
|
|
column_name NAME
|
|
) RETURNS VOID LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
BEGIN
|
|
EXECUTE format(
|
|
$$
|
|
ALTER TABLE IF EXISTS %1$I.%2$I DROP COLUMN %3$I
|
|
$$, schema_name, table_name, column_name);
|
|
END
|
|
$BODY$;
|
|
|
|
-- Changes the default of a column in a table (e.g. main table or root table)
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.exec_alter_column_set_default(
|
|
schema_name NAME,
|
|
table_name NAME,
|
|
column_name NAME,
|
|
new_default_value TEXT
|
|
) RETURNS VOID LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
BEGIN
|
|
EXECUTE format(
|
|
$$
|
|
ALTER TABLE %1$I.%2$I ALTER COLUMN %3$I SET DEFAULT %4$L
|
|
$$, schema_name, table_name, column_name, new_default_value);
|
|
END
|
|
$BODY$;
|
|
|
|
-- Renames a column of a table (e.g. main table or root table)
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.exec_alter_table_rename_column(
|
|
schema_name NAME,
|
|
table_name NAME,
|
|
old_column NAME,
|
|
new_column NAME
|
|
) RETURNS VOID LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
BEGIN
|
|
EXECUTE format(
|
|
$$
|
|
ALTER TABLE %1$I.%2$I RENAME COLUMN %3$I TO %4$I
|
|
$$, schema_name, table_name, old_column, new_column);
|
|
END
|
|
$BODY$;
|
|
|
|
-- Sets a column of a table (e.g. main table or root table) to NOT NULL
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.exec_alter_column_set_not_null(
|
|
schema_name NAME,
|
|
table_name NAME,
|
|
column_name NAME,
|
|
new_not_null BOOLEAN
|
|
) RETURNS VOID LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
BEGIN
|
|
IF new_not_null THEN
|
|
EXECUTE format(
|
|
$$
|
|
ALTER TABLE %1$I.%2$I ALTER COLUMN %3$I SET NOT NULL
|
|
$$, schema_name, table_name, column_name);
|
|
ELSE
|
|
EXECUTE format(
|
|
$$
|
|
ALTER TABLE %1$I.%2$I ALTER COLUMN %3$I DROP NOT NULL
|
|
$$, schema_name, table_name, column_name);
|
|
END IF;
|
|
END
|
|
$BODY$;
|
|
|
|
-- Trigger to modify a column from a hypertable.
|
|
-- Called when the user alters the main table by adding a column or changing
|
|
-- the properties of a column.
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.on_change_column()
|
|
RETURNS TRIGGER LANGUAGE PLPGSQL AS
|
|
$BODY$
|
|
DECLARE
|
|
hypertable_row _iobeamdb_catalog.hypertable;
|
|
update_found BOOLEAN = false;
|
|
BEGIN
|
|
|
|
IF TG_OP = 'INSERT' THEN
|
|
SELECT *
|
|
INTO STRICT hypertable_row
|
|
FROM _iobeamdb_catalog.hypertable AS h
|
|
WHERE h.id = NEW.hypertable_id;
|
|
|
|
-- update root table
|
|
PERFORM _iobeamdb_internal.create_column_on_table(
|
|
hypertable_row.root_schema_name, hypertable_row.root_table_name,
|
|
NEW.name, NEW.attnum, NEW.data_type, NEW.default_value, NEW.not_null);
|
|
IF new.created_on <> current_database() THEN
|
|
PERFORM set_config('io.ignore_ddl_in_trigger', 'true', true);
|
|
-- update main table on others
|
|
PERFORM _iobeamdb_internal.create_column_on_table(
|
|
hypertable_row.schema_name, hypertable_row.table_name,
|
|
NEW.name, NEW.attnum, NEW.data_type, NEW.default_value, NEW.not_null);
|
|
END IF;
|
|
|
|
PERFORM _iobeamdb_internal.create_partition_constraint_for_column(NEW.hypertable_id, NEW.name);
|
|
RETURN NEW;
|
|
ELSIF TG_OP = 'UPDATE' THEN
|
|
SELECT *
|
|
INTO STRICT hypertable_row
|
|
FROM _iobeamdb_catalog.hypertable AS h
|
|
WHERE h.id = NEW.hypertable_id;
|
|
|
|
IF NEW.default_value IS DISTINCT FROM OLD.default_value THEN
|
|
update_found = TRUE;
|
|
-- update root table
|
|
PERFORM _iobeamdb_internal.exec_alter_column_set_default(
|
|
hypertable_row.root_schema_name, hypertable_row.root_table_name,
|
|
NEW.name, NEW.default_value);
|
|
|
|
IF NEW.modified_on <> current_database() THEN
|
|
PERFORM set_config('io.ignore_ddl_in_trigger', 'true', true);
|
|
-- update main table on others
|
|
PERFORM _iobeamdb_internal.exec_alter_column_set_default(
|
|
hypertable_row.schema_name, hypertable_row.table_name,
|
|
NEW.name, NEW.default_value);
|
|
END IF;
|
|
END IF;
|
|
IF NEW.not_null IS DISTINCT FROM OLD.not_null THEN
|
|
update_found = TRUE;
|
|
-- update root table
|
|
PERFORM _iobeamdb_internal.exec_alter_column_set_not_null(
|
|
hypertable_row.root_schema_name, hypertable_row.root_table_name,
|
|
NEW.name, NEW.not_null);
|
|
IF NEW.modified_on <> current_database() THEN
|
|
PERFORM set_config('io.ignore_ddl_in_trigger', 'true', true);
|
|
-- update main table on others
|
|
PERFORM _iobeamdb_internal.exec_alter_column_set_not_null(
|
|
hypertable_row.schema_name, hypertable_row.table_name,
|
|
NEW.name, NEW.not_null);
|
|
END IF;
|
|
END IF;
|
|
IF NEW.name IS DISTINCT FROM OLD.name THEN
|
|
update_found = TRUE;
|
|
-- update root table
|
|
PERFORM _iobeamdb_internal.exec_alter_table_rename_column(
|
|
hypertable_row.root_schema_name, hypertable_row.root_table_name,
|
|
OLD.name, NEW.name);
|
|
IF NEW.modified_on <> current_database() THEN
|
|
PERFORM set_config('io.ignore_ddl_in_trigger', 'true', true);
|
|
-- update main table on others
|
|
PERFORM _iobeamdb_internal.exec_alter_table_rename_column(
|
|
hypertable_row.schema_name, hypertable_row.table_name,
|
|
OLD.name, NEW.name);
|
|
END IF;
|
|
END IF;
|
|
|
|
IF NOT update_found THEN
|
|
RAISE EXCEPTION 'Invalid update type on %', TG_TABLE_NAME
|
|
USING ERRCODE = 'IO101';
|
|
END IF;
|
|
RETURN NEW;
|
|
END IF;
|
|
PERFORM _iobeamdb_internal.on_trigger_error(TG_OP, TG_TABLE_SCHEMA, TG_TABLE_NAME);
|
|
END
|
|
$BODY$;
|
|
|
|
-- Trigger to remove a column from a hypertable.
|
|
-- Called when the user alters the main table by deleting a column.
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_internal.on_change_deleted_column()
|
|
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;
|
|
|
|
SELECT *
|
|
INTO hypertable_row
|
|
FROM _iobeamdb_catalog.hypertable AS h
|
|
WHERE h.id = NEW.hypertable_id;
|
|
|
|
IF hypertable_row IS NULL THEN
|
|
--presumably hypertable has been dropped and this is part of cascade
|
|
RETURN NEW;
|
|
END IF;
|
|
|
|
-- update root table
|
|
PERFORM _iobeamdb_internal.drop_column_on_table(
|
|
hypertable_row.root_schema_name, hypertable_row.root_table_name, NEW.name);
|
|
IF NEW.deleted_on <> current_database() THEN
|
|
PERFORM set_config('io.ignore_ddl_in_trigger', 'true', true);
|
|
-- update main table on others
|
|
PERFORM _iobeamdb_internal.drop_column_on_table(
|
|
hypertable_row.schema_name, hypertable_row.table_name, NEW.name);
|
|
END IF;
|
|
|
|
RETURN NEW;
|
|
END
|
|
$BODY$
|
|
;
|