upsert 2.9.10-java

data/lib/upsert/connection/postgresql.rb

@@ -0,0 +1,24 @@
+class Upsert
+  class Connection
+    # @private
+    module Postgresql
+      def bind_value(v)
+        case v
+        when Array
+          # pg array escaping lifted from https://github.com/tlconnor/activerecord-postgres-array/blob/master/lib/activerecord-postgres-array/array.rb
+          '{' + v.map do |vv|
+            vv = vv.to_s.dup
+            vv.gsub!(/\\/, '\&\&')
+            vv.gsub!(/"/, '\"')
+            %{"#{vv}"}
+          end.join(',') + '}'
+        when Hash
+          # you must require 'pg_hstore' from the 'pg-hstore' gem yourself
+          ::PgHstore.dump v, true
+        else
+          super
+        end
+      end
+    end
+  end
+end

data/lib/upsert/connection/sqlite3.rb

@@ -0,0 +1,19 @@
+class Upsert
+  class Connection
+    # @private
+    module Sqlite3
+      def bind_value(v)
+        case v
+        when BigDecimal
+          v.to_s('F')
+        when TrueClass
+          't'
+        when FalseClass
+          'f'
+        else
+          super
+        end
+      end
+    end
+  end
+end

data/lib/upsert/merge_function.rb

@@ -0,0 +1,73 @@
+require 'zlib'
+require 'upsert/version'
+
+class Upsert
+  # @private
+  class MergeFunction
+    MAX_NAME_LENGTH = 62
+    NAME_PREFIX = "upsert#{Upsert::VERSION.gsub('.', '_')}"
+
+    class << self
+      def unique_name(table_name, selector_keys, setter_keys)
+        parts = [
+          NAME_PREFIX,
+          [*table_name].join("_").gsub(/[^\w_]+/, "_"),
+          'SEL',
+          selector_keys.join('_A_').gsub(" ","_"),
+          'SET',
+          setter_keys.join('_A_').gsub(" ","_")
+        ].join('_')
+        if parts.length > MAX_NAME_LENGTH
+          # maybe i should md5 instead
+          crc32 = Zlib.crc32(parts).to_s
+          [ parts[0..MAX_NAME_LENGTH-10], crc32 ].join
+        else
+          parts
+        end
+      end
+    end
+
+    attr_reader :controller
+    attr_reader :selector_keys
+    attr_reader :setter_keys
+
+    def initialize(controller, selector_keys, setter_keys, assume_function_exists)
+      @controller = controller
+      @selector_keys = selector_keys
+      @setter_keys = setter_keys
+      @assume_function_exists = assume_function_exists
+      validate!
+      create! unless @assume_function_exists
+    end
+
+    def name
+      @name ||= self.class.unique_name table_name, selector_keys, setter_keys
+    end
+
+    def connection
+      controller.connection
+    end
+
+    def table_name
+      controller.table_name
+    end
+
+    def quoted_table_name
+      controller.quoted_table_name
+    end
+
+    def column_definitions
+      controller.column_definitions
+    end
+
+    private
+
+    def validate!
+      possible = column_definitions.map(&:name)
+      invalid = (setter_keys + selector_keys).uniq - possible
+      if invalid.any?
+        raise ArgumentError, "[Upsert] Invalid column(s): #{invalid.map(&:inspect).join(', ')}"
+      end
+    end
+  end
+end

data/lib/upsert/merge_function/Java_ComMysqlJdbc_JDBC4Connection.rb

@@ -0,0 +1,42 @@
+require 'upsert/merge_function/mysql'
+
+class Upsert
+  class MergeFunction
+    # @private
+    class Java_ComMysqlJdbc_JDBC4Connection < MergeFunction
+      include Mysql
+
+      def sql
+        @sql ||= begin
+          bind_params = Array.new(selector_keys.length + setter_keys.length, '?')
+          %{CALL #{name}(#{bind_params.join(', ')})}
+        end
+      end
+
+      def execute(row)
+        first_try = true
+        bind_selector_values = row.selector.values.map { |v| connection.bind_value v }
+        bind_setter_values = row.setter.values.map { |v| connection.bind_value v }
+        begin
+          connection.execute sql, (bind_selector_values + bind_setter_values)
+        rescue com.mysql.jdbc.exceptions.jdbc4.MySQLSyntaxErrorException => e
+          if e.message =~ /PROCEDURE.*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 e
+            end
+          else
+            raise e
+          end
+        end
+      end
+
+
+    end
+  end
+end

data/lib/upsert/merge_function/Java_OrgPostgresqlJdbc_PgConnection.rb

@@ -0,0 +1,27 @@
+require 'upsert/merge_function/postgresql'
+
+class Upsert
+  class MergeFunction
+    # @private
+    class Java_OrgPostgresqlJdbc_PgConnection < MergeFunction
+      ERROR_CLASS = org.postgresql.util.PSQLException
+      include Postgresql
+
+      def execute_parameterized(query, args = [])
+        query_args = []
+        query = query.gsub(/\$(\d+)/) do |str|
+          query_args << args[Regexp.last_match[1].to_i - 1]
+          "?"
+        end
+        controller.connection.execute(query, query_args)
+      end
+
+      def unique_index_on_selector?
+        return @unique_index_on_selector if defined?(@unique_index_on_selector)
+        @unique_index_on_selector = unique_index_columns.any? do |row|
+          row["index_columns"].sort == selector_keys.sort
+        end
+      end
+    end
+  end
+end

data/lib/upsert/merge_function/Java_OrgSqlite_Conn.rb

@@ -0,0 +1,10 @@
+require 'upsert/merge_function/sqlite3'
+
+class Upsert
+  class MergeFunction
+    # @private
+    class Java_OrgSqlite_Conn < MergeFunction
+      include Sqlite3
+    end
+  end
+end

data/lib/upsert/merge_function/Mysql2_Client.rb

@@ -0,0 +1,36 @@
+require 'upsert/merge_function/mysql'
+
+class Upsert
+  class MergeFunction
+    # @private
+    class Mysql2_Client < MergeFunction
+      include Mysql
+
+      def sql(row)
+        quoted_params = (row.selector.values + row.setter.values).map { |v| connection.quote_value v }
+        %{CALL #{name}(#{quoted_params.join(', ')})}
+      end
+
+      def execute(row)
+        first_try = true
+        begin
+          connection.execute sql(row)
+        rescue Mysql2::Error => e
+          if e.message =~ /PROCEDURE.*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 e
+            end
+          else
+            raise e
+          end
+        end
+      end
+    end
+  end
+end

data/lib/upsert/merge_function/PG_Connection.rb

@@ -0,0 +1,26 @@
+require 'upsert/merge_function/postgresql'
+
+class Upsert
+  class MergeFunction
+    # @private
+    class PG_Connection < MergeFunction
+      ERROR_CLASS = PG::Error
+      include Postgresql
+
+      def execute_parameterized(query, args = [])
+        controller.connection.execute(query, args)
+      end
+
+      def unique_index_on_selector?
+        return @unique_index_on_selector if defined?(@unique_index_on_selector)
+
+        type_map = PG::TypeMapByColumn.new([PG::TextDecoder::Array.new])
+        res = unique_index_columns.tap { |r| r.type_map = type_map }
+
+        @unique_index_on_selector = res.values.any? do |row|
+          row.first.sort == selector_keys.sort
+        end
+      end
+    end
+  end
+end

data/lib/upsert/merge_function/SQLite3_Database.rb

@@ -0,0 +1,10 @@
+require 'upsert/merge_function/sqlite3'
+
+class Upsert
+  class MergeFunction
+    # @private
+    class SQLite3_Database < MergeFunction
+      include Sqlite3
+    end
+  end
+end

data/lib/upsert/merge_function/mysql.rb

@@ -0,0 +1,66 @@
+class Upsert
+  class MergeFunction
+    # @private
+    module Mysql
+      def self.included(klass)
+        klass.extend ClassMethods
+      end
+
+      module ClassMethods
+        # http://stackoverflow.com/questions/733349/list-of-stored-procedures-functions-mysql-command-line
+        def clear(connection)
+          connection.execute("SHOW PROCEDURE STATUS WHERE Db = DATABASE() AND Name LIKE '#{MergeFunction::NAME_PREFIX}%'").map do |row|
+            row['Name'] || row['ROUTINE_NAME']
+          end.each do |name|
+            connection.execute "DROP PROCEDURE IF EXISTS #{connection.quote_ident(name)}"
+          end
+        end
+      end
+
+      # http://stackoverflow.com/questions/11371479/how-to-translate-postgresql-merge-db-aka-upsert-function-into-mysql/
+      def create!
+        Upsert.logger.info "[upsert] Creating or replacing database function #{name.inspect} on table #{table_name.inspect} for selector #{selector_keys.map(&:inspect).join(', ')} and setter #{setter_keys.map(&:inspect).join(', ')}"
+        selector_column_definitions = column_definitions.select { |cd| selector_keys.include?(cd.name) }
+        setter_column_definitions = column_definitions.select { |cd| setter_keys.include?(cd.name) }
+        update_column_definitions = setter_column_definitions.select { |cd| cd.name !~ CREATED_COL_REGEX }
+        quoted_name = connection.quote_ident name
+        connection.execute "DROP PROCEDURE IF EXISTS #{quoted_name}"
+        connection.execute(%{
+          CREATE PROCEDURE #{quoted_name}(#{(selector_column_definitions.map(&:to_selector_arg) + setter_column_definitions.map(&:to_setter_arg)).join(', ')})
+          BEGIN
+            DECLARE done BOOLEAN;
+            REPEAT
+              BEGIN
+                -- If there is a unique key constraint error then 
+                -- someone made a concurrent insert. Reset the sentinel
+                -- and try again.
+                DECLARE ER_DUP_UNIQUE CONDITION FOR 23000;
+                DECLARE ER_INTEG CONDITION FOR 1062;
+                DECLARE CONTINUE HANDLER FOR ER_DUP_UNIQUE BEGIN
+                  SET done = FALSE;
+                END;
+                
+                DECLARE CONTINUE HANDLER FOR ER_INTEG BEGIN
+                  SET done = TRUE;
+                END;
+
+                SET done = TRUE;
+                SELECT COUNT(*) INTO @count FROM #{quoted_table_name} WHERE #{selector_column_definitions.map(&:to_selector).join(' AND ')};
+                -- Race condition here. If a concurrent INSERT is made after
+                -- the SELECT but before the INSERT below we'll get a duplicate
+                -- key error. But the handler above will take care of that.
+                IF @count > 0 THEN 
+                  -- UPDATE table_name SET b = b_SET WHERE a = a_SEL;
+                  UPDATE #{quoted_table_name} SET #{update_column_definitions.map(&:to_setter).join(', ')} WHERE #{selector_column_definitions.map(&:to_selector).join(' AND ')};
+                ELSE
+                  -- INSERT INTO table_name (a, b) VALUES (k, data);
+                  INSERT INTO #{quoted_table_name} (#{setter_column_definitions.map(&:quoted_name).join(', ')}) VALUES (#{setter_column_definitions.map(&:to_setter_value).join(', ')});
+                END IF;
+              END;
+            UNTIL done END REPEAT;
+          END
+        })
+      end
+    end
+  end
+end

data/lib/upsert/merge_function/postgresql.rb

@@ -0,0 +1,365 @@
+class Upsert
+  class MergeFunction
+    # @private
+    module Postgresql
+      def self.included(klass)
+        klass.extend ClassMethods
+      end
+
+      module ClassMethods
+        def clear(connection)
+          # http://stackoverflow.com/questions/7622908/postgresql-drop-function-without-knowing-the-number-type-of-parameters
+          connection.execute(%{
+            CREATE OR REPLACE FUNCTION pg_temp.upsert_delfunc(text)
+              RETURNS void AS
+            $BODY$
+            DECLARE
+              _sql text;
+            BEGIN
+            FOR _sql IN
+              SELECT 'DROP FUNCTION ' || quote_ident(n.nspname)
+                               || '.' || quote_ident(p.proname)
+                               || '(' || pg_catalog.pg_get_function_identity_arguments(p.oid) || ');'
+              FROM   pg_catalog.pg_proc p
+              LEFT   JOIN pg_catalog.pg_namespace n ON n.oid = p.pronamespace
+              WHERE  p.proname = $1
+              AND    pg_catalog.pg_function_is_visible(p.oid) -- you may or may not want this
+            LOOP
+              EXECUTE _sql;
+            END LOOP;
+            END;
+            $BODY$
+              LANGUAGE plpgsql;
+          })
+          connection.execute(%{SELECT proname FROM pg_proc WHERE proname LIKE '#{MergeFunction::NAME_PREFIX}%'}).each do |row|
+            k = row['proname']
+            next if k == 'upsert_delfunc'
+            Upsert.logger.info %{[upsert] Dropping function #{k.inspect}}
+            connection.execute %{SELECT pg_temp.upsert_delfunc('#{k}')}
+          end
+        end
+      end
+
+      attr_reader :quoted_setter_names
+      attr_reader :quoted_selector_names
+
+      def initialize(controller, *args)
+        super
+        @quoted_setter_names = setter_keys.map { |k| connection.quote_ident k }
+        @quoted_selector_names = selector_keys.map { |k| connection.quote_ident k }
+      end
+
+      def execute(row)
+        use_pg_native? ? pg_native(row) : pg_function(row)
+      end
+
+      def pg_function(row)
+        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
+
+        first_try = true
+        begin
+          create! if !@assume_function_exists && (connection.in_transaction? && !function_exists?)
+          execute_parameterized(sql, values.map { |v| connection.bind_value v })
+        rescue self.class::ERROR_CLASS => 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
+            end
+            Upsert.logger.info %{[upsert] Failed to create function #{name.inspect} for some reason}
+            raise pg_error
+          else
+            raise pg_error
+          end
+        end
+      end
+
+      def function_exists?
+        @function_exists ||= controller.connection.execute("SELECT count(*) AS cnt FROM pg_proc WHERE lower(proname) = lower('#{name}')").first["cnt"].to_i > 0
+      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
+
+      def use_pg_native?
+        return @use_pg_native if defined?(@use_pg_native)
+
+        @use_pg_native = server_version >= 90500 && unique_index_on_selector?
+        Upsert.logger.warn "[upsert] WARNING: Not using native PG CONFLICT / UPDATE" unless @use_pg_native
+        @use_pg_native
+      end
+
+      def server_version
+        @server_version ||= Upsert::MergeFunction::Postgresql.extract_version(
+          controller.connection.execute("SHOW server_version").first["server_version"]
+        )
+      end
+
+      # Extracted from https://github.com/dr-itz/activerecord-jdbc-adapter/blob/master/lib/arjdbc/postgresql/adapter.rb
+      def self.extract_version(version_string)
+        # Use the same versioning format as jdbc-postgresql and libpq
+        # https://github.com/dr-itz/activerecord-jdbc-adapter/commit/fd79756374c62fa9d009995dd1914d780e6a3dbf
+        # https://github.com/postgres/postgres/blob/master/src/interfaces/libpq/fe-exec.c
+        if (match = version_string.match(/([\d\.]*\d).*?/))
+          version = match[1].split('.').map(&:to_i)
+          # PostgreSQL version representation does not have more than 4 digits
+          # From version 10 onwards, PG has changed its versioning policy to
+          # limit it to only 2 digits. i.e. in 10.x, 10 being the major
+          # version and x representing the patch release
+          # Refer to:
+          #   https://www.postgresql.org/support/versioning/
+          #   https://www.postgresql.org/docs/10/static/libpq-status.html -> PQserverVersion()
+          # for more info
+
+          if version.size >= 3
+            (version[0] * 100 + version[1]) * 100 + version[2]
+          elsif version.size == 2
+            if version[0] >= 10
+              version[0] * 100 * 100 + version[1]
+            else
+              (version[0] * 100 + version[1]) * 100
+            end
+          elsif version.size == 1
+            version[0] * 100 * 100
+          else
+            0
+          end
+        else
+          0
+        end
+      end
+
+      def unique_index_columns
+        if table_name.is_a?(Array) && table_name.length > 1
+          schema_argument = '$2'
+          table_name_arguments = table_name
+        else
+          schema_argument = 'ANY(current_schemas(true)::text[])'
+          table_name_arguments = [*table_name]
+        end
+
+        table_name_arguments.reverse!
+
+        execute_parameterized(
+          %{
+            SELECT
+            ARRAY(
+              SELECT pg_get_indexdef(pg_index.indexrelid, k + 1, TRUE)
+              FROM
+                generate_subscripts(pg_index.indkey, 1) AS k
+              ORDER BY k
+            ) AS index_columns
+              FROM pg_index
+              JOIN pg_class AS idx ON idx.oid = pg_index.indexrelid
+              JOIN pg_class AS tbl ON tbl.oid = pg_index.indrelid
+              JOIN pg_namespace ON pg_namespace.oid = idx.relnamespace
+            WHERE pg_index.indisunique IS TRUE AND pg_namespace.nspname = #{schema_argument} AND tbl.relname = $1
+          },
+          table_name_arguments
+        )
+      end
+
+      def pg_native(row)
+        bind_setter_values = row.setter.values.map { |v| connection.bind_value v }
+        # TODO: Is this needed?
+        row_syntax = server_version >= 100 ? "ROW" : ""
+
+        upsert_sql = %{
+          INSERT INTO #{quoted_table_name} (#{quoted_setter_names.join(',')})
+          VALUES (#{insert_bind_placeholders(row).join(', ')})
+          ON CONFLICT(#{quoted_selector_names.join(', ')})
+          DO UPDATE SET #{quoted_setter_names.zip(conflict_bind_placeholders(row)).map { |n, v| "#{n} = #{v}" }.join(', ')}
+        }
+
+        execute_parameterized(upsert_sql, bind_setter_values)
+      end
+
+      def hstore_delete_function(sql, row, column_definition)
+        parts = []
+        if row.hstore_delete_keys.key?(column_definition.name)
+          parts << "DELETE("
+        end
+        parts << sql
+        if row.hstore_delete_keys.key?(column_definition.name)
+          keys = row.hstore_delete_keys[column_definition.name].map { |k| "'#{k.to_s.gsub("'", "\\'")}'" }
+          parts << ", ARRAY[#{keys.join(', ')}])"
+        end
+
+        parts.join(" ")
+      end
+
+      def insert_bind_placeholders(row)
+        if row.hstore_delete_keys.empty?
+          @insert_bind_placeholders ||= setter_column_definitions.each_with_index.map do |column_definition, i|
+            if column_definition.hstore?
+              "CAST($#{i + 1} AS hstore)"
+            else
+              "$#{i + 1}"
+            end
+          end
+        else
+          setter_column_definitions.each_with_index.map do |column_definition, i|
+            idx = i + 1
+            if column_definition.hstore?
+              hstore_delete_function("CAST($#{idx} AS hstore)", row, column_definition)
+            else
+              "$#{idx}"
+            end
+          end
+        end
+      end
+
+      def conflict_bind_placeholders(row)
+        if row.hstore_delete_keys.empty?
+          @conflict_bind_placeholders ||= setter_column_definitions.each_with_index.map do |column_definition, i|
+            idx = i + 1
+            if column_definition.hstore?
+              "CASE WHEN #{quoted_table_name}.#{column_definition.quoted_name} IS NULL THEN CAST($#{idx} AS hstore) ELSE" \
+                + " (#{quoted_table_name}.#{column_definition.quoted_name} || CAST($#{idx} AS hstore))" \
+                + " END"
+            else
+              "$#{idx}"
+            end
+          end
+        else
+          setter_column_definitions.each_with_index.map do |column_definition, i|
+            idx = i + 1
+            if column_definition.hstore?
+              "CASE WHEN #{quoted_table_name}.#{column_definition.quoted_name} IS NULL THEN " \
+                + hstore_delete_function("CAST($#{idx} AS hstore)", row, column_definition) \
+                + " ELSE " \
+                + hstore_delete_function("(#{quoted_table_name}.#{column_definition.quoted_name} || CAST($#{idx} AS hstore))", row, column_definition) \
+                + " END"
+            else
+              "$#{idx}"
+            end
+          end
+        end
+      end
+
+      class HstoreDeleteHandler
+        attr_reader :merge_function
+        attr_reader :column_definition
+        def initialize(merge_function, column_definition)
+          @merge_function = merge_function
+          @column_definition = column_definition
+        end
+        def name
+          column_definition.name
+        end
+        def to_arg
+          "#{quoted_name} text[]"
+        end
+        # use coalesce(foo, '{}':text[])
+        def to_setter
+          "#{column_definition.quoted_name} = DELETE(#{column_definition.quoted_name}, #{quoted_name})"
+        end
+        def to_pgsql
+          %{
+            IF array_length(#{quoted_name}, 1) > 0 THEN
+              UPDATE #{merge_function.quoted_table_name} SET #{to_setter}
+                WHERE #{merge_function.selector_column_definitions.map(&:to_selector).join(' AND ') };
+            END IF;
+          }.gsub(/\s+/, ' ')
+        end
+        private
+        def quoted_name
+          @quoted_name ||= merge_function.connection.quote_ident "_delete_#{column_definition.name}"
+        end
+      end
+
+      def hstore_delete_handlers
+        @hstore_delete_handlers ||= setter_column_definitions.select do |column_definition|
+          column_definition.hstore?
+        end.map do |column_definition|
+          HstoreDeleteHandler.new self, column_definition
+        end
+      end
+
+      def selector_column_definitions
+        column_definitions.select { |cd| selector_keys.include?(cd.name) }
+      end
+
+      def setter_column_definitions
+        column_definitions.select { |cd| setter_keys.include?(cd.name) }
+      end
+
+      def update_column_definitions
+        setter_column_definitions.select { |cd| cd.name !~ CREATED_COL_REGEX }
+      end
+
+      # the "canonical example" from http://www.postgresql.org/docs/9.1/static/plpgsql-control-structures.html#PLPGSQL-UPSERT-EXAMPLE
+      # differentiate between selector and setter
+      def create!
+        Upsert.logger.info "[upsert] Creating or replacing database function #{name.inspect} on table #{table_name.inspect} for selector #{selector_keys.map(&:inspect).join(', ')} and setter #{setter_keys.map(&:inspect).join(', ')}"
+        first_try = true
+        connection.execute(%{
+          CREATE OR REPLACE FUNCTION #{name}(#{(selector_column_definitions.map(&:to_selector_arg) + setter_column_definitions.map(&:to_setter_arg) + hstore_delete_handlers.map(&:to_arg)).join(', ')}) RETURNS VOID AS
+          $$
+          DECLARE
+            first_try INTEGER := 1;
+          BEGIN
+            LOOP
+              -- first try to update the key
+              UPDATE #{quoted_table_name} SET #{update_column_definitions.map(&:to_setter).join(', ')}
+                WHERE #{selector_column_definitions.map(&:to_selector).join(' AND ') };
+              IF found THEN
+                #{hstore_delete_handlers.map(&:to_pgsql).join(' ')}
+                RETURN;
+              END IF;
+              -- not there, so try to insert the key
+              -- if someone else inserts the same key concurrently,
+              -- we could get a unique-key failure
+              BEGIN
+                INSERT INTO #{quoted_table_name}(#{setter_column_definitions.map(&:quoted_name).join(', ')}) VALUES (#{setter_column_definitions.map(&:to_setter_value).join(', ')});
+                #{hstore_delete_handlers.map(&:to_pgsql).join(' ')}
+                RETURN;
+              EXCEPTION WHEN unique_violation THEN
+                -- seamusabshere 9/20/12 only retry once
+                IF (first_try = 1) THEN
+                  first_try := 0;
+                ELSE
+                  RETURN;
+                END IF;
+                -- Do nothing, and loop to try the UPDATE again.
+              END;
+            END LOOP;
+          END;
+          $$
+          LANGUAGE plpgsql;
+        })
+      rescue
+        if first_try and $!.message =~ /tuple concurrently updated/
+          first_try = false
+          retry
+        else
+          raise $!
+        end
+      end
+    end
+  end
+end