6457419e0d
I would like to reformat all of our SQL to have a concise coding style. This makes it far easier to understand the code for a casual contributor, and lets us spot errors more easily. Most importantly, it makes it much easier to grep (search) the code because it is more likely to be in the same syntax Some key changes: * SQL keywords are always UPPERCASE, e.g. `SELECT WHEN AS END ...` * types, variables, aliases, and field names (identifiers) are always lower case * `LANGUAGE 'plpgsql'` is now `LANGUAGE plpgsql` (no quotes) * a few minor spacing/semicolon cleanups P.S. Per @TomPohys request, `TABLE` is spelled using upper case despite being a type for consistency with PG Docs. Same for `LANGUAGE SQL` vs `LANGUAGE plpgsql`.
98 lines
2.9 KiB
PL/PgSQL
98 lines
2.9 KiB
PL/PgSQL
DROP TRIGGER IF EXISTS trigger_flag ON osm_state_point;
|
|
DROP TRIGGER IF EXISTS trigger_refresh ON place_state.updates;
|
|
|
|
ALTER TABLE osm_state_point
|
|
DROP CONSTRAINT IF EXISTS osm_state_point_rank_constraint;
|
|
|
|
-- etldoc: ne_10m_admin_1_states_provinces -> osm_state_point
|
|
-- etldoc: osm_state_point -> osm_state_point
|
|
|
|
CREATE OR REPLACE FUNCTION update_osm_state_point() RETURNS void AS
|
|
$$
|
|
BEGIN
|
|
|
|
WITH important_state_point AS (
|
|
SELECT osm.geometry,
|
|
osm.osm_id,
|
|
osm.name,
|
|
COALESCE(NULLIF(osm.name_en, ''), ne.name) AS name_en,
|
|
ne.scalerank,
|
|
ne.labelrank,
|
|
ne.datarank
|
|
FROM ne_10m_admin_1_states_provinces AS ne,
|
|
osm_state_point AS osm
|
|
WHERE
|
|
-- We only match whether the point is within the Natural Earth polygon
|
|
-- because name matching is difficult
|
|
ST_Within(osm.geometry, ne.geometry)
|
|
-- We leave out leess important states
|
|
AND ne.scalerank <= 3
|
|
AND ne.labelrank <= 2
|
|
)
|
|
UPDATE osm_state_point AS osm
|
|
-- Normalize both scalerank and labelrank into a ranking system from 1 to 6.
|
|
SET "rank" = LEAST(6, CEILING((scalerank + labelrank + datarank) / 3.0))
|
|
FROM important_state_point AS ne
|
|
WHERE osm.osm_id = ne.osm_id;
|
|
|
|
-- TODO: This shouldn't be necessary? The rank function makes something wrong...
|
|
UPDATE osm_state_point AS osm
|
|
SET "rank" = 1
|
|
WHERE "rank" = 0;
|
|
|
|
DELETE FROM osm_state_point WHERE "rank" IS NULL;
|
|
|
|
UPDATE osm_state_point
|
|
SET tags = update_tags(tags, geometry)
|
|
WHERE COALESCE(tags->'name:latin', tags->'name:nonlatin', tags->'name_int') IS NULL;
|
|
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
SELECT update_osm_state_point();
|
|
|
|
-- ALTER TABLE osm_state_point ADD CONSTRAINT osm_state_point_rank_constraint CHECK("rank" BETWEEN 1 AND 6);
|
|
CREATE INDEX IF NOT EXISTS osm_state_point_rank_idx ON osm_state_point ("rank");
|
|
|
|
-- Handle updates
|
|
|
|
CREATE SCHEMA IF NOT EXISTS place_state;
|
|
|
|
CREATE TABLE IF NOT EXISTS place_state.updates
|
|
(
|
|
id serial PRIMARY KEY,
|
|
t text,
|
|
UNIQUE (t)
|
|
);
|
|
CREATE OR REPLACE FUNCTION place_state.flag() RETURNS trigger AS
|
|
$$
|
|
BEGIN
|
|
INSERT INTO place_state.updates(t) VALUES ('y') ON CONFLICT(t) DO NOTHING;
|
|
RETURN NULL;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
CREATE OR REPLACE FUNCTION place_state.refresh() RETURNS trigger AS
|
|
$$
|
|
BEGIN
|
|
RAISE LOG 'Refresh place_state rank';
|
|
PERFORM update_osm_state_point();
|
|
-- noinspection SqlWithoutWhere
|
|
DELETE FROM place_state.updates;
|
|
RETURN NULL;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
CREATE TRIGGER trigger_flag
|
|
AFTER INSERT OR UPDATE OR DELETE
|
|
ON osm_state_point
|
|
FOR EACH STATEMENT
|
|
EXECUTE PROCEDURE place_state.flag();
|
|
|
|
CREATE CONSTRAINT TRIGGER trigger_refresh
|
|
AFTER INSERT
|
|
ON place_state.updates
|
|
INITIALLY DEFERRED
|
|
FOR EACH ROW
|
|
EXECUTE PROCEDURE place_state.refresh();
|