231 lines
9.5 KiB
PL/PgSQL
231 lines
9.5 KiB
PL/PgSQL
SET client_min_messages TO error;
|
|
\set VERBOSITY terse;
|
|
|
|
CREATE OR REPLACE FUNCTION CDB_CartodbfyTableCheck(tabname regclass, label text)
|
|
RETURNS text AS
|
|
$$
|
|
DECLARE
|
|
sql TEXT;
|
|
id INTEGER;
|
|
rec RECORD;
|
|
lag INTERVAL;
|
|
tmp INTEGER;
|
|
ogc_geom geometry_columns; -- old the_geom record in geometry_columns
|
|
ogc_merc geometry_columns; -- old the_geom_webmercator record in geometry_columns
|
|
BEGIN
|
|
|
|
-- Save current constraints on geometry columns, if any
|
|
ogc_geom = ('','','','',0,0,'GEOMETRY');
|
|
ogc_merc = ogc_geom;
|
|
sql := 'SELECT gc.* FROM geometry_columns gc, pg_class c, pg_namespace n '
|
|
|| 'WHERE c.oid = ' || tabname::oid || ' AND n.oid = c.relnamespace'
|
|
|| ' AND gc.f_table_schema = n.nspname AND gc.f_table_name = c.relname'
|
|
|| ' AND gc.f_geometry_column IN ( ' || quote_literal('the_geom')
|
|
|| ',' || quote_literal('the_geom_webmercator') || ')';
|
|
FOR rec IN EXECUTE sql LOOP
|
|
IF rec.f_geometry_column = 'the_geom' THEN
|
|
ogc_geom := rec;
|
|
ELSE
|
|
ogc_merc := rec;
|
|
END IF;
|
|
END LOOP;
|
|
|
|
PERFORM CDB_CartodbfyTable('public', tabname);
|
|
|
|
sql := 'INSERT INTO ' || tabname::text || '(the_geom) values ( CDB_LatLng(2,1) ) RETURNING cartodb_id';
|
|
EXECUTE sql INTO STRICT id;
|
|
sql := 'SELECT created_at,updated_at,the_geom_webmercator FROM '
|
|
|| tabname::text || ' WHERE cartodb_id = ' || id;
|
|
EXECUTE sql INTO STRICT rec;
|
|
|
|
-- Check created_at and updated_at at creation time
|
|
lag = rec.created_at - now();
|
|
IF lag > '1 second' THEN
|
|
RAISE EXCEPTION 'created_at not defaulting to now() after insert [ valued % ago ]', lag;
|
|
END IF;
|
|
lag = rec.updated_at - now();
|
|
IF lag > '1 second' THEN
|
|
RAISE EXCEPTION 'updated_at not defaulting to now() after insert [ valued % ago ]', lag;
|
|
END IF;
|
|
|
|
-- Check the_geom_webmercator trigger
|
|
IF round(st_x(rec.the_geom_webmercator)) != 111319 THEN
|
|
RAISE EXCEPTION 'the_geom_webmercator X is % (expecting 111319)', round(st_x(rec.the_geom_webmercator));
|
|
END IF;
|
|
IF round(st_y(rec.the_geom_webmercator)) != 222684 THEN
|
|
RAISE EXCEPTION 'the_geom_webmercator Y is % (expecting 222684)', round(st_y(rec.the_geom_webmercator));
|
|
END IF;
|
|
|
|
-- Check CDB_TableMetadata entry
|
|
sql := 'SELECT * FROM CDB_TableMetadata WHERE tabname = ' || tabname::oid;
|
|
EXECUTE sql INTO STRICT rec;
|
|
lag = rec.updated_at - now();
|
|
IF lag > '1 second' THEN
|
|
RAISE EXCEPTION 'updated_at in CDB_TableMetadata not set to now() after insert [ valued % ago ]', lag;
|
|
END IF;
|
|
|
|
-- Check geometry_columns entries
|
|
tmp := 0;
|
|
FOR rec IN
|
|
SELECT
|
|
CASE WHEN gc.f_geometry_column = 'the_geom' THEN 4326
|
|
ELSE 3857 END as expsrid,
|
|
CASE WHEN gc.f_geometry_column = 'the_geom' THEN ogc_geom.type
|
|
ELSE ogc_merc.type END as exptype, gc.*
|
|
FROM geometry_columns gc, pg_class c, pg_namespace n
|
|
WHERE c.oid = tabname::oid AND n.oid = c.relnamespace
|
|
AND gc.f_table_schema = n.nspname AND gc.f_table_name = c.relname
|
|
AND gc.f_geometry_column IN ( 'the_geom', 'the_geom_webmercator')
|
|
LOOP
|
|
tmp := tmp + 1;
|
|
-- Check SRID constraint
|
|
IF rec.srid != rec.expsrid THEN
|
|
RAISE EXCEPTION 'SRID of % in geometry_columns is %, expected %',
|
|
rec.f_geometry_column, rec.srid, rec.expsrid;
|
|
END IF;
|
|
-- Check TYPE constraint didn't change
|
|
IF rec.type != rec.exptype THEN
|
|
RAISE EXCEPTION 'type of % in geometry_columns is %, expected %',
|
|
rec.f_geometry_column, rec.type, rec.exptype;
|
|
END IF;
|
|
-- check coord_dimension ?
|
|
END LOOP;
|
|
IF tmp != 2 THEN
|
|
RAISE EXCEPTION '% entries found for table % in geometry_columns, expected 2', tmp, tabname;
|
|
END IF;
|
|
|
|
-- Check GiST index
|
|
sql := 'SELECT a.attname, count(ri.relname) FROM'
|
|
|| ' pg_index i, pg_class c, pg_class ri, pg_attribute a, pg_opclass o'
|
|
|| ' WHERE i.indrelid = c.oid AND ri.oid = i.indexrelid'
|
|
|| ' AND a.attrelid = ri.oid AND o.oid = i.indclass[0]'
|
|
|| ' AND a.attname IN ( ' || quote_literal('the_geom')
|
|
|| ',' || quote_literal('the_geom_webmercator') || ')'
|
|
|| ' AND ri.relnatts = 1 AND o.opcname = '
|
|
|| quote_literal('gist_geometry_ops_2d')
|
|
|| ' AND c.oid = ' || tabname::oid
|
|
|| ' GROUP BY a.attname';
|
|
RAISE NOTICE 'sql: %', sql;
|
|
EXECUTE sql;
|
|
GET DIAGNOSTICS tmp = ROW_COUNT;
|
|
IF tmp != 2 THEN
|
|
RAISE EXCEPTION '% gist indices found on the_geom and the_geom_webmercator, expected 2', tmp;
|
|
END IF;
|
|
|
|
-- Check null constraint on cartodb_id, created_at, updated_at
|
|
SELECT count(*) FROM pg_attribute a, pg_class c WHERE c.oid = tabname::oid
|
|
AND a.attrelid = c.oid AND NOT a.attisdropped AND a.attname in
|
|
( 'cartodb_id', 'created_at', 'updated_at' )
|
|
AND NOT a.attnotnull INTO strict tmp;
|
|
IF tmp > 0 THEN
|
|
RAISE EXCEPTION 'cartodb_id or created_at or updated_at are missing not-null constraint';
|
|
END IF;
|
|
|
|
-- Cleanup
|
|
sql := 'DELETE FROM ' || tabname::text || ' WHERE cartodb_id = ' || id;
|
|
EXECUTE sql;
|
|
|
|
RETURN label || ' cartodbfied fine';
|
|
END;
|
|
$$
|
|
LANGUAGE 'plpgsql';
|
|
|
|
-- table with single non-geometrical column
|
|
CREATE TABLE t AS SELECT 1::int as a;
|
|
SELECT CDB_CartodbfyTable('public', 't'); -- should fail
|
|
SELECT CDB_SetUserQuotaInBytes(0); -- Set user quota to infinite
|
|
SELECT CDB_CartodbfyTableCheck('t', 'single non-geometrical column');
|
|
DROP TABLE t;
|
|
|
|
-- table with existing srid-unconstrained (but type-constrained) the_geom
|
|
CREATE TABLE t AS SELECT ST_SetSRID(ST_MakePoint(0,0),4326)::geometry(point) as the_geom;
|
|
SELECT CDB_CartodbfyTableCheck('t', 'srid-unconstrained the_geom');
|
|
DROP TABLE t;
|
|
|
|
-- table with mixed-srid the_geom values
|
|
CREATE TABLE t AS SELECT ST_SetSRID(ST_MakePoint(-1,-1),4326) as the_geom
|
|
UNION ALL SELECT ST_SetSRID(ST_MakePoint(0,0),3857);
|
|
SELECT CDB_CartodbfyTableCheck('t', 'mixed-srid the_geom');
|
|
SELECT 'extent',ST_Extent(ST_SnapToGrid(the_geom,0.2)) FROM t;
|
|
DROP TABLE t;
|
|
|
|
-- table with wrong srid-constrained the_geom values
|
|
CREATE TABLE t AS SELECT 'SRID=3857;LINESTRING(222638.981586547 222684.208505545, 111319.490793274 111325.142866385)'::geometry(geometry,3857) as the_geom;
|
|
SELECT CDB_CartodbfyTableCheck('t', 'wrong srid-constrained the_geom');
|
|
SELECT 'extent',ST_Extent(ST_SnapToGrid(the_geom,0.2)),ST_Extent(ST_SnapToGrid(the_geom_webmercator,1)) FROM t;
|
|
DROP TABLE t;
|
|
|
|
-- table with wrong srid-constrained the_geom_webmercator values (and no the_geom!)
|
|
CREATE TABLE t AS SELECT 'SRID=4326;LINESTRING(1 1,2 2)'::geometry(geometry,4326) as the_geom_webmercator;
|
|
SELECT CDB_CartodbfyTableCheck('t', 'wrong srid-constrained the_geom_webmercator');
|
|
-- expect the_geom to be populated from the_geom_webmercator
|
|
SELECT 'extent',ST_Extent(ST_SnapToGrid(the_geom,0.2)) FROM t;
|
|
DROP TABLE t;
|
|
|
|
-- table with existing triggered the_geom
|
|
CREATE TABLE t AS SELECT 'SRID=4326;LINESTRING(1 1,2 2)'::geometry(geometry) as the_geom;
|
|
CREATE TRIGGER update_the_geom_webmercator_trigger BEFORE UPDATE OF the_geom ON t
|
|
FOR EACH ROW EXECUTE PROCEDURE _CDB_update_the_geom_webmercator();
|
|
SELECT CDB_CartodbfyTableCheck('t', 'trigger-protected the_geom');
|
|
SELECT 'extent',ST_Extent(ST_SnapToGrid(the_geom,0.2)) FROM t;
|
|
DROP TABLE t;
|
|
|
|
-- table with existing updated_at and created_at fields ot type text
|
|
CREATE TABLE t AS SELECT NOW()::text as created_at,
|
|
NOW()::text as updated_at,
|
|
NOW() as reftime;
|
|
SELECT CDB_CartodbfyTableCheck('t', 'text timestamps');
|
|
SELECT extract(secs from reftime-created_at),
|
|
extract(secs from reftime-updated_at) FROM t;
|
|
CREATE VIEW v AS SELECT * FROM t;
|
|
SELECT CDB_CartodbfyTableCheck('t', 'cartodbfied with view');
|
|
DROP VIEW v;
|
|
DROP TABLE t;
|
|
|
|
-- table with existing cartodb_id field of type text
|
|
CREATE TABLE t AS SELECT 10::text as cartodb_id;
|
|
SELECT CDB_CartodbfyTableCheck('t', 'text cartodb_id');
|
|
select cartodb_id/2 FROM t;
|
|
DROP TABLE t;
|
|
|
|
-- table with existing cartodb_id field of type text not casting
|
|
CREATE TABLE t AS SELECT 'nan' as cartodb_id;
|
|
SELECT CDB_CartodbfyTableCheck('t', 'uncasting text cartodb_id');
|
|
select cartodb_id,_cartodb_id0 FROM t;
|
|
DROP TABLE t;
|
|
|
|
-- table with existing cartodb_id field of type int4 not sequenced
|
|
CREATE TABLE t AS SELECT 1::int4 as cartodb_id;
|
|
SELECT CDB_CartodbfyTableCheck('t', 'unsequenced cartodb_id');
|
|
select cartodb_id FROM t;
|
|
DROP TABLE t;
|
|
|
|
-- table with existing cartodb_id serial primary key
|
|
CREATE TABLE t ( cartodb_id serial primary key );
|
|
SELECT CDB_CartodbfyTableCheck('t', 'cartodb_id serial primary key');
|
|
SELECT c.conname, a.attname FROM pg_constraint c, pg_attribute a
|
|
WHERE c.conrelid = 't'::regclass and a.attrelid = c.conrelid
|
|
AND c.conkey[1] = a.attnum AND NOT a.attisdropped;
|
|
DROP TABLE t;
|
|
|
|
-- table with existing the_geom and created_at and containing null values
|
|
-- Really, a test for surviving an longstanding PostgreSQL bug:
|
|
-- http://www.postgresql.org/message-id/20140530143150.GA11051@localhost
|
|
CREATE TABLE t (
|
|
the_geom geometry(Geometry,4326),
|
|
created_at timestamptz,
|
|
updated_at timestamptz
|
|
);
|
|
COPY t (the_geom, created_at, updated_at) FROM stdin;
|
|
0106000020E610000001000000010300000001000000050000009EB8244146435BC017B65E062AD343409EB8244146435BC0F51AF6E2708044400B99891683765AC0F51AF6E2708044400B99891683765AC017B65E062AD343409EB8244146435BC017B65E062AD34340 2012-06-06 21:59:08 2013-06-10 20:17:20
|
|
0106000020E61000000100000001030000000100000005000000DA7763431A1A5CC0FBCEE869313C3A40DA7763431A1A5CC09C1B8F55BC494440F9F4A9C7993356C09C1B8F55BC494440F9F4A9C7993356C0FBCEE869313C3A40DA7763431A1A5CC0FBCEE869313C3A40 2012-06-06 21:59:08 2013-06-10 20:17:20
|
|
\N \N \N
|
|
\.
|
|
SELECT CDB_CartodbfyTableCheck('t', 'null geom and timestamp values');
|
|
DROP TABLE t;
|
|
|
|
-- TODO: table with existing custom-triggered the_geom
|
|
|
|
DROP FUNCTION CDB_CartodbfyTableCheck(regclass, text);
|
|
DROP FUNCTION _CDB_UserQuotaInBytes();
|