sql/partition-functions.src.sql

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