Skip to content

Commit

Permalink
Merge pull request #215 from stevenharman/optimize_reservation_on_new…
Browse files Browse the repository at this point in the history
…er_pgs

Optimize PG reservation query for PG >=9.5
  • Loading branch information
albus522 authored Aug 14, 2024
2 parents 15d836b + 38613db commit a50c45b
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion lib/delayed/backend/active_record.rb
Original file line number Diff line number Diff line change
Expand Up @@ -133,9 +133,17 @@ def self.reserve_with_scope_using_optimized_postgres(ready_scope, worker, now)
# Note: active_record would attempt to generate UPDATE...LIMIT like
# SQL for Postgres if we use a .limit() filter, but it would not
# use 'FOR UPDATE' and we would have many locking conflicts
# On PostgreSQL >= 9.5 we leverage SKIP LOCK to avoid multiple workers blocking each other
# when attempting to get the next available job
# https://www.postgresql.org/docs/9.5/sql-select.html#SQL-FOR-UPDATE-SHARE
skip_locked = ""
if connection.respond_to?(:postgresql_version) && connection.postgresql_version >= 90500
skip_locked = " SKIP LOCKED"
end

quoted_name = connection.quote_table_name(table_name)
subquery = ready_scope.limit(1).lock(true).select("id").to_sql
sql = "UPDATE #{quoted_name} SET locked_at = ?, locked_by = ? WHERE id IN (#{subquery}) RETURNING *"
sql = "UPDATE #{quoted_name} SET locked_at = ?, locked_by = ? WHERE id IN (#{subquery}#{skip_locked}) RETURNING *"
reserved = find_by_sql([sql, now, worker.name])
reserved[0]
end
Expand Down

0 comments on commit a50c45b

Please sign in to comment.