sql/partition-functions.src.sql

   1 create or replace function getNearFeatures(in_partition INTEGER, point GEOMETRY, maxrank INTEGER, isin_tokens INT[]) RETURNS setof nearfeaturecentr AS $$
   2 DECLARE
   3   r nearfeaturecentr%rowtype;
   4 BEGIN
   5
   6 -- start
   7   IF in_partition = -partition- THEN
   8     FOR r IN
   9       SELECT place_id, keywords, rank_address, rank_search, min(ST_Distance(point, centroid)) as distance, isguess, centroid FROM (
  10         SELECT * FROM location_area_large_-partition- WHERE ST_Contains(geometry, point) and rank_search < maxrank
  11         UNION ALL
  12         SELECT * FROM location_area_country WHERE ST_Contains(geometry, point) and rank_search < maxrank
  13       ) as location_area
  14       GROUP BY place_id, keywords, rank_address, rank_search, isguess, centroid
  15       ORDER BY rank_address, isin_tokens && keywords desc, isguess asc,
  16         ST_Distance(point, centroid) *
  17           CASE
  18                WHEN rank_address = 16 AND rank_search = 15 THEN 0.2 -- capital city
  19                WHEN rank_address = 16 AND rank_search = 16 THEN 0.25 -- city
  20                WHEN rank_address = 16 AND rank_search = 17 THEN 0.5 -- town
  21                ELSE 1 END ASC -- everything else
  22     LOOP
  23       RETURN NEXT r;
  24     END LOOP;
  25     RETURN;
  26   END IF;
  27 -- end
  28
  29   RAISE EXCEPTION 'Unknown partition %', in_partition;
  30 END
  31 $$
  32 LANGUAGE plpgsql;
  33
  34 create or replace function deleteLocationArea(in_partition INTEGER, in_place_id BIGINT, in_rank_search INTEGER) RETURNS BOOLEAN AS $$
  35 DECLARE
  36 BEGIN
  37
  38   IF in_rank_search <= 4 THEN
  39     DELETE from location_area_country WHERE place_id = in_place_id;
  40     RETURN TRUE;
  41   END IF;
  42
  43 -- start
  44   IF in_partition = -partition- THEN
  45     DELETE from location_area_large_-partition- WHERE place_id = in_place_id;
  46     RETURN TRUE;
  47   END IF;
  48 -- end
  49
  50   RAISE EXCEPTION 'Unknown partition %', in_partition;
  51
  52   RETURN FALSE;
  53 END
  54 $$
  55 LANGUAGE plpgsql;
  56
  57 create or replace function insertLocationAreaLarge(
  58   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), in_keywords INTEGER[],
  59   in_rank_search INTEGER, in_rank_address INTEGER, in_estimate BOOLEAN,
  60   in_centroid GEOMETRY, in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
  61 DECLARE
  62 BEGIN
  63
  64   IF in_rank_search <= 4 THEN
  65     INSERT INTO location_area_country values (in_partition, in_place_id, in_country_code, in_keywords, in_rank_search, in_rank_address, in_estimate, in_centroid, in_geometry);
  66     RETURN TRUE;
  67   END IF;
  68
  69 -- start
  70   IF in_partition = -partition- THEN
  71     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);
  72     RETURN TRUE;
  73   END IF;
  74 -- end
  75
  76   RAISE EXCEPTION 'Unknown partition %', in_partition;
  77   RETURN FALSE;
  78 END
  79 $$
  80 LANGUAGE plpgsql;
  81
  82 create or replace function getNearestNamedFeature(in_partition INTEGER, point GEOMETRY, maxrank INTEGER, isin_token INTEGER) RETURNS setof nearfeature AS $$
  83 DECLARE
  84   r nearfeature%rowtype;
  85 BEGIN
  86
  87 -- start
  88   IF in_partition = -partition- THEN
  89     FOR r IN
  90       SELECT place_id, name_vector, address_rank, search_rank,
  91           ST_Distance(centroid, point) as distance, null as isguess
  92           FROM search_name_-partition-
  93           WHERE name_vector @> ARRAY[isin_token]
  94           AND search_rank < maxrank
  95       UNION ALL
  96       SELECT place_id, name_vector, address_rank, search_rank,
  97           ST_Distance(centroid, point) as distance, null as isguess
  98           FROM search_name_country
  99           WHERE name_vector @> ARRAY[isin_token]
 100           AND search_rank < maxrank
 101       ORDER BY distance ASC limit 1
 102     LOOP
 103       RETURN NEXT r;
 104     END LOOP;
 105     RETURN;
 106   END IF;
 107 -- end
 108
 109   RAISE EXCEPTION 'Unknown partition %', in_partition;
 110 END
 111 $$
 112 LANGUAGE plpgsql;
 113
 114 create or replace function getNearestNamedRoadFeature(in_partition INTEGER, point GEOMETRY, isin_token INTEGER)
 115   RETURNS setof nearfeature AS $$
 116 DECLARE
 117   r nearfeature%rowtype;
 118 BEGIN
 119
 120 -- start
 121   IF in_partition = -partition- THEN
 122     FOR r IN
 123       SELECT place_id, name_vector, address_rank, search_rank,
 124           ST_Distance(centroid, point) as distance, null as isguess
 125           FROM search_name_-partition-
 126           WHERE name_vector @> ARRAY[isin_token]
 127           AND ST_DWithin(centroid, point, 0.03)
 128           AND search_rank between 22 and 27
 129       ORDER BY distance ASC limit 1
 130     LOOP
 131       RETURN NEXT r;
 132     END LOOP;
 133     RETURN;
 134   END IF;
 135 -- end
 136
 137   RAISE EXCEPTION 'Unknown partition %', in_partition;
 138 END
 139 $$
 140 LANGUAGE plpgsql;
 141
 142 create or replace function getNearestPostcode(in_partition INTEGER, point GEOMETRY)
 143   RETURNS TEXT AS $$
 144 DECLARE
 145   out_postcode TEXT;
 146 BEGIN
 147
 148 -- start
 149   IF in_partition = -partition- THEN
 150     SELECT postcode
 151         FROM location_area_large_-partition- join placex using (place_id)
 152         WHERE st_contains(location_area_large_-partition-.geometry, point)
 153         AND class = 'place' and type = 'postcode'
 154       ORDER BY st_distance(location_area_large_-partition-.centroid, point) ASC limit 1
 155       INTO out_postcode;
 156     RETURN out_postcode;
 157   END IF;
 158 -- end
 159
 160   RAISE EXCEPTION 'Unknown partition %', in_partition;
 161 END
 162 $$
 163 LANGUAGE plpgsql;
 164
 165 create or replace function insertSearchName(
 166   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2),
 167   in_name_vector INTEGER[], in_nameaddress_vector INTEGER[],
 168   in_rank_search INTEGER, in_rank_address INTEGER, in_importance FLOAT,
 169   in_centroid GEOMETRY) RETURNS BOOLEAN AS $$
 170 DECLARE
 171 BEGIN
 172
 173   DELETE FROM search_name WHERE place_id = in_place_id;
 174   INSERT INTO search_name values (in_place_id, in_rank_search, in_rank_address, in_importance, in_country_code,
 175     in_name_vector, in_nameaddress_vector, in_centroid);
 176
 177   IF in_rank_search <= 4 THEN
 178     DELETE FROM search_name_country WHERE place_id = in_place_id;
 179     INSERT INTO search_name_country values (in_place_id, in_rank_search, in_rank_address, in_importance, in_country_code,
 180       in_name_vector, in_nameaddress_vector, in_centroid);
 181     RETURN TRUE;
 182   END IF;
 183
 184 -- start
 185   IF in_partition = -partition- THEN
 186     DELETE FROM search_name_-partition- values WHERE place_id = in_place_id;
 187     INSERT INTO search_name_-partition- values (in_place_id, in_rank_search, in_rank_address, 0, in_country_code,
 188       in_name_vector, in_nameaddress_vector, in_centroid);
 189     RETURN TRUE;
 190   END IF;
 191 -- end
 192
 193   RAISE EXCEPTION 'Unknown partition %', in_partition;
 194   RETURN FALSE;
 195 END
 196 $$
 197 LANGUAGE plpgsql;
 198
 199 create or replace function deleteSearchName(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
 200 DECLARE
 201 BEGIN
 202
 203   DELETE from search_name WHERE place_id = in_place_id;
 204   DELETE from search_name_country WHERE place_id = in_place_id;
 205
 206 -- start
 207   IF in_partition = -partition- THEN
 208     DELETE from search_name_-partition- WHERE place_id = in_place_id;
 209     RETURN TRUE;
 210   END IF;
 211 -- end
 212
 213   RAISE EXCEPTION 'Unknown partition %', in_partition;
 214
 215   RETURN FALSE;
 216 END
 217 $$
 218 LANGUAGE plpgsql;
 219
 220 create or replace function insertLocationRoad(
 221   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
 222 DECLARE
 223 BEGIN
 224
 225 -- start
 226   IF in_partition = -partition- THEN
 227     DELETE FROM location_road_-partition- where place_id = in_place_id;
 228     INSERT INTO location_road_-partition- values (in_partition, in_place_id, in_country_code, in_geometry);
 229     RETURN TRUE;
 230   END IF;
 231 -- end
 232
 233   RAISE EXCEPTION 'Unknown partition %', in_partition;
 234   RETURN FALSE;
 235 END
 236 $$
 237 LANGUAGE plpgsql;
 238
 239 create or replace function deleteRoad(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
 240 DECLARE
 241 BEGIN
 242
 243 -- start
 244   IF in_partition = -partition- THEN
 245     DELETE FROM location_road_-partition- where place_id = in_place_id;
 246     RETURN TRUE;
 247   END IF;
 248 -- end
 249
 250   RAISE EXCEPTION 'Unknown partition %', in_partition;
 251
 252   RETURN FALSE;
 253 END
 254 $$
 255 LANGUAGE plpgsql;
 256
 257 create or replace function getNearestRoadFeature(in_partition INTEGER, point GEOMETRY) RETURNS setof nearfeature AS $$
 258 DECLARE
 259   r nearfeature%rowtype;
 260   search_diameter FLOAT;
 261 BEGIN
 262
 263 -- start
 264   IF in_partition = -partition- THEN
 265     search_diameter := 0.00005;
 266     WHILE search_diameter < 0.1 LOOP
 267       FOR r IN
 268         SELECT place_id, null, null, null,
 269             ST_Distance(geometry, point) as distance, null as isguess
 270             FROM location_road_-partition-
 271             WHERE ST_DWithin(geometry, point, search_diameter)
 272         ORDER BY distance ASC limit 1
 273       LOOP
 274         RETURN NEXT r;
 275         RETURN;
 276       END LOOP;
 277       search_diameter := search_diameter * 2;
 278     END LOOP;
 279     RETURN;
 280   END IF;
 281 -- end
 282
 283   RAISE EXCEPTION 'Unknown partition %', in_partition;
 284 END
 285 $$
 286 LANGUAGE plpgsql;
 287
 288 create or replace function getNearestParellelRoadFeature(in_partition INTEGER, line GEOMETRY) RETURNS setof nearfeature AS $$
 289 DECLARE
 290   r nearfeature%rowtype;
 291   search_diameter FLOAT;
 292   p1 GEOMETRY;
 293   p2 GEOMETRY;
 294   p3 GEOMETRY;
 295 BEGIN
 296
 297   IF st_geometrytype(line) not in ('ST_LineString') THEN
 298     RETURN;
 299   END IF;
 300
 301   p1 := ST_Line_Interpolate_Point(line,0);
 302   p2 := ST_Line_Interpolate_Point(line,0.5);
 303   p3 := ST_Line_Interpolate_Point(line,1);
 304
 305 -- start
 306   IF in_partition = -partition- THEN
 307     search_diameter := 0.0005;
 308     WHILE search_diameter < 0.01 LOOP
 309       FOR r IN
 310         SELECT place_id, null, null, null,
 311             ST_Distance(geometry, line) as distance, null as isguess
 312             FROM location_road_-partition-
 313             WHERE ST_DWithin(line, geometry, search_diameter)
 314             ORDER BY (ST_distance(geometry, p1)+
 315                       ST_distance(geometry, p2)+
 316                       ST_distance(geometry, p3)) ASC limit 1
 317       LOOP
 318         RETURN NEXT r;
 319         RETURN;
 320       END LOOP;
 321       search_diameter := search_diameter * 2;
 322     END LOOP;
 323     RETURN;
 324   END IF;
 325 -- end
 326
 327   RAISE EXCEPTION 'Unknown partition %', in_partition;
 328 END
 329 $$
 330 LANGUAGE plpgsql;