小编tbu*_*ehl的帖子

我需要以原子方式递增模型计数器并使用其新值(由Sidekiq作业处理).

目前,我使用

Group.increment_counter :tasks_count, @task.id

在我的模型中,原子地递增计数器.

但是,如果计数器具有例如值,我还需要其新值来发送通知50.有任何想法吗？锁定桌子/行还是有更简单的方法？

编辑/解决方案

基于mu太短的答案和Rails的update_counters方法,我实现了一个实例方法(用PostgreSQL测试).

def self.increment_counter_and_return_value(counter_name, id)
  quoted_column = connection.quote_column_name(counter_name)
  quoted_table = connection.quote_table_name(table_name)
  quoted_primary_key = connection.quote_column_name(primary_key)
  quoted_primary_key_value = connection.quote(id)

  sql = "UPDATE #{quoted_table} SET #{quoted_column} = COALESCE(#{quoted_column}, 0) + 1 WHERE #{quoted_table}.#{quoted_primary_key} = #{quoted_primary_key_value} RETURNING #{quoted_column}"
  connection.select_value(sql).to_i
end

使用它像:

Group.increment_counter_and_return_value(:tasks_count, @task.id)

它用于RETURNING在同一查询中获取新值.