我需要以原子方式递增模型计数器并使用其新值(由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
在同一查询中获取新值。
答案 0 :(得分:5)
您的Group.increment_counter
调用会将这样的SQL发送到数据库:
update groups
set tasks_count = coalesce(tasks_counter, 0) + 1
where id = X
其中X
为@task.id
。获取新tasks_counter
值的SQL方法是包含RETURNING子句:
update groups
set tasks_count = coalesce(tasks_counter, 0) + 1
where id = X
returning tasks_count
我不知道有任何方便的Railsy方法将SQL提供给数据库。通常的Rails方法是做一堆锁定并重新加载@task
或跳过锁定并希望最好:
Group.increment_counter :tasks_count, @task.id
@task.reload
# and now look at @task.tasks_count to get the new value
你可以像这样使用RETURNING:
new_count = Group.connection.execute(%Q{
update groups
set tasks_count = coalesce(tasks_counter, 0) + 1
where id = #{Group.connection.quote(@task.id)}
returning tasks_count
}).first['tasks_count'].to_i
您可能希望隐藏Group
上方法背后的混乱,以便您可以这样说:
n = Group.increment_tasks_count_for(@task)
# or
n = @task.increment_tasks_count