Contents

module SchemaPlusDbDefault
  module Middleware

    module Query
      module ExecCache

        module Postgresql

          # Middleware to replace each ActiveRecord::DB_DEFAULT with a literal
          # DEFAULT in the sql string, for postgresql.  The underlying pg gem provides no
          # way to bind a value that will replace $n with DEFAULT.
          def before(env)
            if env.binds.any?{ |col, val| val.equal? ::ActiveRecord::DB_DEFAULT}
              j = 0
              env.binds.each_with_index do |(col, val), i|
                if val.equal? ::ActiveRecord::DB_DEFAULT
                  env.sql = env.sql.sub(/\$#{i+1}/, 'DEFAULT')
                else
                  env.sql = env.sql.sub(/\$#{i+1}/, "$#{j+1}") if i != j
                  j += 1
                end
              end
              env.binds = env.binds.reject{|col, val| val.equal? ::ActiveRecord::DB_DEFAULT}
            end
          end
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
schema_plus-2.0.0.pre3	lib/schema_plus_db_default/middleware.rb