You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1356 lines
42 KiB
1356 lines
42 KiB
-- Depends on:
|
|
-- * CDB_ExtensionUtils.sql
|
|
-- * CDB_TransformToWebmercator.sql
|
|
-- * CDB_TableMetadata.sql
|
|
-- * CDB_Quota.sql
|
|
-- * _CDB_UserQuotaInBytes() function, installed by rails
|
|
-- (user.rebuild_quota_trigger, called by rake task cartodb:db:update_test_quota_trigger)
|
|
|
|
-- 1) Required checks before running cartodbfication
|
|
-- Either will pass silenty or raise an exception
|
|
CREATE OR REPLACE FUNCTION _CDB_check_prerequisites(schema_name TEXT, reloid REGCLASS)
|
|
RETURNS void
|
|
AS $$
|
|
DECLARE
|
|
sql TEXT;
|
|
BEGIN
|
|
IF cartodb.schema_exists(schema_name) = false THEN
|
|
RAISE EXCEPTION 'Invalid schema name "%"', schema_name;
|
|
END IF;
|
|
|
|
-- TODO: Check that user quota is set ?
|
|
BEGIN
|
|
EXECUTE FORMAT('SELECT %I._CDB_UserQuotaInBytes();', schema_name::text) INTO sql;
|
|
EXCEPTION WHEN undefined_function THEN
|
|
RAISE EXCEPTION 'Please set user quota before cartodbfying tables.';
|
|
END;
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
|
|
-- 2) Drop cartodb triggers (might prevent changing columns)
|
|
CREATE OR REPLACE FUNCTION _CDB_drop_triggers(reloid REGCLASS)
|
|
RETURNS void
|
|
AS $$
|
|
DECLARE
|
|
sql TEXT;
|
|
BEGIN
|
|
-- "track_updates"
|
|
sql := 'DROP TRIGGER IF EXISTS track_updates ON ' || reloid::text;
|
|
EXECUTE sql;
|
|
|
|
-- "update_the_geom_webmercator"
|
|
sql := 'DROP TRIGGER IF EXISTS update_the_geom_webmercator_trigger ON ' || reloid::text;
|
|
EXECUTE sql;
|
|
|
|
-- "test_quota" and "test_quota_per_row"
|
|
sql := 'DROP TRIGGER IF EXISTS test_quota ON ' || reloid::text;
|
|
EXECUTE sql;
|
|
sql := 'DROP TRIGGER IF EXISTS test_quota_per_row ON ' || reloid::text;
|
|
EXECUTE sql;
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
|
|
-- 3) Cartodb_id creation & validation or renaming if invalid
|
|
CREATE OR REPLACE FUNCTION _CDB_create_cartodb_id_column(reloid REGCLASS)
|
|
RETURNS void
|
|
AS $$
|
|
DECLARE
|
|
sql TEXT;
|
|
rec RECORD;
|
|
rec2 RECORD;
|
|
had_column BOOLEAN;
|
|
i INTEGER;
|
|
new_name TEXT;
|
|
cartodb_id_name TEXT;
|
|
BEGIN
|
|
<< cartodb_id_setup >>
|
|
LOOP --{
|
|
had_column := FALSE;
|
|
BEGIN
|
|
sql := 'ALTER TABLE ' || reloid::text || ' ADD cartodb_id SERIAL NOT NULL UNIQUE';
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
cartodb_id_name := 'cartodb_id';
|
|
EXIT cartodb_id_setup;
|
|
EXCEPTION
|
|
WHEN duplicate_column THEN
|
|
RAISE NOTICE 'Column cartodb_id already exists';
|
|
had_column := TRUE;
|
|
WHEN others THEN
|
|
RAISE EXCEPTION 'Cartodbfying % (cartodb_id): % (%)', reloid, SQLERRM, SQLSTATE;
|
|
END;
|
|
|
|
IF had_column THEN
|
|
SELECT pg_catalog.pg_get_serial_sequence(reloid::text, 'cartodb_id')
|
|
AS seq INTO rec2;
|
|
|
|
-- Check data type is an integer
|
|
SELECT
|
|
pg_catalog.pg_get_serial_sequence(reloid::text, 'cartodb_id') as seq,
|
|
t.typname, t.oid, a.attnotnull FROM pg_type t, pg_attribute a
|
|
WHERE a.atttypid = t.oid AND a.attrelid = reloid AND NOT a.attisdropped AND a.attname = 'cartodb_id'
|
|
INTO STRICT rec;
|
|
|
|
-- 20=int2, 21=int4, 23=int8
|
|
IF rec.oid NOT IN (20,21,23) THEN -- {
|
|
RAISE NOTICE 'Existing cartodb_id field is of invalid type % (need int2, int4 or int8), renaming', rec.typname;
|
|
ELSIF rec.seq IS NULL THEN -- }{
|
|
RAISE NOTICE 'Existing cartodb_id field does not have an associated sequence, renaming';
|
|
ELSE -- }{
|
|
sql := 'ALTER TABLE ' || reloid::text || ' ALTER COLUMN cartodb_id SET NOT NULL';
|
|
IF NOT EXISTS ( SELECT c.conname FROM pg_constraint c, pg_attribute a
|
|
WHERE c.conkey = ARRAY[a.attnum] AND c.conrelid = reloid
|
|
AND a.attrelid = reloid
|
|
AND NOT a.attisdropped
|
|
AND a.attname = 'cartodb_id'
|
|
AND c.contype IN ( 'u', 'p' ) ) -- unique or pkey
|
|
THEN
|
|
sql := sql || ', ADD unique(cartodb_id)';
|
|
END IF;
|
|
BEGIN
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
cartodb_id_name := 'cartodb_id';
|
|
EXIT cartodb_id_setup;
|
|
EXCEPTION
|
|
WHEN unique_violation OR not_null_violation THEN
|
|
RAISE NOTICE '%, renaming', SQLERRM;
|
|
WHEN others THEN
|
|
RAISE EXCEPTION 'Cartodbfying % (cartodb_id): % (%)', reloid, SQLERRM, SQLSTATE;
|
|
END;
|
|
END IF; -- }
|
|
|
|
-- invalid column, need rename and re-create it
|
|
i := 0;
|
|
<< rename_column >>
|
|
LOOP --{
|
|
new_name := '_cartodb_id' || i;
|
|
BEGIN
|
|
sql := 'ALTER TABLE ' || reloid::text || ' RENAME COLUMN cartodb_id TO ' || new_name;
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
EXCEPTION
|
|
WHEN duplicate_column THEN
|
|
i := i+1;
|
|
CONTINUE rename_column;
|
|
WHEN others THEN
|
|
RAISE EXCEPTION 'Cartodbfying % (renaming cartodb_id): % (%)', reloid, SQLERRM, SQLSTATE;
|
|
END;
|
|
cartodb_id_name := new_name;
|
|
EXIT rename_column;
|
|
END LOOP; --}
|
|
CONTINUE cartodb_id_setup;
|
|
END IF;
|
|
END LOOP; -- }
|
|
|
|
-- Try to copy data from new name if possible
|
|
IF new_name IS NOT NULL THEN
|
|
RAISE NOTICE 'Trying to recover data from % column', new_name;
|
|
BEGIN
|
|
-- Copy existing values to new field
|
|
-- NOTE: using ALTER is a workaround to a PostgreSQL bug and is also known to be faster for tables with many rows
|
|
-- See http://www.postgresql.org/message-id/20140530143150.GA11051@localhost
|
|
sql := 'ALTER TABLE ' || reloid::text
|
|
|| ' ALTER cartodb_id TYPE int USING '
|
|
|| new_name || '::int4';
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
|
|
-- Find max value
|
|
sql := 'SELECT max(cartodb_id) FROM ' || reloid::text;
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql INTO rec;
|
|
|
|
-- Find sequence name
|
|
SELECT pg_catalog.pg_get_serial_sequence(reloid::text, 'cartodb_id')
|
|
AS seq INTO rec2;
|
|
|
|
-- Reset sequence name
|
|
sql := 'ALTER SEQUENCE ' || rec2.seq::text
|
|
|| ' RESTART WITH ' || rec.max + 1;
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
|
|
-- Drop old column (all went fine if we got here)
|
|
sql := 'ALTER TABLE ' || reloid::text || ' DROP ' || new_name;
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
|
|
EXCEPTION
|
|
WHEN others THEN
|
|
RAISE NOTICE 'Could not initialize cartodb_id with existing values: % (%)',
|
|
SQLERRM, SQLSTATE;
|
|
END;
|
|
END IF;
|
|
|
|
-- Set primary key of the table if not already present (e.g. tables created from SQL API)
|
|
IF cartodb_id_name IS NULL THEN
|
|
RAISE EXCEPTION 'Cartodbfying % (Didnt get cartodb_id field name)', reloid;
|
|
END IF;
|
|
BEGIN
|
|
sql := 'ALTER TABLE ' || reloid::text || ' ADD PRIMARY KEY (cartodb_id)';
|
|
EXECUTE sql;
|
|
EXCEPTION
|
|
WHEN others THEN
|
|
RAISE DEBUG 'Table % Already had PRIMARY KEY', reloid;
|
|
END;
|
|
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
|
|
|
|
-- 5) the_geom and the_geom_webmercator creation & validation or renaming if invalid
|
|
CREATE OR REPLACE FUNCTION _CDB_create_the_geom_columns(reloid REGCLASS)
|
|
RETURNS BOOLEAN[]
|
|
AS $$
|
|
DECLARE
|
|
sql TEXT;
|
|
rec RECORD;
|
|
rec2 RECORD;
|
|
had_column BOOLEAN;
|
|
i INTEGER;
|
|
new_name TEXT;
|
|
exists_geom_cols BOOLEAN[];
|
|
BEGIN
|
|
-- We need the_geom and the_geom_webmercator
|
|
FOR rec IN SELECT * FROM ( VALUES ('the_geom',4326), ('the_geom_webmercator',3857) ) t(cname,csrid) LOOP --{
|
|
<< column_setup >> LOOP --{
|
|
BEGIN
|
|
sql := 'ALTER TABLE ' || reloid::text || ' ADD ' || rec.cname
|
|
|| ' GEOMETRY(geometry,' || rec.csrid || ')';
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
sql := 'CREATE INDEX ON ' || reloid::text || ' USING GIST ( ' || rec.cname || ')';
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
exists_geom_cols := array_append(exists_geom_cols, false);
|
|
EXIT column_setup;
|
|
EXCEPTION
|
|
WHEN duplicate_column THEN
|
|
exists_geom_cols := array_append(exists_geom_cols, true);
|
|
RAISE NOTICE 'Column % already exists', rec.cname;
|
|
WHEN others THEN
|
|
RAISE EXCEPTION 'Cartodbfying % (%): % (%)', reloid, rec.cname, SQLERRM, SQLSTATE;
|
|
END;
|
|
|
|
<< column_fixup >>
|
|
LOOP --{
|
|
-- Check data type is a GEOMETRY
|
|
SELECT t.typname, t.oid, a.attnotnull,
|
|
postgis_typmod_srid(a.atttypmod) as srid,
|
|
postgis_typmod_type(a.atttypmod) as gtype
|
|
FROM pg_type t, pg_attribute a
|
|
WHERE a.atttypid = t.oid AND a.attrelid = reloid AND NOT a.attisdropped AND a.attname = rec.cname
|
|
INTO STRICT rec2;
|
|
|
|
IF rec2.typname NOT IN ('geometry') THEN -- {
|
|
RAISE NOTICE 'Existing % field is of invalid type % (need geometry), renaming', rec.cname, rec2.typname;
|
|
EXIT column_fixup; -- cannot fix
|
|
END IF; -- }
|
|
|
|
IF rec2.srid != rec.csrid THEN -- {
|
|
BEGIN
|
|
sql := 'ALTER TABLE ' || reloid::text || ' ALTER ' || rec.cname
|
|
|| ' TYPE geometry(' || rec2.gtype || ',' || rec.csrid || ') USING ST_Transform('
|
|
|| rec.cname || ',' || rec.csrid || ')';
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
EXCEPTION
|
|
WHEN others THEN
|
|
RAISE NOTICE 'Could not enforce SRID % to column %: %, renaming', rec.csrid, rec.cname, SQLERRM;
|
|
EXIT column_fixup; -- cannot fix, will rename
|
|
END;
|
|
END IF; -- }
|
|
|
|
-- add gist indices if not there already
|
|
IF NOT EXISTS ( SELECT ir.relname
|
|
FROM pg_am am, pg_class ir,
|
|
pg_class c, pg_index i,
|
|
pg_attribute a
|
|
WHERE c.oid = reloid AND i.indrelid = c.oid
|
|
AND a.attname = rec.cname
|
|
AND i.indexrelid = ir.oid AND i.indnatts = 1
|
|
AND i.indkey[0] = a.attnum AND a.attrelid = c.oid
|
|
AND NOT a.attisdropped AND am.oid = ir.relam
|
|
AND am.amname = 'gist' )
|
|
THEN -- {
|
|
BEGIN
|
|
sql := 'CREATE INDEX ON ' || reloid::text || ' USING GIST ( ' || rec.cname || ')';
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
EXCEPTION
|
|
WHEN others THEN
|
|
RAISE EXCEPTION 'Cartodbfying % (% index): % (%)', reloid, rec.cname, SQLERRM, SQLSTATE;
|
|
END;
|
|
END IF; -- }
|
|
|
|
-- if we reached this line, all went good
|
|
EXIT column_setup;
|
|
END LOOP; -- } column_fixup
|
|
|
|
-- invalid column, need rename and re-create it
|
|
i := 0;
|
|
<< rename_column >>
|
|
LOOP --{
|
|
new_name := '_' || rec.cname || i;
|
|
BEGIN
|
|
sql := 'ALTER TABLE ' || reloid::text || ' RENAME COLUMN ' || rec.cname || ' TO ' || new_name;
|
|
RAISE DEBUG 'Running %', sql;
|
|
EXECUTE sql;
|
|
EXCEPTION
|
|
WHEN duplicate_column THEN
|
|
i := i+1;
|
|
CONTINUE rename_column;
|
|
WHEN others THEN
|
|
RAISE EXCEPTION 'Cartodbfying % (rename %): % (%)', reloid, rec.cname, SQLERRM, SQLSTATE;
|
|
END;
|
|
EXIT rename_column;
|
|
END LOOP; --}
|
|
CONTINUE column_setup;
|
|
END LOOP; -- } column_setup
|
|
END LOOP; -- } on expected geometry columns
|
|
|
|
RETURN exists_geom_cols;
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
|
|
-- 6) Initialize the_geom with values from the_geom_webmercator
|
|
-- do this only if the_geom_webmercator was found (not created) and the_geom was NOT found.
|
|
CREATE OR REPLACE FUNCTION _CDB_populate_the_geom_from_the_geom_webmercator(reloid REGCLASS, geom_columns_exist BOOLEAN[])
|
|
RETURNS void
|
|
AS $$
|
|
DECLARE
|
|
sql TEXT;
|
|
BEGIN
|
|
IF geom_columns_exist[2] AND NOT geom_columns_exist[1] THEN
|
|
sql := 'UPDATE ' || reloid::text || ' SET the_geom = ST_Transform(the_geom_webmercator, 4326) ';
|
|
EXECUTE sql;
|
|
END IF;
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
|
|
-- 7) Initialize the_geom_webmercator with values from the_geom
|
|
-- do this only if the_geom was found (not created) and the_geom_webmercator was NOT found.
|
|
CREATE OR REPLACE FUNCTION _CDB_populate_the_geom_webmercator_from_the_geom(reloid REGCLASS, geom_columns_exist BOOLEAN[])
|
|
RETURNS void
|
|
AS $$
|
|
DECLARE
|
|
sql TEXT;
|
|
BEGIN
|
|
IF geom_columns_exist[1] AND NOT geom_columns_exist[2] THEN
|
|
sql := 'UPDATE ' || reloid::text || ' SET the_geom_webmercator = public.CDB_TransformToWebmercator(the_geom) ';
|
|
EXECUTE sql;
|
|
END IF;
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
|
|
-- 8.a) Create all triggers
|
|
-- NOTE: drop/create has the side-effect of re-enabling disabled triggers
|
|
CREATE OR REPLACE FUNCTION _CDB_create_triggers(schema_name TEXT, reloid REGCLASS)
|
|
RETURNS void
|
|
AS $$
|
|
DECLARE
|
|
sql TEXT;
|
|
BEGIN
|
|
-- "track_updates"
|
|
sql := 'CREATE trigger track_updates AFTER INSERT OR UPDATE OR DELETE OR TRUNCATE ON '
|
|
|| reloid::text
|
|
|| ' FOR EACH STATEMENT EXECUTE PROCEDURE public.cdb_tablemetadata_trigger()';
|
|
EXECUTE sql;
|
|
|
|
-- "update_the_geom_webmercator"
|
|
-- TODO: why _before_ and not after ?
|
|
sql := 'CREATE trigger update_the_geom_webmercator_trigger BEFORE INSERT OR UPDATE OF the_geom ON '
|
|
|| reloid::text
|
|
|| ' FOR EACH ROW EXECUTE PROCEDURE public._CDB_update_the_geom_webmercator()';
|
|
EXECUTE sql;
|
|
|
|
-- "test_quota" and "test_quota_per_row"
|
|
|
|
sql := 'CREATE TRIGGER test_quota BEFORE UPDATE OR INSERT ON '
|
|
|| reloid::text
|
|
|| ' EXECUTE PROCEDURE public.CDB_CheckQuota(1, ''-1'', '''
|
|
|| schema_name::text
|
|
|| ''')';
|
|
EXECUTE sql;
|
|
|
|
sql := 'CREATE TRIGGER test_quota_per_row BEFORE UPDATE OR INSERT ON '
|
|
|| reloid::text
|
|
|| ' FOR EACH ROW EXECUTE PROCEDURE public.CDB_CheckQuota(0.001, ''-1'', '''
|
|
|| schema_name::text
|
|
|| ''')';
|
|
EXECUTE sql;
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
-- 8.b) Create all raster triggers
|
|
-- NOTE: drop/create has the side-effect of re-enabling disabled triggers
|
|
CREATE OR REPLACE FUNCTION _CDB_create_raster_triggers(schema_name TEXT, reloid REGCLASS)
|
|
RETURNS void
|
|
AS $$
|
|
DECLARE
|
|
sql TEXT;
|
|
BEGIN
|
|
-- "track_updates"
|
|
sql := 'CREATE trigger track_updates AFTER INSERT OR UPDATE OR DELETE OR TRUNCATE ON '
|
|
|| reloid::text
|
|
|| ' FOR EACH STATEMENT EXECUTE PROCEDURE public.cdb_tablemetadata_trigger()';
|
|
EXECUTE sql;
|
|
|
|
-- "test_quota" and "test_quota_per_row"
|
|
|
|
sql := 'CREATE TRIGGER test_quota BEFORE UPDATE OR INSERT ON '
|
|
|| reloid::text
|
|
|| ' EXECUTE PROCEDURE public.CDB_CheckQuota(1, ''-1'', '''
|
|
|| schema_name::text
|
|
|| ''')';
|
|
EXECUTE sql;
|
|
|
|
sql := 'CREATE TRIGGER test_quota_per_row BEFORE UPDATE OR INSERT ON '
|
|
|| reloid::text
|
|
|| ' FOR EACH ROW EXECUTE PROCEDURE public.CDB_CheckQuota(0.001, ''-1'', '''
|
|
|| schema_name::text
|
|
|| ''')';
|
|
EXECUTE sql;
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
|
|
|
|
-- Update the_geom_webmercator
|
|
CREATE OR REPLACE FUNCTION _CDB_update_the_geom_webmercator()
|
|
RETURNS trigger
|
|
AS $$
|
|
BEGIN
|
|
NEW.the_geom_webmercator := public.CDB_TransformToWebmercator(NEW.the_geom);
|
|
RETURN NEW;
|
|
END;
|
|
$$ LANGUAGE plpgsql VOLATILE;
|
|
|
|
|
|
-- Auxiliary function
|
|
CREATE OR REPLACE FUNCTION cartodb._CDB_is_raster_table(schema_name TEXT, reloid REGCLASS)
|
|
RETURNS BOOLEAN
|
|
AS $$
|
|
DECLARE
|
|
sql TEXT;
|
|
is_raster BOOLEAN;
|
|
rel_name TEXT;
|
|
BEGIN
|
|
IF cartodb.schema_exists(schema_name) = FALSE THEN
|
|
RAISE EXCEPTION 'Invalid schema name "%"', schema_name;
|
|
END IF;
|
|
|
|
SELECT relname FROM pg_class WHERE oid=reloid INTO rel_name;
|
|
|
|
BEGIN
|
|
sql := 'SELECT the_raster_webmercator FROM '
|
|
|| quote_ident(schema_name::TEXT)
|
|
|| '.'
|
|
|| quote_ident(rel_name::TEXT)
|
|
|| ' LIMIT 1';
|
|
is_raster = TRUE;
|
|
EXECUTE sql;
|
|
|
|
EXCEPTION WHEN undefined_column THEN
|
|
is_raster = FALSE;
|
|
END;
|
|
|
|
RETURN is_raster;
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
|
|
|
|
-- ////////////////////////////////////////////////////
|
|
|
|
-- Ensure a table is a "cartodb" table (See https://github.com/CartoDB/cartodb/wiki/CartoDB-user-table)
|
|
-- Rails code replicates this call at User.cartodbfy()
|
|
CREATE OR REPLACE FUNCTION CDB_CartodbfyTable(schema_name TEXT, reloid REGCLASS)
|
|
RETURNS void
|
|
AS $$
|
|
DECLARE
|
|
exists_geom_cols BOOLEAN[];
|
|
is_raster BOOLEAN;
|
|
BEGIN
|
|
|
|
PERFORM cartodb._CDB_check_prerequisites(schema_name, reloid);
|
|
|
|
PERFORM cartodb._CDB_drop_triggers(reloid);
|
|
|
|
-- Ensure required fields exist
|
|
PERFORM cartodb._CDB_create_cartodb_id_column(reloid);
|
|
|
|
SELECT cartodb._CDB_is_raster_table(schema_name, reloid) INTO is_raster;
|
|
IF is_raster THEN
|
|
PERFORM cartodb._CDB_create_raster_triggers(schema_name, reloid);
|
|
ELSE
|
|
SELECT cartodb._CDB_create_the_geom_columns(reloid) INTO exists_geom_cols;
|
|
|
|
-- Both only populate if proceeds
|
|
PERFORM cartodb._CDB_populate_the_geom_from_the_geom_webmercator(reloid, exists_geom_cols);
|
|
PERFORM cartodb._CDB_populate_the_geom_webmercator_from_the_geom(reloid, exists_geom_cols);
|
|
|
|
PERFORM cartodb._CDB_create_triggers(schema_name, reloid);
|
|
END IF;
|
|
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
CREATE OR REPLACE FUNCTION CDB_CartodbfyTable(reloid REGCLASS)
|
|
RETURNS void
|
|
AS $$
|
|
BEGIN
|
|
PERFORM cartodb.CDB_CartodbfyTable('public', reloid);
|
|
END;
|
|
$$ LANGUAGE PLPGSQL;
|
|
|
|
|
|
-- -=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
|
|
--
|
|
-- NEW CARTODBFY CODE FROM HERE ON DOWN
|
|
--
|
|
-- -=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
|
|
--
|
|
-- CDB_CartodbfyTable2(reloid REGCLASS, destschema TEXT DEFAULT NULL)
|
|
--
|
|
-- Main function, calls the following functions, with a little
|
|
-- logic before the table re-write to avoid re-writing if the table
|
|
-- already has all the necessary columns in place.
|
|
--
|
|
-- (1) _CDB_drop_triggers
|
|
-- As before, this drops all the metadata and geom sync triggers
|
|
--
|
|
-- (2) _CDB_Has_Usable_Primary_ID()
|
|
-- Returns TRUE if it can find a unique integer primary key named
|
|
-- 'cartodb_id' or can rename an existing key.
|
|
-- Returns FALSE otherwise.
|
|
--
|
|
-- (3) _CDB_Has_Usable_Geom()
|
|
-- Looks for existing EPSG:4326 and EPSG:3857 geometry columns, and
|
|
-- renames them to the standard names if it can find them, returning TRUE.
|
|
-- If it cannot find both columns in the right EPSG, returns FALSE.
|
|
--
|
|
-- (4) _CDB_Rewrite_Table()
|
|
-- If table does not have a usable primary key and both usable geom
|
|
-- columns it needs to be re-written. Function constructs an appropriate
|
|
-- CREATE TABLE AS SELECT... query and executes it.
|
|
--
|
|
-- (5) _CDB_Add_Indexes()
|
|
-- Checks the primary key column for primary key constraint, adds it if
|
|
-- missing. Check geometry columns for GIST indexes and adds them if missing.
|
|
--
|
|
-- (6) _CDB_create_triggers()
|
|
-- Adds the system metadata and geometry column update triggers back
|
|
-- onto the table.
|
|
--
|
|
-- -=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
|
|
|
|
|
|
CREATE OR REPLACE FUNCTION _CDB_Columns(OUT pkey TEXT, OUT geomcol TEXT, OUT mercgeomcol TEXT)
|
|
RETURNS record
|
|
AS $$
|
|
BEGIN
|
|
|
|
pkey := 'cartodb_id';
|
|
geomcol := 'the_geom';
|
|
mercgeomcol := 'the_geom_webmercator';
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
CREATE OR REPLACE FUNCTION _CDB_Error(message TEXT, funcname TEXT DEFAULT '_CDB_Error')
|
|
RETURNS void
|
|
AS $$
|
|
BEGIN
|
|
|
|
RAISE EXCEPTION 'CDB(%): %', funcname, message;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
CREATE OR REPLACE FUNCTION _CDB_SQL(sql TEXT, funcname TEXT DEFAULT '_CDB_SQL')
|
|
RETURNS void
|
|
AS $$
|
|
BEGIN
|
|
|
|
RAISE DEBUG 'CDB(%): %', funcname, sql;
|
|
EXECUTE sql;
|
|
|
|
EXCEPTION
|
|
WHEN others THEN
|
|
RAISE EXCEPTION 'CDB(%:%:%): %', funcname, SQLSTATE, SQLERRM, sql;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
-- Find a unique relation name in the given schema, starting from the
|
|
-- template given. If the template is already unique, just return it;
|
|
-- otherwise, append an increasing integer until you find a unique variant.
|
|
CREATE OR REPLACE FUNCTION _CDB_Unique_Relation_Name(schemaname TEXT, relationname TEXT)
|
|
RETURNS TEXT
|
|
AS $$
|
|
DECLARE
|
|
rec RECORD;
|
|
i INTEGER;
|
|
newrelname TEXT;
|
|
BEGIN
|
|
|
|
i := 0;
|
|
newrelname := relationname;
|
|
LOOP
|
|
|
|
SELECT c.relname, n.nspname
|
|
INTO rec
|
|
FROM pg_class c
|
|
JOIN pg_namespace n ON c.relnamespace = n.oid
|
|
WHERE c.relname = newrelname
|
|
AND n.nspname = schemaname;
|
|
|
|
IF NOT FOUND THEN
|
|
RETURN newrelname;
|
|
END IF;
|
|
|
|
i := i + 1;
|
|
newrelname := relationname || '_' || i;
|
|
|
|
IF i > 100 THEN
|
|
PERFORM _CDB_Error('looping too far', '_CDB_Unique_Relation_Name');
|
|
END IF;
|
|
|
|
END LOOP;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
-- Find a unique column name in the given relation, starting from the
|
|
-- column name given. If the column name is already unique, just return it;
|
|
-- otherwise, append an increasing integer until you find a unique variant.
|
|
CREATE OR REPLACE FUNCTION _CDB_Unique_Column_Name(reloid REGCLASS, columnname TEXT)
|
|
RETURNS TEXT
|
|
AS $$
|
|
DECLARE
|
|
rec RECORD;
|
|
i INTEGER;
|
|
newcolname TEXT;
|
|
BEGIN
|
|
|
|
i := 0;
|
|
newcolname := columnname;
|
|
LOOP
|
|
|
|
SELECT a.attname
|
|
INTO rec
|
|
FROM pg_class c
|
|
JOIN pg_attribute a ON a.attrelid = c.oid
|
|
WHERE NOT a.attisdropped
|
|
AND a.attnum > 0
|
|
AND c.oid = reloid
|
|
AND a.attname = newcolname;
|
|
|
|
IF NOT FOUND THEN
|
|
RETURN newcolname;
|
|
END IF;
|
|
|
|
i := i + 1;
|
|
newcolname := columnname || '_' || i;
|
|
|
|
IF i > 100 THEN
|
|
PERFORM _CDB_Error('looping too far', '_CDB_Unique_Column_Name');
|
|
END IF;
|
|
|
|
END LOOP;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
-- Return the geometry SRID from the column metadata or
|
|
-- the geometry of the very first entry in a given column.
|
|
CREATE OR REPLACE FUNCTION _CDB_Geometry_SRID(reloid REGCLASS, columnname TEXT)
|
|
RETURNS INTEGER
|
|
AS $$
|
|
DECLARE
|
|
rec RECORD;
|
|
BEGIN
|
|
|
|
RAISE DEBUG 'CDB(%): %', '_CDB_Geometry_SRID', 'entered function';
|
|
|
|
EXECUTE Format('SELECT ST_SRID(%s) AS srid FROM %s LIMIT 1', columnname, reloid::text)
|
|
INTO rec;
|
|
|
|
IF FOUND THEN
|
|
RETURN rec.srid;
|
|
END IF;
|
|
|
|
RETURN 0;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
-- Find out if the table already has a usable primary key
|
|
-- If the table has both a usable key and usable geometry
|
|
-- we can no-op on the table copy and just ensure that the
|
|
-- indexes and triggers are in place
|
|
CREATE OR REPLACE FUNCTION _CDB_Has_Usable_Primary_ID(reloid REGCLASS)
|
|
RETURNS BOOLEAN
|
|
AS $$
|
|
DECLARE
|
|
rec RECORD;
|
|
const RECORD;
|
|
i INTEGER;
|
|
sql TEXT;
|
|
useable_key BOOLEAN = false;
|
|
BEGIN
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Primary_ID): %', 'entered function';
|
|
|
|
-- Read in the names of the CartoDB columns
|
|
const := _CDB_Columns();
|
|
|
|
-- Do we already have a properly named column?
|
|
SELECT a.attname, i.indisprimary, i.indisunique, a.attnotnull, a.atttypid
|
|
INTO rec
|
|
FROM pg_class c
|
|
JOIN pg_attribute a ON a.attrelid = c.oid
|
|
JOIN pg_type t ON a.atttypid = t.oid
|
|
LEFT JOIN pg_index i ON c.oid = i.indrelid AND a.attnum = ANY(i.indkey)
|
|
WHERE c.oid = reloid
|
|
AND NOT a.attisdropped
|
|
AND a.attname = const.pkey;
|
|
|
|
-- Found something named right...
|
|
IF FOUND THEN
|
|
|
|
-- And it's an integer column...
|
|
IF rec.atttypid IN (20,21,23) THEN
|
|
|
|
-- And it's a unique primary key! Done!
|
|
IF rec.indisprimary AND rec.indisunique AND rec.attnotnull THEN
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Primary_ID): %', Format('found good ''%s''', const.pkey);
|
|
RETURN true;
|
|
|
|
-- Check and see if the column values are unique,
|
|
-- if they are, we can use this column...
|
|
ELSE
|
|
|
|
-- Assume things are OK until proven otherwise...
|
|
useable_key := true;
|
|
|
|
BEGIN
|
|
sql := Format('ALTER TABLE %s ADD CONSTRAINT %s_unique UNIQUE (%s)', reloid::text, const.pkey, const.pkey);
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Primary_ID): %', sql;
|
|
EXECUTE sql;
|
|
EXCEPTION
|
|
-- Failed unique check...
|
|
WHEN unique_violation THEN
|
|
RAISE NOTICE 'CDB(_CDB_Has_Usable_Primary_ID): %', Format('column %s is not unique', const.pkey);
|
|
useable_key := false;
|
|
-- Other fatal error
|
|
WHEN others THEN
|
|
PERFORM _CDB_Error(sql, '_CDB_Has_Usable_Primary_ID');
|
|
END;
|
|
|
|
-- Clean up test constraint
|
|
IF useable_key THEN
|
|
PERFORM _CDB_SQL(Format('ALTER TABLE %s DROP CONSTRAINT %s_unique', reloid::text, const.pkey));
|
|
|
|
-- Move non-unique column out of the way
|
|
ELSE
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Primary_ID): %',
|
|
Format('found non-unique ''%s'', renaming it', const.pkey);
|
|
|
|
PERFORM _CDB_SQL(
|
|
Format('ALTER TABLE %s RENAME COLUMN %s TO %s',
|
|
reloid::text, rec.attname,
|
|
_CDB_Unique_Column_Name(reloid, const.pkey)),
|
|
'_CDB_Has_Usable_Primary_ID');
|
|
|
|
END IF;
|
|
|
|
return useable_key;
|
|
|
|
END IF;
|
|
|
|
-- It's not an integer column, we have to rename it
|
|
ELSE
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Primary_ID): %',
|
|
Format('found non-integer ''%s'', renaming it', const.pkey);
|
|
|
|
PERFORM _CDB_SQL(
|
|
Format('ALTER TABLE %s RENAME COLUMN %s TO %s',
|
|
reloid::text, rec.attname, _CDB_Unique_Column_Name(reloid, const.pkey)),
|
|
'_CDB_Has_Usable_Primary_ID');
|
|
|
|
END IF;
|
|
|
|
-- There's no column there named pkey
|
|
ELSE
|
|
|
|
-- Is there another suitable primary key already?
|
|
SELECT a.attname
|
|
INTO rec
|
|
FROM pg_class c
|
|
JOIN pg_attribute a ON a.attrelid = c.oid
|
|
JOIN pg_type t ON a.atttypid = t.oid
|
|
LEFT JOIN pg_index i ON c.oid = i.indrelid AND a.attnum = ANY(i.indkey)
|
|
WHERE c.oid = reloid AND NOT a.attisdropped
|
|
AND i.indisprimary AND i.indisunique AND a.attnotnull AND a.atttypid IN (20,21,23);
|
|
|
|
-- Yes! Ok, rename it.
|
|
IF FOUND THEN
|
|
PERFORM _CDB_SQL(Format('ALTER TABLE %s RENAME COLUMN %s TO %s', reloid::text, rec.attname, const.pkey),'_CDB_Has_Usable_Primary_ID');
|
|
RETURN true;
|
|
ELSE
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Primary_ID): %',
|
|
Format('found no useful column for ''%s''', const.pkey);
|
|
END IF;
|
|
|
|
END IF;
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Primary_ID): %', 'function complete';
|
|
|
|
-- Didn't find re-usable key, so return FALSE
|
|
RETURN false;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
CREATE OR REPLACE FUNCTION _CDB_Has_Usable_Geom(reloid REGCLASS)
|
|
RETURNS BOOLEAN
|
|
AS $$
|
|
DECLARE
|
|
r1 RECORD;
|
|
r2 RECORD;
|
|
const RECORD;
|
|
has_geom BOOLEAN := false;
|
|
has_mercgeom BOOLEAN := false;
|
|
has_geom_name TEXT;
|
|
has_mercgeom_name TEXT;
|
|
str TEXT;
|
|
sql TEXT;
|
|
BEGIN
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Geom): %', 'entered function';
|
|
|
|
-- Read in the names of the CartoDB columns
|
|
const := _CDB_Columns();
|
|
|
|
-- Do we have a column we can use?
|
|
FOR r1 IN
|
|
SELECT
|
|
a.attname,
|
|
CASE WHEN t.typname = 'geometry' THEN postgis_typmod_srid(a.atttypmod) ELSE NULL END AS srid,
|
|
t.typname,
|
|
f.desired_attname, f.desired_srid
|
|
FROM pg_class c
|
|
JOIN pg_attribute a ON a.attrelid = c.oid
|
|
JOIN pg_type t ON a.atttypid = t.oid,
|
|
(VALUES (const.geomcol, 4326), (const.mercgeomcol, 3857) ) as f(desired_attname, desired_srid)
|
|
WHERE c.oid = reloid
|
|
AND a.attnum > 0
|
|
AND NOT a.attisdropped
|
|
AND postgis_typmod_srid(a.atttypmod) IN (4326, 3857, 0)
|
|
ORDER BY t.oid ASC
|
|
LOOP
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Geom): %', Format('checking column ''%s''', r1.attname);
|
|
|
|
-- Name collision: right name but wrong type, rename it!
|
|
IF r1.typname != 'geometry' AND r1.attname = r1.desired_attname THEN
|
|
str := _CDB_Unique_Column_Name(reloid, r1.attname);
|
|
sql := Format('ALTER TABLE %s RENAME COLUMN %s TO %s', reloid::text, r1.attname, str);
|
|
PERFORM _CDB_SQL(sql,'_CDB_Has_Usable_Geom');
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Geom): %',
|
|
Format('%s is the wrong type, renamed to %s', r1.attname, str);
|
|
|
|
-- Found a geometry column!
|
|
ELSIF r1.typname = 'geometry' THEN
|
|
|
|
-- If it's the right SRID, we can use it in place without
|
|
-- transforming it!
|
|
IF r1.srid = r1.desired_srid OR _CDB_Geometry_SRID(reloid, r1.attname) = r1.desired_srid THEN
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Has_Usable_Geom): %', Format('found acceptable ''%s''', r1.attname);
|
|
|
|
IF r1.desired_attname = const.geomcol THEN
|
|
has_geom := true;
|
|
has_geom_name := r1.attname;
|
|
ELSIF r1.desired_attname = const.mercgeomcol THEN
|
|
has_mercgeom := true;
|
|
has_mercgeom_name := r1.attname;
|
|
END IF;
|
|
|
|
END IF;
|
|
|
|
END IF;
|
|
|
|
END LOOP;
|
|
|
|
-- If geom is the wrong name, just rename it.
|
|
IF has_geom AND has_geom_name != const.geomcol THEN
|
|
sql := Format('ALTER TABLE %s RENAME COLUMN %s TO %s', reloid::text, has_geom_name, const.geomcol);
|
|
PERFORM _CDB_SQL(sql,'_CDB_Has_Usable_Geom');
|
|
END IF;
|
|
|
|
-- If mercgeom is the wrong name, just rename it.
|
|
IF has_mercgeom AND has_mercgeom_name != const.mercgeomcol THEN
|
|
sql := Format('ALTER TABLE %s RENAME COLUMN %s TO %s', reloid::text, has_mercgeom_name, const.mercgeomcol);
|
|
PERFORM _CDB_SQL(sql,'_CDB_Has_Usable_Geom');
|
|
END IF;
|
|
|
|
-- If table is perfect (no transforms required), return TRUE!
|
|
RETURN has_geom AND has_mercgeom;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
-- Create a copy of the table. Assumes that the "Has usable" functions
|
|
-- have already been run, so that if there is a 'cartodb_id' column, it is
|
|
-- a "good" one, and the same for the geometry columns. If all the required
|
|
-- columns are in place already, it no-ops and just renames the table to
|
|
-- the destination if necessary.
|
|
CREATE OR REPLACE FUNCTION _CDB_Rewrite_Table(reloid REGCLASS, destschema TEXT DEFAULT NULL)
|
|
RETURNS BOOLEAN
|
|
AS $$
|
|
DECLARE
|
|
|
|
relname TEXT;
|
|
relschema TEXT;
|
|
|
|
destoid REGCLASS;
|
|
destname TEXT;
|
|
destseq TEXT;
|
|
destseqmax INTEGER;
|
|
|
|
salt TEXT := md5(random()::text || now());
|
|
copyname TEXT;
|
|
|
|
column_name_sql TEXT;
|
|
geom_transform_sql TEXT := NULL;
|
|
geom_column_source TEXT := '';
|
|
|
|
rec RECORD;
|
|
const RECORD;
|
|
sql TEXT;
|
|
str TEXT;
|
|
table_srid INTEGER;
|
|
|
|
has_usable_primary_key BOOLEAN;
|
|
has_usable_geoms BOOLEAN;
|
|
|
|
BEGIN
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Rewrite_Table): %', 'entered function';
|
|
|
|
-- Read CartoDB standard column names in
|
|
const := _CDB_Columns();
|
|
|
|
-- Save the raw schema/table names for later
|
|
SELECT n.nspname, c.relname, c.relname
|
|
INTO STRICT relschema, relname, destname
|
|
FROM pg_class c JOIN pg_namespace n ON c.relnamespace = n.oid
|
|
WHERE c.oid = reloid;
|
|
|
|
-- Default the destination to current schema if unspecified
|
|
IF destschema IS NULL THEN
|
|
destschema := relschema;
|
|
END IF;
|
|
|
|
-- See if there is a primary key column we need to carry along to the
|
|
-- new table. If this is true, it implies there is an indexed
|
|
-- primary key of integer type named (by default) cartodb_id
|
|
SELECT _CDB_Has_Usable_Primary_ID(reloid) AS has_usable_primary_key
|
|
INTO STRICT has_usable_primary_key;
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Rewrite_Table): has_usable_primary_key %', has_usable_primary_key;
|
|
|
|
-- See if the geometry columns we need are already available
|
|
-- on the table. If they are, we don't need to do any bulk
|
|
-- transformation of the table, we can just ensure proper
|
|
-- indexes are in place and apply a rename
|
|
SELECT _CDB_Has_Usable_Geom(reloid) AS has_usable_geoms
|
|
INTO STRICT has_usable_geoms;
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Rewrite_Table): has_usable_geoms %', has_usable_geoms;
|
|
|
|
-- We can only avoid a rewrite if both the key and
|
|
-- geometry are usable
|
|
|
|
-- No table re-write is required, BUT a rename is required to
|
|
-- a destination schema, so do that now
|
|
IF has_usable_primary_key AND has_usable_geoms AND destschema != relschema THEN
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Rewrite_Table): perfect table needs to be moved to schema (%)', destschema;
|
|
PERFORM _CDB_SQL(Format('ALTER TABLE %s SET SCHEMA %s', reloid::text, destschema), '_CDB_Rewrite_Table');
|
|
RETURN true;
|
|
|
|
-- Don't move anything, just make sure our destination information is set right
|
|
ELSIF has_usable_primary_key AND has_usable_geoms AND destschema = relschema THEN
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Rewrite_Table): perfect table in the perfect place';
|
|
RETURN true;
|
|
|
|
END IF;
|
|
|
|
-- We must rewrite, so here we go...
|
|
|
|
|
|
-- Put the primary key sequence in the right schema
|
|
-- If the new table is not moving, better ensure the sequence name
|
|
-- is unique
|
|
destseq := relname || '_' || const.pkey || '_seq';
|
|
destseq := _CDB_Unique_Relation_Name(destschema, destseq);
|
|
destseq := Format('%s.%s', destschema, destseq);
|
|
PERFORM _CDB_SQL(Format('CREATE SEQUENCE %s', destseq), '_CDB_Rewrite_Table');
|
|
|
|
-- Salt a temporary table name if we are re-writing in place
|
|
IF destschema = relschema THEN
|
|
copyname := destschema || '.' || destname || '_' || salt;
|
|
ELSE
|
|
copyname := destschema || '.' || destname;
|
|
END IF;
|
|
|
|
-- Start building the SQL!
|
|
sql := 'CREATE TABLE ' || copyname || ' AS SELECT ';
|
|
|
|
-- Add cartodb ID!
|
|
IF has_usable_primary_key THEN
|
|
sql := sql || const.pkey;
|
|
ELSE
|
|
sql := sql || 'nextval(''' || destseq || ''') AS ' || const.pkey;
|
|
END IF;
|
|
|
|
-- Add the geometry columns!
|
|
IF has_usable_geoms THEN
|
|
sql := sql || ',' || const.geomcol || ',' || const.mercgeomcol;
|
|
ELSE
|
|
|
|
-- This gets complicated: we have to make sure the
|
|
-- geometry column we are using can be transformed into
|
|
-- geographics, which means it needs to have a valid
|
|
-- SRID.
|
|
SELECT a.attname
|
|
INTO rec
|
|
FROM pg_class c
|
|
JOIN pg_attribute a ON a.attrelid = c.oid
|
|
JOIN pg_type t ON a.atttypid = t.oid
|
|
WHERE c.oid = reloid
|
|
AND t.typname = 'geometry'
|
|
AND a.attnum > 0
|
|
AND NOT a.attisdropped
|
|
ORDER BY a.attnum
|
|
LIMIT 1;
|
|
|
|
IF NOT FOUND THEN
|
|
-- If there is no geometry column, we continue making a
|
|
-- non-spatial table. This is important for folks who want
|
|
-- their tables to invalidate the SQL API
|
|
-- cache on update/insert/delete.
|
|
geom_column_source := '';
|
|
|
|
ELSE
|
|
|
|
-- table_srid = _CDB_Geometry_SRID(reloid, rec.attname);
|
|
|
|
EXECUTE Format('SELECT ST_SRID(%s) AS srid FROM %s LIMIT 1', rec.attname, reloid::text)
|
|
INTO rec;
|
|
|
|
-- The geometry columns weren't in the right projection,
|
|
-- so we need to find the first decent geometry column
|
|
-- in the table and wrap it in two transforms, one to 4326
|
|
-- and another to 3857. Then remember its name so we can
|
|
-- ignore it when we build the list of other columns to
|
|
-- add to the output table
|
|
SELECT ',ST_Transform('
|
|
|| a.attname
|
|
|| ',4326)::Geometry('
|
|
|| postgis_typmod_type(a.atttypmod)
|
|
|| ', 4326) AS '
|
|
|| const.geomcol
|
|
|| ', ST_Transform('
|
|
|| a.attname
|
|
|| ',3857)::Geometry('
|
|
|| postgis_typmod_type(a.atttypmod)
|
|
|| ', 3857) AS '
|
|
|| const.mercgeomcol,
|
|
a.attname
|
|
INTO geom_transform_sql, geom_column_source
|
|
FROM pg_class c
|
|
JOIN pg_attribute a ON a.attrelid = c.oid
|
|
JOIN pg_type t ON a.atttypid = t.oid,
|
|
( SELECT rec.srid AS srid ) AS srid
|
|
-- ( SELECT table_srid AS srid ) AS srid
|
|
WHERE c.oid = reloid
|
|
AND t.typname = 'geometry'
|
|
AND a.attnum > 0
|
|
AND NOT a.attisdropped
|
|
AND (postgis_typmod_srid(a.atttypmod) > 0 OR srid.srid > 0)
|
|
ORDER BY a.attnum
|
|
LIMIT 1;
|
|
|
|
IF FOUND THEN
|
|
sql := sql || geom_transform_sql;
|
|
END IF;
|
|
|
|
END IF;
|
|
|
|
END IF;
|
|
|
|
-- Add now add all the rest of the columns
|
|
-- by selecting their names into an array and
|
|
-- joining the array with a comma
|
|
SELECT
|
|
',' || array_to_string(array_agg(a.attname),',') AS column_name_sql,
|
|
Count(*) AS count
|
|
INTO rec
|
|
FROM pg_class c
|
|
JOIN pg_attribute a ON a.attrelid = c.oid
|
|
JOIN pg_type t ON a.atttypid = t.oid
|
|
WHERE c.oid = reloid
|
|
AND a.attnum > 0
|
|
AND a.attname NOT IN (const.geomcol, const.mercgeomcol, const.pkey, geom_column_source)
|
|
AND NOT a.attisdropped;
|
|
|
|
|
|
-- No non-cartodb columns? Possible, I guess.
|
|
IF rec.count = 0 THEN
|
|
RAISE DEBUG 'CDB(_CDB_Rewrite_Table): %', 'found no extra columns';
|
|
column_name_sql := '';
|
|
ELSE
|
|
RAISE DEBUG 'CDB(_CDB_Rewrite_Table): %', Format('found extra columns columns ''%s''', rec.column_name_sql);
|
|
column_name_sql := rec.column_name_sql;
|
|
END IF;
|
|
|
|
-- Add the source table to the SQL
|
|
sql := sql || column_name_sql || ' FROM ' || reloid::text;
|
|
RAISE DEBUG 'CDB(_CDB_Rewrite_Table): %', sql;
|
|
|
|
-- Run it!
|
|
PERFORM _CDB_SQL(sql, '_CDB_Rewrite_Table');
|
|
|
|
-- Set up the primary key sequence
|
|
-- If we copied the primary key from the original data, we need
|
|
-- to set the sequence to the maximum value of that key
|
|
IF has_usable_primary_key THEN
|
|
|
|
EXECUTE Format('SELECT max(%s) FROM %s',
|
|
const.pkey, copyname)
|
|
INTO destseqmax;
|
|
|
|
IF FOUND AND destseqmax IS NOT NULL THEN
|
|
PERFORM _CDB_SQL(Format('SELECT setval(''%s'', %s)', destseq, destseqmax), '_CDB_Rewrite_Table');
|
|
END IF;
|
|
|
|
END IF;
|
|
|
|
-- Make the primary key use the sequence as its default value
|
|
sql := Format('ALTER TABLE %s ALTER COLUMN %I SET DEFAULT nextval(''%s'')',
|
|
copyname, const.pkey, destseq);
|
|
PERFORM _CDB_SQL(sql, '_CDB_Rewrite_Table');
|
|
|
|
-- Make the sequence owned by the table, so when the table drops,
|
|
-- the sequence does too
|
|
sql := Format('ALTER SEQUENCE %s OWNED BY %s.%s', destseq, copyname, const.pkey);
|
|
PERFORM _CDB_SQL(sql,'_CDB_Rewrite_Table');
|
|
|
|
-- We just made a copy, so we can drop the original now
|
|
sql := Format('DROP TABLE %s', reloid::text);
|
|
PERFORM _CDB_SQL(sql, '_CDB_Rewrite_Table');
|
|
|
|
-- If we used a temporary destination table
|
|
-- we can now rename it into place
|
|
IF destschema = relschema THEN
|
|
sql := Format('ALTER TABLE %s RENAME TO %s', copyname, destname);
|
|
PERFORM _CDB_SQL(sql, '_CDB_Rewrite_Table');
|
|
END IF;
|
|
|
|
RETURN true;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
-- Assumes the table already has the right metadata columns
|
|
-- (primary key and two geometry columns) and adds primary key
|
|
-- and geometry indexes if necessary.
|
|
CREATE OR REPLACE FUNCTION _CDB_Add_Indexes(reloid REGCLASS)
|
|
RETURNS BOOLEAN
|
|
AS $$
|
|
DECLARE
|
|
rec RECORD;
|
|
const RECORD;
|
|
iname TEXT;
|
|
sql TEXT;
|
|
relname TEXT;
|
|
BEGIN
|
|
|
|
RAISE DEBUG 'CDB(_CDB_Add_Indexes): %', 'entered function';
|
|
|
|
-- Read CartoDB standard column names in
|
|
const := _CDB_Columns();
|
|
|
|
-- Extract just the relname to use for the index names
|
|
SELECT c.relname
|
|
INTO STRICT relname
|
|
FROM pg_class c
|
|
WHERE c.oid = reloid;
|
|
|
|
-- Is there already a primary key on this table for
|
|
-- a column other than our chosen primary key?
|
|
SELECT ci.relname AS pkey
|
|
INTO rec
|
|
FROM pg_class c
|
|
JOIN pg_attribute a ON a.attrelid = c.oid
|
|
LEFT JOIN pg_index i ON c.oid = i.indrelid AND a.attnum = ANY(i.indkey)
|
|
JOIN pg_class ci ON i.indexrelid = ci.oid
|
|
WHERE c.oid = reloid
|
|
AND NOT a.attisdropped
|
|
AND a.attname != const.pkey
|
|
AND i.indisprimary;
|
|
|
|
-- Yes? Then drop it, we're adding our own PK to the column
|
|
-- we prefer.
|
|
IF FOUND THEN
|
|
RAISE DEBUG 'CDB(_CDB_Add_Indexes): dropping unwanted primary key ''%''', rec.pkey;
|
|
sql := Format('ALTER TABLE %s DROP CONSTRAINT IF EXISTS %s', reloid::text, rec.pkey);
|
|
PERFORM _CDB_SQL(sql, '_CDB_Add_Indexes');
|
|
END IF;
|
|
|
|
|
|
-- Is the default primary key flagged as primary?
|
|
SELECT a.attname
|
|
INTO rec
|
|
FROM pg_class c
|
|
JOIN pg_attribute a ON a.attrelid = c.oid
|
|
JOIN pg_index i ON c.oid = i.indrelid AND a.attnum = ANY(i.indkey)
|
|
JOIN pg_class ci ON ci.oid = i.indexrelid
|
|
WHERE attnum > 0
|
|
AND c.oid = reloid
|
|
AND a.attname = const.pkey
|
|
AND i.indisprimary
|
|
AND i.indisunique
|
|
AND NOT attisdropped;
|
|
|
|
-- No primary key? Add one.
|
|
IF NOT FOUND THEN
|
|
sql := Format('ALTER TABLE %s ADD PRIMARY KEY (%s)', reloid::text, const.pkey);
|
|
PERFORM _CDB_SQL(sql, '_CDB_Add_Indexes');
|
|
END IF;
|
|
|
|
-- Add geometry indexes to all "special geometry columns" that
|
|
-- don't have one (either have no index at all, or have a non-GIST index)
|
|
FOR rec IN
|
|
SELECT a.attname, n.nspname
|
|
FROM pg_class c
|
|
JOIN pg_namespace n ON n.oid = c.relnamespace
|
|
JOIN pg_attribute a ON a.attrelid = c.oid AND attnum > 0
|
|
LEFT JOIN pg_index i ON c.oid = i.indrelid AND a.attnum = ANY(i.indkey)
|
|
WHERE NOT attisdropped
|
|
AND a.attname IN (const.geomcol, const.mercgeomcol)
|
|
AND c.oid = reloid
|
|
AND i.indexrelid IS NULL
|
|
UNION
|
|
SELECT a.attname, n.nspname
|
|
FROM pg_class c
|
|
JOIN pg_namespace n ON n.oid = c.relnamespace
|
|
JOIN pg_attribute a ON a.attrelid = c.oid AND attnum > 0
|
|
JOIN pg_index i ON c.oid = i.indrelid AND a.attnum = ANY(i.indkey)
|
|
JOIN pg_class ci ON ci.oid = i.indexrelid
|
|
JOIN pg_am am ON ci.relam = am.oid
|
|
WHERE NOT attisdropped
|
|
AND a.attname IN (const.geomcol, const.mercgeomcol)
|
|
AND c.oid = reloid
|
|
AND am.amname != 'gist'
|
|
LOOP
|
|
sql := Format('CREATE INDEX %s_%s_gix ON %s USING GIST (%s)', relname, rec.attname, reloid::text, rec.attname);
|
|
PERFORM _CDB_SQL(sql, '_CDB_Add_Indexes');
|
|
END LOOP;
|
|
|
|
RETURN true;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|
|
|
|
|
|
CREATE OR REPLACE FUNCTION CDB_CartodbfyTable2(reloid REGCLASS, destschema TEXT DEFAULT NULL)
|
|
RETURNS void
|
|
AS $$
|
|
DECLARE
|
|
|
|
is_raster BOOLEAN;
|
|
relname TEXT;
|
|
relschema TEXT;
|
|
|
|
destoid REGCLASS;
|
|
destname TEXT;
|
|
|
|
rec RECORD;
|
|
|
|
BEGIN
|
|
|
|
-- Save the raw schema/table names for later
|
|
SELECT n.nspname, c.relname, c.relname
|
|
INTO STRICT relschema, relname, destname
|
|
FROM pg_class c JOIN pg_namespace n ON c.relnamespace = n.oid
|
|
WHERE c.oid = reloid;
|
|
|
|
-- Check destination schema exists
|
|
-- Throws an exception of there is no matching schema
|
|
IF destschema IS NOT NULL THEN
|
|
|
|
SELECT n.nspname
|
|
INTO rec FROM pg_namespace n WHERE n.nspname = destschema;
|
|
IF NOT FOUND THEN
|
|
RAISE EXCEPTION 'Schema ''%'' does not exist', destschema;
|
|
END IF;
|
|
|
|
ELSE
|
|
destschema := relschema;
|
|
END IF;
|
|
|
|
-- Drop triggers first
|
|
PERFORM _CDB_drop_triggers(reloid);
|
|
|
|
-- Rasters only get a cartodb_id and a limited selection of triggers
|
|
-- underlying assumption is that they are already formed up correctly
|
|
SELECT cartodb._CDB_is_raster_table(destschema, reloid) INTO is_raster;
|
|
IF is_raster THEN
|
|
|
|
PERFORM cartodb._CDB_create_cartodb_id_column(reloid);
|
|
PERFORM cartodb._CDB_create_raster_triggers(destschema, reloid);
|
|
|
|
ELSE
|
|
|
|
-- Rewrite (or rename) the table to the new location
|
|
PERFORM _CDB_Rewrite_Table(reloid, destschema);
|
|
|
|
-- The old regclass might not be valid anymore if we re-wrote the table...
|
|
destoid := (destschema || '.' || destname)::regclass;
|
|
|
|
-- Add indexes to the destination table, as necessary
|
|
PERFORM _CDB_Add_Indexes(destoid);
|
|
|
|
-- Add triggers to the destination table, as necessary
|
|
PERFORM _CDB_create_triggers(destschema, destoid);
|
|
|
|
END IF;
|
|
|
|
END;
|
|
$$ LANGUAGE 'plpgsql';
|