Sha256: 149ceb84aa9da74f753f44863951e817a1bf2682ddc213bb38a5be3179e23b33
Contents?: true
Size: 1.68 KB
Versions: 6
Compression:
Stored size: 1.68 KB
Contents
require 'upsert/merge_function/postgresql' class Upsert class MergeFunction # @private class PG_Connection < MergeFunction include Postgresql def execute(row) first_try = true values = [] values += row.selector.values values += row.setter.values hstore_delete_handlers.each do |hstore_delete_handler| values << row.hstore_delete_keys.fetch(hstore_delete_handler.name, []) end Upsert.logger.debug do %{[upsert]\n\tSelector: #{row.selector.inspect}\n\tSetter: #{row.setter.inspect}} end begin connection.execute sql, values.map { |v| connection.bind_value v } rescue PG::Error => pg_error if pg_error.message =~ /function #{name}.* does not exist/i if first_try Upsert.logger.info %{[upsert] Function #{name.inspect} went missing, trying to recreate} first_try = false create! retry else Upsert.logger.info %{[upsert] Failed to create function #{name.inspect} for some reason} raise pg_error end else raise pg_error end end end # strangely ? can't be used as a placeholder def sql @sql ||= begin bind_params = [] i = 1 (selector_keys.length + setter_keys.length).times do bind_params << "$#{i}" i += 1 end hstore_delete_handlers.length.times do bind_params << "$#{i}::text[]" i += 1 end %{SELECT #{name}(#{bind_params.join(', ')})} end end end end end
Version data entries
6 entries across 6 versions & 1 rubygems