class Upsert

# File lib/upsert.rb, line 128
def adapter(metal)
  metal_class_name = class_name metal
  METAL_CLASS_ALIAS.fetch(metal_class_name, metal_class_name).gsub /\W+/, '_'
end

# File lib/upsert.rb, line 96
def batch(connection, table_name, options = {})
  upsert = new connection, table_name, options
  yield upsert
end

# File lib/upsert.rb, line 78
def binary(v)
  Binary.new v
end

# File lib/upsert.rb, line 105
def class_name(metal)
  if RUBY_PLATFORM == 'java'
    metal.class.name || metal.get_class.name
  else
    metal.class.name
  end
end

# File lib/upsert.rb, line 70
def clear_database_functions(connection)
  dummy = new(connection, :dummy)
  dummy.clear_database_functions
end

# File lib/upsert.rb, line 114
def flavor(metal)
  case class_name(metal)
  when /sqlite/i
    'Sqlite3'
  when /mysql/i
    'Mysql'
  when /pg/i, /postgres/i
    'Postgresql'
  else
    raise "[upsert] #{metal} not supported"
  end
end

# File lib/upsert.rb, line 21
def logger
  @logger || MUTEX_FOR_PERFORM.synchronize do
    @logger ||= if defined?(::Rails) and (rails_logger = ::Rails.logger)
      rails_logger
    elsif defined?(::ActiveRecord) and ::ActiveRecord.const_defined?(:Base) and (ar_logger = ::ActiveRecord::Base.logger)
      ar_logger
    else
      my_logger = Logger.new $stderr
      case ENV['UPSERT_DEBUG']
      when 'true'
        my_logger.level = Logger::DEBUG
      when 'false'
        my_logger.level = Logger::INFO
      end
      my_logger
    end
  end
end

# File lib/upsert.rb, line 134
def metal(connection)
  metal = connection.respond_to?(:raw_connection) ? connection.raw_connection : connection
  if metal.class.name.to_s.start_with?('ActiveRecord::ConnectionAdapters')
    metal = metal.connection
  end
  metal
end

# File lib/upsert.rb, line 44
def mutex_for_function(upsert, row)
  retrieve_mutex(upsert.table_name, row.selector.keys, row.setter.keys)
end

# File lib/upsert.rb, line 40
def mutex_for_row(upsert, row)
  retrieve_mutex(upsert.table_name, row.selector.keys)
end

# File lib/upsert.rb, line 212
def initialize(connection, table_name, options = {})
  @table_name = self.class.normalize_table_name(table_name)
  metal = Upsert.metal connection
  @flavor = Upsert.flavor metal
  @adapter = Upsert.adapter metal
  # todo memoize
  Dir[File.expand_path("../upsert/**/{#{flavor.downcase},#{adapter}}.rb", __FILE__)].each do |path|
    require path
  end
  @connection = Connection.const_get(adapter).new self, metal
  @merge_function_class = MergeFunction.const_get adapter
  @merge_function_cache = {}
  @assume_function_exists = options.fetch :assume_function_exists, @flavor != "Postgresql"

  @merge_function_mutex = Mutex.new
  @row_mutex = Mutex.new
end

# File lib/upsert.rb, line 277
def self.normalize_table_name(table_name)
  if defined?(Sequel) && table_name.is_a?(::Sequel::SQL::QualifiedIdentifier)
    [table_name.table, table_name.column]
  else
    [*table_name].map(&:to_s)
  end
end

# File lib/upsert.rb, line 49
def retrieve_mutex(*args)
  # ||= isn't an atomic operation
  MUTEX_FOR_PERFORM.synchronize do
    @mutex_cache ||= {}
  end

  @mutex_cache.fetch(args.flatten.join('::')) do |k|
    MUTEX_FOR_PERFORM.synchronize do
      # We still need the ||= because this block could have
      # theoretically been entered simultaneously by two threads
      # but the actual assignment is protected by the mutex
      @mutex_cache[k] ||= Mutex.new
    end
  end
end

# File lib/upsert.rb, line 143
def utc(time)
  if time.is_a? DateTime
    usec = time.sec_fraction * SEC_FRACTION
    if time.offset != 0
      time = time.new_offset(0)
    end
    Time.utc time.year, time.month, time.day, time.hour, time.min, time.sec, usec
  elsif time.utc?
    time
  else
    time.utc
  end
end

# File lib/upsert.rb, line 158
def utc_iso8601(time, tz = true)
  t = utc time
  s = t.strftime(ISO8601_DATETIME) + '.' + (USEC_SPRINTF % t.usec)
  tz ? (s + UTC_TZ) : s
end

# File lib/upsert.rb, line 204
def assume_function_exists?
  @assume_function_exists
end

# File lib/upsert.rb, line 254
def clear_database_functions
  merge_function_class.clear connection
end

# File lib/upsert.rb, line 272
def column_definitions
  @column_definitions ||= ColumnDefinition.const_get(flavor).all connection, quoted_table_name
end

# File lib/upsert.rb, line 258
def merge_function(row)
  cache_key = [row.selector.keys, row.setter.keys]
  self.class.mutex_for_function(self, row).synchronize do
    @merge_function_cache[cache_key] ||=
      merge_function_class.new(self, row.selector.keys, row.setter.keys, assume_function_exists?)
  end
end

# File lib/upsert.rb, line 267
def quoted_table_name
  @quoted_table_name ||= table_name.map { |t| connection.quote_ident(t) }.join(".")
end

# File lib/upsert.rb, line 245
def row(selector, setter = {}, options = nil)
  row_object = Row.new(selector, setter, options)
  self.class.mutex_for_row(self, row_object).synchronize do
    merge_function(row_object).execute(row_object)
    nil
  end
end