mirror of
https://github.com/Freika/dawarich.git
synced 2026-01-10 01:01:39 -05:00
100 lines
2.3 KiB
Ruby
100 lines
2.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class Users::ImportData::Places
|
|
BATCH_SIZE = 5000
|
|
|
|
def initialize(user, places_data = nil, batch_size: BATCH_SIZE, logger: Rails.logger)
|
|
@user = user
|
|
@places_data = places_data
|
|
@batch_size = batch_size
|
|
@logger = logger
|
|
@buffer = []
|
|
@created = 0
|
|
end
|
|
|
|
def call
|
|
return 0 unless places_data.respond_to?(:each)
|
|
|
|
enumerate(places_data) do |place_data|
|
|
add(place_data)
|
|
end
|
|
|
|
finalize
|
|
end
|
|
|
|
def add(place_data)
|
|
return unless place_data.is_a?(Hash)
|
|
|
|
@buffer << place_data
|
|
flush_batch if @buffer.size >= batch_size
|
|
end
|
|
|
|
def finalize
|
|
flush_batch
|
|
logger.info "Places import completed. Created: #{@created}"
|
|
@created
|
|
end
|
|
|
|
private
|
|
|
|
attr_reader :user, :places_data, :batch_size, :logger
|
|
|
|
def enumerate(collection, &block)
|
|
collection.each(&block)
|
|
end
|
|
|
|
def collection_description(collection)
|
|
return collection.size if collection.respond_to?(:size)
|
|
|
|
'streamed'
|
|
end
|
|
|
|
def flush_batch
|
|
return if @buffer.empty?
|
|
|
|
logger.debug "Processing places batch of #{@buffer.size}"
|
|
@buffer.each do |place_data|
|
|
place = find_or_create_place_for_import(place_data)
|
|
@created += 1 if place&.respond_to?(:previously_new_record?) && place.previously_new_record?
|
|
end
|
|
|
|
@buffer.clear
|
|
end
|
|
|
|
def find_or_create_place_for_import(place_data)
|
|
name = place_data['name']
|
|
latitude = place_data['latitude']&.to_f
|
|
longitude = place_data['longitude']&.to_f
|
|
|
|
unless name.present? && latitude.present? && longitude.present?
|
|
return nil
|
|
end
|
|
|
|
existing_place = Place.where(
|
|
name: name,
|
|
latitude: latitude,
|
|
longitude: longitude,
|
|
user_id: nil
|
|
).first
|
|
|
|
if existing_place
|
|
existing_place.define_singleton_method(:previously_new_record?) { false }
|
|
return existing_place
|
|
end
|
|
|
|
place_attributes = place_data.except('created_at', 'updated_at', 'latitude', 'longitude')
|
|
place_attributes['lonlat'] = "POINT(#{longitude} #{latitude})"
|
|
place_attributes['latitude'] = latitude
|
|
place_attributes['longitude'] = longitude
|
|
place_attributes.delete('user')
|
|
|
|
begin
|
|
place = Place.create!(place_attributes)
|
|
place.define_singleton_method(:previously_new_record?) { true }
|
|
|
|
place
|
|
rescue ActiveRecord::RecordInvalid => e
|
|
nil
|
|
end
|
|
end
|
|
end
|