Add unique index to poll voters table
Note that Rails 7.1 changes `find_or_create_by!` so it calls `create_or_find_by!` when no record is found, meaning we'll rarely get `RecordNotUnique` exceptions when using this method during a race condition. Adding this index means we need to remove the uniqueness validation. According to the `create_or_find_by` documentation [1]: > Columns with unique database constraints should not have uniqueness > validations defined, otherwise create will fail due to validation > errors and find_by will never be called. We're adding a test that checks what happens when using `create_or_find_by!`. Note that we're creating voters combining `create_with` with `find_or_create_by!`. Using `find_or_create_by!(...)` with all attributes (including non-key ones like `origin`) fails when a voter already exists with different values, e.g. an existing `origin: "web"` and an incoming `"booth"`. In this situation the existing record is not matched and the unique index raises an exception. `create_with(...).find_or_create_by!(user: ..., poll: ...)` searches by the unique key only and applies the extra attributes only on creation. Existing voters are returned unchanged, which is the intended behavior. For the `take_votes_from` method, we're handling a (highly unlikely, but theoretically possible) scenario where a user votes at the same time as taking voters from another user. For that, we're doing something similar to what `create_or_find_by!` does: we're updating the `user_id` column inside a new transaction (using a new transactions avoids a `PG::InFailedSqlTransaction` exception when there are duplicate records), and deleting the existing voter when we get a `RecordNotUnique` exception. On `Poll::WebVote` we're simply raising an exception when there's already a user who's voted via booth, because the `Poll::WebVote#update` method should never be called in this case. We still need to use `with_lock` in `Poll::WebVote`, but not due to duplicate voters (`find_or_create_by!` method will now handle the unique record scenario, even in the case of simultaneous transactions), but because we use a uniqueness validation in `Poll::Answer`; this validation would cause an error in simultaneous transactions. [1] https://api.rubyonrails.org/v7.1/classes/ActiveRecord/Relation.html#method-i-create_or_find_by
This commit is contained in:
@@ -15,7 +15,6 @@ class Poll
|
||||
validates :officer_assignment_id, presence: true, if: ->(voter) { voter.origin == "booth" }
|
||||
|
||||
validates :document_number, presence: true, unless: :skip_user_verification?
|
||||
validates :user_id, uniqueness: { scope: [:poll_id], message: :has_voted }
|
||||
validates :origin, inclusion: { in: ->(*) { VALID_ORIGINS }}
|
||||
|
||||
before_validation :set_demographic_info, :set_document_info, :set_denormalized_booth_assignment_id
|
||||
|
||||
@@ -289,14 +289,10 @@ class User < ApplicationRecord
|
||||
def take_votes_from(other_user)
|
||||
return if other_user.blank?
|
||||
|
||||
with_lock do
|
||||
Poll::Voter.where(user_id: other_user.id).find_each do |poll_voter|
|
||||
if Poll::Voter.where(poll: poll_voter.poll, user_id: id).any?
|
||||
poll_voter.delete
|
||||
else
|
||||
poll_voter.update_column(:user_id, id)
|
||||
end
|
||||
end
|
||||
Poll::Voter.where(user_id: other_user.id).find_each do |poll_voter|
|
||||
transaction(requires_new: true) { poll_voter.update_column(:user_id, id) }
|
||||
rescue ActiveRecord::RecordNotUnique
|
||||
poll_voter.delete
|
||||
end
|
||||
|
||||
Budget::Ballot.where(user_id: other_user.id).update_all(user_id: id)
|
||||
|
||||
Reference in New Issue
Block a user