cartodb/app/models/synchronization/member.rb

575 lines
20 KiB
Ruby
Raw Permalink Normal View History

2020-06-15 10:58:47 +08:00
require 'virtus'
require_relative 'adapter'
require_relative '../../../services/importer/lib/importer'
require_relative '../visualization/collection'
require_relative '../../../services/importer/lib/importer/datasource_downloader'
require_relative '../../../services/datasources/lib/datasources'
require_relative '../log'
require_relative '../../../services/importer/lib/importer/unp'
require_relative '../../../services/importer/lib/importer/post_import_handler'
require_relative '../../../services/importer/lib/importer/overviews'
require_relative '../../../lib/cartodb/errors'
require_relative '../../../lib/cartodb/import_error_codes'
require_relative '../../../services/platform-limits/platform_limits'
require_dependency 'carto/configuration'
include CartoDB::Datasources
module CartoDB
module Synchronization
class << self
attr_accessor :repository
end
class Member
include Carto::Configuration
include Virtus.model
MAX_RETRIES = 10
MIN_INTERVAL_SECONDS = 15 * 60
# Seconds required between manual sync now
SYNC_NOW_TIMESPAN = 900
STATE_CREATED = 'created'
# Already at resque, waiting for slot
STATE_QUEUED = 'queued'
# Actually syncing
STATE_SYNCING = 'syncing'
STATE_SUCCESS = 'success'
STATE_FAILURE = 'failure'
attribute :id, String
attribute :name, String
attribute :interval, Integer, default: 3600
attribute :url, String
attribute :state, String, default: STATE_CREATED
attribute :user_id, String
attribute :created_at, Time
attribute :updated_at, Time
attribute :run_at, Time
attribute :ran_at, Time
attribute :modified_at, Time
attribute :etag, String
attribute :checksum, String
attribute :log_id, String
attribute :error_code, Integer
attribute :error_message, String
attribute :retried_times, Integer, default: 0
attribute :service_name, String
attribute :service_item_id, String
attribute :type_guessing, Boolean, default: true
attribute :quoted_fields_guessing, Boolean, default: true
attribute :content_guessing, Boolean, default: false
attribute :visualization_id, String
def initialize(attributes={}, repository=Synchronization.repository)
super(attributes)
@log = nil
self.log_trace = nil
@repository = repository
self.id ||= @repository.next_id
self.state ||= STATE_CREATED
self.ran_at ||= Time.now
self.interval ||= 3600
self.run_at ||= Time.now + interval
self.retried_times ||= 0
self.log_id ||= log.id unless log.nil?
self.service_name ||= nil
self.service_item_id ||= nil
self.checksum ||= ''
raise InvalidInterval.new unless self.interval >= MIN_INTERVAL_SECONDS
end
def to_s
"<CartoDB::Synchronization::Member id:\"#{@id}\" name:\"#{@name}\" ran_at:\"#{@ran_at}\" run_at:\"#{@run_at}\" " \
"interval:\"#{@interval}\" state:\"#{@state}\" retried_times:\"#{@retried_times}\" log_id:\"#{self.log_id}\" " \
"service_name:\"#{@service_name}\" service_item_id:\"#{@service_item_id}\" checksum:\"#{@checksum}\" " \
"url:\"#{@url}\" error_code:\"#{@error_code}\" error_message:\"#{@error_message}\" modified_at:\"#{@modified_at}\" " \
" user_id:\"#{@user_id}\" type_guessing:\"#{@type_guessing}\" " \
"quoted_fields_guessing:\"#{@quoted_fields_guessing}\" visualization_id:\"#{@visualization_id}\">"
end
def synchronizations_logger
@@synchronizations_logger ||= CartoDB.unformatted_logger(log_file_path("synchronizations.log"))
end
def interval=(seconds=3600)
super(seconds.to_i)
if seconds
self.run_at = Time.now + (seconds.to_i || 3600)
end
seconds
end
def store
raise CartoDB::InvalidMember unless self.valid?
set_timestamps
repository.store(id, attributes.to_hash)
self
end
def fetch
data = repository.fetch(id)
raise KeyError if data.nil?
self.attributes = data
self
end
def fetch_by_visualization_id
data = repository.fetch(visualization_id, "visualization_id")
raise KeyError if data.nil?
self.attributes = data
self
end
def delete
repository.delete(id)
self.attributes.keys.each { |key| self.send("#{key}=", nil) }
self
end
def enqueue
Resque.enqueue(Resque::SynchronizationJobs, job_id: id)
self.error_code = nil
self.error_message = nil
self.state = CartoDB::Synchronization::Member::STATE_QUEUED
self.store
end
# @return bool
def can_manually_sync?
# Last sync ok, last sync failed, or too much time in queued state
( self.state == STATE_SUCCESS ||
self.state == STATE_FAILURE ||
(self.state == STATE_QUEUED && self.updated_at + SYNC_NOW_TIMESPAN < Time.now)
) && (self.ran_at + SYNC_NOW_TIMESPAN < Time.now)
end
# @return bool
def should_auto_sync?
self.state == STATE_SUCCESS && (self.run_at < Time.now)
end
# This should be joined with data_import to stop the madness of duplicated code
def run
importer = nil
self.state = STATE_SYNCING
self.store
# TODO: See if we can remove this code
# First import is a "normal import" so still has no id, then run gets called and will get log first time
# but we need this to fix old logs
if log.nil?
@log = CartoDB::Log.new(type: CartoDB::Log::TYPE_SYNCHRONIZATION, user_id: user.id)
@log.store
self.log_id = @log.id
store
else
@log.type = CartoDB::Log::TYPE_SYNCHRONIZATION
@log.clear
@log.store
end
if user.nil?
raise "Couldn't instantiate synchronization user. Data: #{to_s}"
end
raise "Can't run a synchronization for inactive user: #{user.username}" unless user.reload.active?
if !authorize?(user)
raise CartoDB::Datasources::AuthError.new('User is not authorized to sync tables')
end
runner = service_name == 'connector' ? get_connector : get_runner
database = user.in_database
overviews_creator = CartoDB::Importer2::Overviews.new(runner, user)
importer = CartoDB::Synchronization::Adapter.new(name, runner, database, user, overviews_creator, id)
importer.run
self.ran_at = Time.now
self.run_at = Time.now + interval
if importer.success?
set_success_state_from(importer)
else
set_failure_state_from(importer)
end
store
notify
rescue => exception
if exception.is_a? CartoDB::Datasources::NotFoundDownloadError
CartoDB::Logger.debug(exception: exception, sync_id: id)
else
CartoDB::Logger.error(exception: exception, sync_id: id)
end
log.append_and_store exception.message, truncate = false
log.append exception.backtrace.join("\n"), truncate = false
if importer.nil?
if exception.is_a?(NotFoundDownloadError)
set_general_failure_state_from(exception, 1017, 'File not found, you must import it again')
elsif exception.is_a?(CartoDB::Importer2::FileTooBigError)
set_general_failure_state_from(exception, exception.error_code,
CartoDB::IMPORTER_ERROR_CODES[exception.error_code][:title])
elsif exception.is_a?(AuthError)
set_general_failure_state_from(exception, 1011, 'Unauthorized')
else
set_general_failure_state_from(exception)
end
else
set_failure_state_from(importer)
end
store
if exception.is_a?(TokenExpiredOrInvalidError)
begin
user.oauths.remove(exception.service_name)
rescue => ex
log.append "Exception removing OAuth: #{ex.message}"
log.append ex.backtrace
end
end
notify
self
ensure
CartoDB::PlatformLimits::Importer::UserConcurrentSyncsAmount.new(
user: user, redis: { db: $users_metadata }
).decrement!
end
def get_runner
downloader = get_downloader
post_import_handler = CartoDB::Importer2::PostImportHandler.new
unless downloader.datasource.nil?
case downloader.datasource.class::DATASOURCE_NAME
when Url::ArcGIS::DATASOURCE_NAME
post_import_handler.add_fix_geometries_task
when Search::Twitter::DATASOURCE_NAME
post_import_handler.add_transform_geojson_geom_column
end
end
runner = CartoDB::Importer2::Runner.new(
pg: pg_options,
downloader: downloader,
log: log,
user: user,
unpacker: CartoDB::Importer2::Unp.new(Cartodb.config[:importer], Cartodb.config[:ogr2ogr]),
post_import_handler: post_import_handler,
importer_config: Cartodb.config[:importer]
)
runner.loader_options = ogr2ogr_options.merge content_guessing_options
runner.include_additional_errors_mapping(
AuthError => 1011,
DataDownloadError => 1011,
TokenExpiredOrInvalidError => 1012,
DatasourceBaseError => 1012,
InvalidServiceError => 1012,
MissingConfigurationError => 1012,
UninitializedError => 1012
)
runner
end
def get_connector
CartoDB::Importer2::ConnectorRunner.check_availability!(user)
CartoDB::Importer2::ConnectorRunner.new(
service_item_id,
user: user,
pg: pg_options,
log: log
)
end
def notify
sync_log = {
'name' => self.name,
'sync_time' => self.updated_at - self.created_at,
'sync_timestamp' => Time.now,
'user' => user.username,
'queue_server' => `hostname`.strip,
'resque_ppid' => Process.ppid,
'state' => self.state,
'user_timeout' => ::DataImport.http_timeout_for(user)
}
synchronizations_logger.info(sync_log.to_json)
end
def get_downloader
datasource_name = (service_name.nil? || service_name.size == 0) ? Url::PublicUrl::DATASOURCE_NAME : service_name
if service_item_id.nil? || service_item_id.size == 0
self.service_item_id = url
end
datasource_provider = get_datasource(datasource_name)
if datasource_provider.nil?
raise CartoDB::DataSourceError.new("Datasource #{datasource_name} could not be instantiated")
end
if service_item_id.nil?
raise CartoDB::DataSourceError.new("Datasource #{datasource_name} without item id")
end
log.append "Fetching datasource #{datasource_provider.to_s} metadata for item id #{service_item_id} from user #{user.id}"
metadata = datasource_provider.get_resource_metadata(service_item_id)
if hit_platform_limit?(datasource_provider, metadata, user)
raise CartoDB::Importer2::FileTooBigError.new(metadata.inspect)
end
if datasource_provider.providers_download_url?
metadata_url = metadata[:url]
resource_url = (metadata_url.present? && datasource_provider.providers_download_url?) ? metadata_url : url
raise CartoDB::DataSourceError.new("Missing resource URL to download. Data:#{self}") unless resource_url
log.append "File will be downloaded from #{resource_url}"
http_options = {
http_timeout: DataImport.http_timeout_for(user),
etag: etag,
last_modified: modified_at,
checksum: checksum,
verify_ssl_cert: false
}
CartoDB::Importer2::Downloader.new(user.id,
resource_url,
http_options,
importer_config: Cartodb.config[:importer])
else
log.append 'Downloading file data from datasource'
options = {
http_timeout: DataImport.http_timeout_for(user),
checksum: checksum,
importer_config: Cartodb.config[:importer],
user_id: user.id
}
CartoDB::Importer2::DatasourceDownloader.new(datasource_provider, metadata, options, log)
end
end
def hit_platform_limit?(datasource, metadata, user)
if datasource.has_resource_size?(metadata)
CartoDB::PlatformLimits::Importer::InputFileSize.new({ user: user })
.is_over_limit!(metadata[:size])
else
false
end
end
def set_success_state_from(importer)
log.append_and_store '******** synchronization succeeded ********'
self.log_trace = importer.runner_log_trace
self.state = STATE_SUCCESS
self.etag = importer.etag
self.checksum = importer.checksum
self.error_code = nil
self.error_message = nil
self.retried_times = 0
self.run_at = Time.now + interval
self.modified_at = importer.last_modified
geocode_table
rescue => exception
CartoDB::Logger.error(exception: exception,
message: 'Error updating state for sync table',
sync_id: self.id)
self
end
def set_failure_state_from(importer)
log.append_and_store '******** synchronization failed ********'
self.log_trace = importer.runner_log_trace
log.append "*** Runner log: #{self.log_trace} \n***" unless self.log_trace.nil?
self.state = STATE_FAILURE
self.error_code = importer.error_code.blank? ? 9999 : importer.error_code
self.error_message = importer.error_message
# Try to fill empty messages with the list
if self.error_message.blank? && !self.error_code.nil?
default_message = CartoDB::IMPORTER_ERROR_CODES.fetch(self.error_code, {})
self.error_message = default_message.fetch(:title, '')
end
if self.retried_times < MAX_RETRIES
self.retried_times += 1
self.run_at = Time.now + interval
else
::Resque.enqueue(::Resque::UserJobs::Mail::Sync::MaxRetriesReached, self.user_id,
self.visualization_id, self.name, self.error_code, self.error_message)
end
end
def set_general_failure_state_from(exception, error_code = 99999, error_message = 'Unknown error, please try again')
log.append_and_store '******** synchronization raised exception ********'
self.log_trace = exception.message + ' ' + exception.backtrace.join("\n")
self.state = STATE_FAILURE
self.error_code = error_code
self.error_message = error_message
if self.retried_times < MAX_RETRIES
self.retried_times += 1
self.run_at = Time.now + interval
else
::Resque.enqueue(::Resque::UserJobs::Mail::Sync::MaxRetriesReached, self.user_id,
self.visualization_id, self.name, self.error_code, self.error_message)
end
rescue => e
CartoDB.notify_exception(e,
{
error_code: error_code,
error_message: error_message,
retried_times: self.retried_times
}
)
end
# Tries to run automatic geocoding if present
def geocode_table
return unless table && table.automatic_geocoding
log.append 'Running automatic geocoding...'
table.automatic_geocoding.run
rescue => e
log.append "Error while running automatic geocoding: #{e.message}"
end # geocode_table
def to_hash
attributes.merge({from_external_source: from_external_source?}).to_hash
end
def to_json(*args)
attributes.merge({from_external_source: from_external_source?}).to_json(*args)
end
def valid?
true
end
def set_timestamps
self.created_at ||= Time.now
self.updated_at = Time.now
self
end
def user
@user ||= ::User.where(id: user_id).first
end
def table
if @table.nil?
@table = ::Table.new(name: name, user_id: user.id)
end
@table
end
def visualization
@visualization ||= CartoDB::Visualization::Member.new(id: @visualization_id).fetch
rescue KeyErrror
@visualization = nil
end
def authorize?(user)
user.id == user_id && (!!user.sync_tables_enabled || from_external_source?)
end
def pg_options
SequelRails.configuration.environment_for(Rails.env)
.merge(
username: user.database_username,
password: user.database_password,
database: user.database_name,
host: user.user_database_host
)
end
def ogr2ogr_options
options = Cartodb.config.fetch(:ogr2ogr, {})
if options['binary'].nil? || options['csv_guessing'].nil?
{}
else
ogr_options = {
ogr2ogr_binary: options['binary'],
ogr2ogr_csv_guessing: options['csv_guessing'] && @type_guessing,
quoted_fields_guessing: @quoted_fields_guessing,
}
if options['memory_limit'].present?
ogr_options.merge!(ogr2ogr_memory_limit: options['memory_limit'])
end
return ogr_options
end
end
# TODO code duplicated from data_import.rb, refactor
def content_guessing_options
guessing_config = Cartodb.config.fetch(:importer, {}).deep_symbolize_keys.fetch(:content_guessing, {})
geocoder_config = Cartodb.config.fetch(:geocoder, {}).deep_symbolize_keys
if guessing_config[:enabled] and self.content_guessing and geocoder_config
{ guessing: guessing_config, geocoder: geocoder_config }
else
{ guessing: { enabled: false } }
end
end
def log
return @log unless @log.nil?
log_attributes = {
id: self.log_id
}
log_attributes.merge(user_id: user.id) if user
@log = CartoDB::Log.where(log_attributes).first
end
def valid_uuid?(text)
!!UUIDTools::UUID.parse(text)
rescue TypeError
false
rescue ArgumentError
false
end
def get_datasource(datasource_name)
begin
datasource = DatasourcesFactory.get_datasource(datasource_name, user, {
http_timeout: ::DataImport.http_timeout_for(user)
})
if datasource.kind_of? BaseOAuth
oauth = user.oauths.select(datasource_name)
datasource.token = oauth.token unless oauth.nil?
end
rescue => ex
log.append "Exception: #{ex.message}"
log.append ex.backtrace
datasource = nil
end
datasource
end
def from_external_source?
::ExternalDataImport.where(synchronization_id: self.id).first != nil
end
attr_reader :repository
attr_accessor :log_trace, :service_name, :service_item_id
end
end
end