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 getNearestNamedRoadFeature(in_partition INTEGER, point GEOMETRY, isin_token INTEGER[])
  88   RETURNS setof nearfeature AS $$
  89 DECLARE
  90   r nearfeature%rowtype;
  91 BEGIN
  92
  93 -- start
  94   IF in_partition = -partition- THEN
  95     FOR r IN
  96       SELECT place_id, name_vector, address_rank, search_rank,
  97           ST_Distance(centroid, point) as distance, null as isguess
  98           FROM search_name_-partition-
  99           WHERE name_vector && isin_token
 100           AND ST_DWithin(centroid, point, 0.015)
 101           AND search_rank between 26 and 27
 102       ORDER BY distance ASC limit 1
 103     LOOP
 104       RETURN NEXT r;
 105     END LOOP;
 106     RETURN;
 107   END IF;
 108 -- end
 109
 110   RAISE EXCEPTION 'Unknown partition %', in_partition;
 111 END
 112 $$
 113 LANGUAGE plpgsql;
 114
 115 create or replace function getNearestNamedPlaceFeature(in_partition INTEGER, point GEOMETRY, isin_token INTEGER[])
 116   RETURNS setof nearfeature AS $$
 117 DECLARE
 118   r nearfeature%rowtype;
 119 BEGIN
 120
 121 -- start
 122   IF in_partition = -partition- THEN
 123     FOR r IN
 124       SELECT place_id, name_vector, address_rank, search_rank,
 125           ST_Distance(centroid, point) as distance, null as isguess
 126           FROM search_name_-partition-
 127           WHERE name_vector && isin_token
 128           AND ST_DWithin(centroid, point, 0.04)
 129           AND search_rank between 16 and 22
 130       ORDER BY distance ASC limit 1
 131     LOOP
 132       RETURN NEXT r;
 133     END LOOP;
 134     RETURN;
 135   END IF;
 136 -- end
 137
 138   RAISE EXCEPTION 'Unknown partition %', in_partition;
 139 END
 140 $$
 141 LANGUAGE plpgsql;
 142
 143
 144 create or replace function insertSearchName(
 145   in_partition INTEGER, in_place_id BIGINT, in_name_vector INTEGER[],
 146   in_rank_search INTEGER, in_rank_address INTEGER, in_geometry GEOMETRY)
 147 RETURNS BOOLEAN AS $$
 148 DECLARE
 149 BEGIN
 150 -- start
 151   IF in_partition = -partition- THEN
 152     DELETE FROM search_name_-partition- values WHERE place_id = in_place_id;
 153     IF in_rank_address > 0 THEN
 154       INSERT INTO search_name_-partition- (place_id, search_rank, address_rank, name_vector, centroid)
 155         values (in_place_id, in_rank_search, in_rank_address, in_name_vector, in_geometry);
 156     END IF;
 157     RETURN TRUE;
 158   END IF;
 159 -- end
 160
 161   RAISE EXCEPTION 'Unknown partition %', in_partition;
 162   RETURN FALSE;
 163 END
 164 $$
 165 LANGUAGE plpgsql;
 166
 167 create or replace function deleteSearchName(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
 168 DECLARE
 169 BEGIN
 170 -- start
 171   IF in_partition = -partition- THEN
 172     DELETE from search_name_-partition- WHERE place_id = in_place_id;
 173     RETURN TRUE;
 174   END IF;
 175 -- end
 176
 177   RAISE EXCEPTION 'Unknown partition %', in_partition;
 178
 179   RETURN FALSE;
 180 END
 181 $$
 182 LANGUAGE plpgsql;
 183
 184 create or replace function insertLocationRoad(
 185   in_partition INTEGER, in_place_id BIGINT, in_country_code VARCHAR(2), in_geometry GEOMETRY) RETURNS BOOLEAN AS $$
 186 DECLARE
 187 BEGIN
 188
 189 -- start
 190   IF in_partition = -partition- THEN
 191     DELETE FROM location_road_-partition- where place_id = in_place_id;
 192     INSERT INTO location_road_-partition- (partition, place_id, country_code, geometry)
 193       values (in_partition, in_place_id, in_country_code, in_geometry);
 194     RETURN TRUE;
 195   END IF;
 196 -- end
 197
 198   RAISE EXCEPTION 'Unknown partition %', in_partition;
 199   RETURN FALSE;
 200 END
 201 $$
 202 LANGUAGE plpgsql;
 203
 204 create or replace function deleteRoad(in_partition INTEGER, in_place_id BIGINT) RETURNS BOOLEAN AS $$
 205 DECLARE
 206 BEGIN
 207
 208 -- start
 209   IF in_partition = -partition- THEN
 210     DELETE FROM location_road_-partition- where place_id = in_place_id;
 211     RETURN TRUE;
 212   END IF;
 213 -- end
 214
 215   RAISE EXCEPTION 'Unknown partition %', in_partition;
 216
 217   RETURN FALSE;
 218 END
 219 $$
 220 LANGUAGE plpgsql;
 221
 222 create or replace function getNearestRoadFeature(in_partition INTEGER, point GEOMETRY) RETURNS setof nearfeature AS $$
 223 DECLARE
 224   r nearfeature%rowtype;
 225   search_diameter FLOAT;
 226 BEGIN
 227
 228 -- start
 229   IF in_partition = -partition- THEN
 230     search_diameter := 0.00005;
 231     WHILE search_diameter < 0.1 LOOP
 232       FOR r IN
 233         SELECT place_id, null, null, null,
 234             ST_Distance(geometry, point) as distance, null as isguess
 235             FROM location_road_-partition-
 236             WHERE ST_DWithin(geometry, point, search_diameter)
 237         ORDER BY distance ASC limit 1
 238       LOOP
 239         RETURN NEXT r;
 240         RETURN;
 241       END LOOP;
 242       search_diameter := search_diameter * 2;
 243     END LOOP;
 244     RETURN;
 245   END IF;
 246 -- end
 247
 248   RAISE EXCEPTION 'Unknown partition %', in_partition;
 249 END
 250 $$
 251 LANGUAGE plpgsql;
 252
 253 create or replace function getNearestParellelRoadFeature(in_partition INTEGER, line GEOMETRY) RETURNS setof nearfeature AS $$
 254 DECLARE
 255   r nearfeature%rowtype;
 256   search_diameter FLOAT;
 257   p1 GEOMETRY;
 258   p2 GEOMETRY;
 259   p3 GEOMETRY;
 260 BEGIN
 261
 262   IF st_geometrytype(line) not in ('ST_LineString') THEN
 263     RETURN;
 264   END IF;
 265
 266   p1 := ST_LineInterpolatePoint(line,0);
 267   p2 := ST_LineInterpolatePoint(line,0.5);
 268   p3 := ST_LineInterpolatePoint(line,1);
 269
 270 -- start
 271   IF in_partition = -partition- THEN
 272     search_diameter := 0.0005;
 273     WHILE search_diameter < 0.01 LOOP
 274       FOR r IN
 275         SELECT place_id, null, null, null,
 276             ST_Distance(geometry, line) as distance, null as isguess
 277             FROM location_road_-partition-
 278             WHERE ST_DWithin(line, geometry, search_diameter)
 279             ORDER BY (ST_distance(geometry, p1)+
 280                       ST_distance(geometry, p2)+
 281                       ST_distance(geometry, p3)) ASC limit 1
 282       LOOP
 283         RETURN NEXT r;
 284         RETURN;
 285       END LOOP;
 286       search_diameter := search_diameter * 2;
 287     END LOOP;
 288     RETURN;
 289   END IF;
 290 -- end
 291
 292   RAISE EXCEPTION 'Unknown partition %', in_partition;
 293 END
 294 $$
 295 LANGUAGE plpgsql;