X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/31b205ecf96ddf2573479f875d6030ffc82ef92b..bfe56e6a4575b9697dcfb496649889a57c681931:/sql/partitions.src.sql diff --git a/sql/partitions.src.sql b/sql/partitions.src.sql index b6acd0f2..4fe06ea2 100644 --- a/sql/partitions.src.sql +++ b/sql/partitions.src.sql @@ -14,28 +14,41 @@ CREATE TABLE location_area_country () INHERITS (location_area_large); CREATE INDEX idx_location_area_country_place_id ON location_area_country USING BTREE (place_id); CREATE INDEX idx_location_area_country_geometry ON location_area_country USING GIST (geometry); +CREATE TABLE search_name_country () INHERITS (search_name_blank); +CREATE INDEX idx_search_name_country_place_id ON search_name_country USING BTREE (place_id); +CREATE INDEX idx_search_name_country_centroid ON search_name_country USING GIST (centroid); +CREATE INDEX idx_search_name_country_name_vector ON search_name_country USING GIN (name_vector gin__int_ops); +CREATE INDEX idx_search_name_country_nameaddress_vector ON search_name_country USING GIN (nameaddress_vector gin__int_ops); + -- start CREATE TABLE location_area_large_-partition- () INHERITS (location_area_large); CREATE INDEX idx_location_area_large_-partition-_place_id ON location_area_large_-partition- USING BTREE (place_id); CREATE INDEX idx_location_area_large_-partition-_geometry ON location_area_large_-partition- USING GIST (geometry); CREATE TABLE location_area_roadnear_-partition- () INHERITS (location_area_roadnear); -CREATE INDEX idx_location_area_roadnear_-partition-_place_id ON location_area_roadfar_-partition- USING BTREE (place_id); +CREATE INDEX idx_location_area_roadnear_-partition-_place_id ON location_area_roadnear_-partition- USING BTREE (place_id); CREATE INDEX idx_location_area_roadnear_-partition-_geometry ON location_area_roadnear_-partition- USING GIST (geometry); CREATE TABLE location_area_roadfar_-partition- () INHERITS (location_area_roadfar); CREATE INDEX idx_location_area_roadfar_-partition-_place_id ON location_area_roadfar_-partition- USING BTREE (place_id); CREATE INDEX idx_location_area_roadfar_-partition-_geometry ON location_area_roadfar_-partition- USING GIST (geometry); + +CREATE TABLE search_name_-partition- () INHERITS (search_name_blank); +CREATE INDEX idx_search_name_-partition-_place_id ON search_name_-partition- USING BTREE (place_id); +CREATE INDEX idx_search_name_-partition-_centroid ON search_name_-partition- USING GIST (centroid); +CREATE INDEX idx_search_name_-partition-_name_vector ON search_name_-partition- USING GIN (name_vector gin__int_ops); +CREATE INDEX idx_search_name_-partition-_nameaddress_vector ON search_name_-partition- USING GIN (nameaddress_vector gin__int_ops); + -- end -create or replace function getNearRoads(in_partition TEXT, point GEOMETRY) RETURNS setof nearplace AS $$ +create or replace function getNearRoads(in_partition INTEGER, point GEOMETRY) RETURNS setof nearplace AS $$ DECLARE r nearplace%rowtype; a BOOLEAN; BEGIN -- start - IF in_partition = '-partition-' THEN + IF in_partition = -partition- THEN a := FALSE; FOR r IN SELECT place_id FROM location_area_roadnear_-partition- WHERE ST_Contains(geometry, point) ORDER BY ST_Distance(point, centroid) ASC LIMIT 1 LOOP a := TRUE; @@ -55,13 +68,13 @@ END $$ LANGUAGE plpgsql; -create or replace function getNearFeatures(in_partition TEXT, point GEOMETRY, maxrank INTEGER, isin_tokens INT[]) RETURNS setof nearfeature AS $$ +create or replace function getNearFeatures(in_partition INTEGER, point GEOMETRY, maxrank INTEGER, isin_tokens INT[]) RETURNS setof nearfeature AS $$ DECLARE r nearfeature%rowtype; BEGIN -- start - IF in_partition = '-partition-' THEN + IF in_partition = -partition- THEN FOR r IN SELECT place_id, keywords, rank_address, rank_search, ST_Distance(point, centroid) as distance FROM ( SELECT * FROM location_area_large_-partition- WHERE ST_Contains(geometry, point) and rank_search < maxrank @@ -81,12 +94,12 @@ END $$ LANGUAGE plpgsql; -create or replace function deleteLocationArea(in_partition TEXT, in_place_id integer) RETURNS BOOLEAN AS $$ +create or replace function deleteLocationArea(in_partition INTEGER, in_place_id integer) RETURNS BOOLEAN AS $$ DECLARE BEGIN -- start - IF in_partition = '-partition-' THEN + IF in_partition = -partition- THEN -- DELETE from location_area_large_-partition- WHERE place_id = in_place_id; -- DELETE from location_area_roadnear_-partition- WHERE place_id = in_place_id; -- DELETE from location_area_roadfar_-partition- WHERE place_id = in_place_id; @@ -102,7 +115,7 @@ $$ LANGUAGE plpgsql; create or replace function insertLocationAreaLarge( - in_partition TEXT, in_place_id integer, in_country_code VARCHAR(2), in_keywords INTEGER[], + in_partition INTEGER, in_place_id integer, in_country_code VARCHAR(2), in_keywords INTEGER[], in_rank_search INTEGER, in_rank_address INTEGER, in_estimate BOOLEAN, in_centroid GEOMETRY, in_geometry GEOMETRY) RETURNS BOOLEAN AS $$ DECLARE @@ -114,7 +127,7 @@ BEGIN END IF; -- start - IF in_partition = '-partition-' THEN + IF in_partition = -partition- THEN INSERT INTO location_area_large_-partition- values (in_partition, in_place_id, in_country_code, in_keywords, in_rank_search, in_rank_address, in_estimate, in_centroid, in_geometry); RETURN TRUE; END IF; @@ -127,14 +140,14 @@ $$ LANGUAGE plpgsql; create or replace function insertLocationAreaRoadNear( - in_partition TEXT, in_place_id integer, in_country_code VARCHAR(2), in_keywords INTEGER[], + in_partition INTEGER, in_place_id integer, in_country_code VARCHAR(2), in_keywords INTEGER[], in_rank_search INTEGER, in_rank_address INTEGER, in_estimate BOOLEAN, in_centroid GEOMETRY, in_geometry GEOMETRY) RETURNS BOOLEAN AS $$ DECLARE BEGIN -- start - IF in_partition = '-partition-' THEN + IF in_partition = -partition- THEN INSERT INTO location_area_roadnear_-partition- values (in_partition, in_place_id, in_country_code, in_keywords, in_rank_search, in_rank_address, in_estimate, in_centroid, in_geometry); RETURN TRUE; END IF; @@ -147,14 +160,14 @@ $$ LANGUAGE plpgsql; create or replace function insertLocationAreaRoadFar( - in_partition TEXT, in_place_id integer, in_country_code VARCHAR(2), in_keywords INTEGER[], + in_partition INTEGER, in_place_id integer, in_country_code VARCHAR(2), in_keywords INTEGER[], in_rank_search INTEGER, in_rank_address INTEGER, in_estimate BOOLEAN, in_centroid GEOMETRY, in_geometry GEOMETRY) RETURNS BOOLEAN AS $$ DECLARE BEGIN -- start - IF in_partition = '-partition-' THEN + IF in_partition = -partition- THEN INSERT INTO location_area_roadfar_-partition- values (in_partition, in_place_id, in_country_code, in_keywords, in_rank_search, in_rank_address, in_estimate, in_centroid, in_geometry); RETURN TRUE; END IF; @@ -165,3 +178,114 @@ BEGIN END $$ LANGUAGE plpgsql; + +create or replace function getNearestNamedFeature(in_partition INTEGER, point GEOMETRY, maxrank INTEGER, isin_token INTEGER) RETURNS setof nearfeature AS $$ +DECLARE + r nearfeature%rowtype; +BEGIN + +-- start + IF in_partition = -partition- THEN + FOR r IN + SELECT place_id, name_vector, address_rank, search_rank, + ST_Distance(centroid, point) as distance + FROM search_name_-partition- + WHERE name_vector @> ARRAY[isin_token] + AND search_rank < maxrank + UNION ALL + SELECT place_id, name_vector, address_rank, search_rank, + ST_Distance(centroid, point) as distance + FROM search_name_country + WHERE name_vector @> ARRAY[isin_token] + AND search_rank < maxrank + ORDER BY distance ASC limit 1 + LOOP + RETURN NEXT r; + END LOOP; + RETURN; + END IF; +-- end + + RAISE EXCEPTION 'Unknown partition %', in_partition; +END +$$ +LANGUAGE plpgsql; + +create or replace function getNearestNamedRoadFeature(in_partition INTEGER, point GEOMETRY, isin_token INTEGER) RETURNS setof nearfeature AS $$ +DECLARE + r nearfeature%rowtype; +BEGIN + +-- start + IF in_partition = -partition- THEN + FOR r IN + SELECT place_id, name_vector, address_rank, search_rank, + ST_Distance(centroid, point) as distance + FROM search_name_-partition- + WHERE name_vector @> ARRAY[isin_token] + AND ST_DWithin(centroid, point, 0.01) + AND search_rank between 22 and 27 + ORDER BY distance ASC limit 1 + LOOP + RETURN NEXT r; + END LOOP; + RETURN; + END IF; +-- end + + RAISE EXCEPTION 'Unknown partition %', in_partition; +END +$$ +LANGUAGE plpgsql; + +create or replace function insertSearchName( + in_partition INTEGER, in_place_id integer, in_country_code VARCHAR(2), + in_name_vector INTEGER[], in_nameaddress_vector INTEGER[], + in_rank_search INTEGER, in_rank_address INTEGER, + in_centroid GEOMETRY) RETURNS BOOLEAN AS $$ +DECLARE +BEGIN + + INSERT INTO search_name values (in_place_id, in_rank_search, in_rank_address, 0, in_country_code, + in_name_vector, in_nameaddress_vector, in_centroid); + + IF in_rank_search <= 4 THEN + INSERT INTO search_name_country values (in_place_id, in_rank_search, in_rank_address, 0, in_country_code, + in_name_vector, in_nameaddress_vector, in_centroid); + RETURN TRUE; + END IF; + +-- start + IF in_partition = -partition- THEN + INSERT INTO search_name_-partition- values (in_place_id, in_rank_search, in_rank_address, 0, in_country_code, + in_name_vector, in_nameaddress_vector, in_centroid); + RETURN TRUE; + END IF; +-- end + + RAISE EXCEPTION 'Unknown partition %', in_partition; + RETURN FALSE; +END +$$ +LANGUAGE plpgsql; + +create or replace function deleteSearchName(in_partition INTEGER, in_place_id integer) RETURNS BOOLEAN AS $$ +DECLARE +BEGIN + + DELETE from search_name WHERE place_id = in_place_id; + DELETE from search_name_country WHERE place_id = in_place_id; + +-- start + IF in_partition = -partition- THEN + DELETE from search_name_-partition- WHERE place_id = in_place_id; + RETURN TRUE; + END IF; +-- end + + RAISE EXCEPTION 'Unknown partition %', in_partition; + + RETURN FALSE; +END +$$ +LANGUAGE plpgsql;