cartodb-4.42/services/table-geocoder/lib/internal-geocoder/admin1_column_polygons.rb
2024-04-06 05:25:13 +00:00

41 lines
1.6 KiB
Ruby

require_relative 'abstract_query_generator'
module CartoDB
module InternalGeocoder
class Admin1ColumnPolygons < AbstractQueryGenerator
def search_terms_query(page)
%Q{
SELECT DISTINCT
trim(quote_nullable(#{@internal_geocoder.column_name})) as region,
trim(quote_nullable(#{@internal_geocoder.country_column})) as country
FROM #{@internal_geocoder.qualified_table_name}
WHERE cartodb_georef_status IS NULL
LIMIT #{@internal_geocoder.batch_size} OFFSET #{page * @internal_geocoder.batch_size}
}
end
def dataservices_query(search_terms)
region = search_terms.map { |row| row[:region] }.join(',')
countries = search_terms.map { |row| row[:country] }.join(',')
"WITH geo_function AS (SELECT (geocode_admin1_polygons(Array[#{region}], Array[#{countries}])).*) SELECT q, c, null AS a1, geom, success FROM geo_function"
end
def copy_results_to_table_query
%Q{
UPDATE #{dest_table} AS dest
SET the_geom = CASE WHEN orig.cartodb_georef_status THEN orig.the_geom ELSE dest.the_geom END,
cartodb_georef_status = orig.cartodb_georef_status
FROM #{@internal_geocoder.temp_table_name} AS orig
WHERE trim(dest.#{@internal_geocoder.column_name}::text) = trim(orig.geocode_string)
AND trim(dest.#{@internal_geocoder.country_column}::text) = trim(orig.country)
AND dest.cartodb_georef_status IS NULL
}
end
end # Admin1ColumnPolygons
end # InternalGeocoder
end # CartoDB