cartodb-4.42/lib/carto/connector/provider.rb

191 lines
5.5 KiB
Ruby
Raw Normal View History

2024-04-06 13:25:13 +08:00
# Base class for Connector Providers
#
# This is an abstract class; concrete classes derived from this one
# must implement these methods:
#
# * `copy_table(schema_name:, table_name:, limits:)`
# * `list_tables(limits:)`
# * `check_connection`
# * `remote_data_updated?`
# * `table_name`
# * `required_parameters`
# * `optional_parameters`
# * `features_information`
#
module Carto
class Connector
class Provider
# Provider identifier (internal name, used in APIs, etc)
def self.provider_id
must_be_defined_in_derived_class
end
# Human-readable name of the provider
def self.friendly_name
must_be_defined_in_derived_class
end
# This means that the provider is publicly announced (so it is accessible through UI, visible in lists of
# providers, etc.) A provider may be available or not (see Connector.limits) independently of its public status,
# so that a public provider may not be available for all users, and non-public providers may be available to
# some users.
def self.public?
# Providers are public by default
true
end
def initialize(parameters: {}, user: nil, logger: nil, previous_last_modified: nil)
@params = Parameters.new(parameters, required: required_parameters + [:provider], optional: optional_parameters)
@user = user
@logger = logger
# previous_last_modified is the time at which external data had been modified at the previous synchronization
@previous_last_modified = previous_last_modified
end
def errors(only_for: nil)
@params.errors(only_for: only_for)
end
def valid?
errors.empty?
end
def validate!(only: nil)
errors = self.errors(only_for: only)
raise InvalidParametersError.new(message: errors * "\n") if errors.present?
end
def copy_table(schema_name:, table_name:, limits:)
must_be_defined_in_derived_class schema_name: schema_name, table_name: table_name, limits: limits
end
def list_tables(limits:)
must_be_defined_in_derived_class limits: limits
end
def list_projects
must_be_defined_in_derived_class
end
def list_project_datasets(project_id)
must_be_defined_in_derived_class
end
def list_project_dataset_tables(project_id, dataset_id)
must_be_defined_in_derived_class
end
def check_connection
must_be_defined_in_derived_class
end
def dry_run
must_be_defined_in_derived_class
end
def remote_data_updated?
# By default connectors don't check for updated data
true
end
def last_modified
# By default connectors don't support last modified time
nil
end
# Name of the table to be imported
def table_name
must_be_defined_in_derived_class
end
# Parameters required by this connector provider
def required_parameters
must_be_defined_in_derived_class
end
# Optional parameters accepted by this connector provider
def optional_parameters
must_be_defined_in_derived_class
end
# Parameters accepted by this connector provider
def accepted_parameters
required_parameters + optional_parameters
end
def self.information
# For convenience we'll use instance methods to provide the information
# en each class. Otherwise all the information needed by such methods
# would have to be defined in class methods too.
test_provider = new(parameters: {})
{
features: test_provider.features_information,
parameters: test_provider.parameters_information
}
end
def features_information
must_be_defined_in_derived_class
end
def parameters_information
# TODO: add templates with parameter descriptions
info = {}
required_parameters.each do |name|
# TODO: description = load template for parameter name of @provider.name
info[name.to_s] = {
required: true
}
end
optional_parameters.each do |name|
# TODO: description = load template for parameter name of @provider.name
info[name.to_s] = {
required: false
}
end
info
end
def provider_id
self.class.provider_id
end
METADATA_KEYS = {
id: :provider_id,
name: :friendly_name,
public: :public?
}
class <<self
def metadata(options)
options.each do |key, value|
method = METADATA_KEYS[key]
raise "Invalid Provider metadata key: #{key.inspect}" unless method
define_singleton_method(method) { value.freeze }
end
end
def optional_parameters(params)
define_method(:optional_parameters) { params.freeze }
end
def required_parameters(params)
define_method(:required_parameters) { params.freeze }
end
end
def log(message, truncate = true)
@logger.append message, truncate if @logger
end
private
def must_be_defined_in_derived_class(*_)
raise NotImplementedError, "Method \"#{caller_locations(1, 1)[0].label}\" must be defined in derived class"
end
def self.must_be_defined_in_derived_class(*_)
raise NotImplementedError, "Class method \"#{caller_locations(1, 1)[0].label}\" must be defined in derived class"
end
end
end
end