Affected rows for ActiveRecord::Base.connection.ex

2019-05-01 05:49发布

问题:

With Rails 4.1.1, using mysql2 adapter:

I am using an ActiveRecord connection to execute a multiple insert in a MySQL table:

ActiveRecord::Base.connection.execute %Q{
    INSERT INTO table (`user_id`, `item_id`) 
    SELECT 1, id FROM items WHERE items.condition IS NOT NULL
}

This works fine, do the job, and returns nil.

Is there a way to get the number of affected rows? (avoiding the need to execute another query)

I have found the documentation of the execute method somewhat sparse.

回答1:

You can use connection.update method which executes expression and returns affected rows count.

ActiveRecord::Base.connection
  .update("INSERT INTO accounts (`name`) VALUES ('first'), ('second')")

=> 2

Rails v4.2.7 doc - http://api.rubyonrails.org/v4.2.7/classes/ActiveRecord/ConnectionAdapters/DatabaseStatements.html#method-i-update

Rails latest doc - http://api.rubyonrails.org/classes/ActiveRecord/ConnectionAdapters/DatabaseStatements.html#method-i-update



回答2:

Here is an easy way to get the number of affected rows using the mysql2 adapter. Put this in a file that gets loaded by your app:

class ActiveRecord::ConnectionAdapters::Mysql2Adapter
  def affected_rows
    @connection.affected_rows
  end
end

Then you will be able to run ActiveRecord::Base.connection.affected_rows to get the number of affected rows.