Rails 7 adds optional transaction arguments to with_lock

0
44
Saeloun Logo


When it comes to concurrency control, there are specifically two mechanisms
around it – pessimistic
and optimistic locking.

Optimistic locking

The optimistic locking model is a concurrency control technique in which
multiple users are allowed to update the same record without informing the users
that others are also attempting to update the record.
The record changes are validated only when the record is committed.
If one user successfully updates the record,
the other users attempting to commit their concurrent updates are informed that a conflict exists.

An advantage of the optimistic locking model is that
it avoids the overhead of locking a record for the duration of the action.
If there are no simultaneous updates,
then this model provides fast updates.

Pessimistic locking

The pessimistic locking model prevents simultaneous updates to records.
As soon as one user starts to update a record,
a lock is placed on it.
Other users who attempt to update this record are informed
that another user has an update in progress.
The other users must wait until the first user has finished committing their changes,
thereby releasing the record lock.
Only then can another user make changes based on the previous user’s changes.

An advantage of the pessimistic locking model is that
it avoids the issue of conflict resolution by preventing conflicts.
Updates are serialized
and each subsequent update starts with
the committed record changes from the previous user.

lock!

In Rails
ActiveRecord::Locking::Pessimistic
provides support for row-level locking using the lock! method wrapped inside a transaction for example:-

If two users press the like button for an article,
at the same time then instead of the like_count of that article going up to 2,
it will only increment to 1, because both users pressed increment
from 0 to 1 at the same time.
To fix this we can use lock! wrapped inside a transaction.

  ActiveRecord::Base.transaction do
    article = Article.find("00000000-0000-0000-0000-000000000001").lock!("FOR UPDATE NOWAIT")
    article.like_count += 1
    article.save! 
  end

What the code does here is
first, it starts a database transaction.
Second, it acquires a pessimistic database lock.
Once the lock is acquired, the record is reloaded in memory,
so that the values on the record match those in the locked database row.
The lock will prevent others from reading or writing to that row
and anyone else trying to acquire a lock will have to wait for the lock to be released.

Also, we can pass various
locking strategies
to the lock! method as supported by the underlying Database
for example, we used FOR UPDATE NOWAIT on Postgres DB
what it means is other transactions that attempt UPDATE, DELETE,
or SELECT FOR UPDATE on this row will be blocked
until the current transaction ends
and suppose another transaction tries to acquire a lock on the same record
then it will result in the below error.

TRANSACTION (1.9ms)  ROLLBACK
/Users/murtazabagwala/.rvm/gems/ruby-3.0.1/gems/activerecord-6.1.4.1/lib/active_record/connection_adapters/postgresql_adapter.rb:672:in `exec_params': PG::LockNotAvailable: ERROR:  could not obtain lock on row in relation "articles" (ActiveRecord::LockWaitTimeout)
/Users/murtazabagwala/.rvm/gems/ruby-3.0.1/gems/activerecord-6.1.4.1/lib/active_record/connection_adapters/postgresql_adapter.rb:672:in `exec_params': ERROR:  could not obtain lock on row in relation "articles" (PG::LockNotAvailable)

with_lock

We can make it more concise using with_lock.
with_lock does the same thing it creates a transaction
and applies a lock on the record under the hood.

  article = Article.find("00000000-0000-0000-0000-000000000001")
  article.with_lock("FOR UPDATE NOWAIT") do
    article.like_count += 1
    article.save! 
  end

But, before Rails 7
there was no way to specify transaction arguments like
isolation, requires_new and joinable to with_lock
so, for example before Rails 7
if we had to create the nested transactions
we had to use multiple transaction blocks with lock!.

Before

  ActiveRecord::Base.transaction do
    article = Article.find("00000000-0000-0000-0000-000000000001").lock!("FOR UPDATE NOWAIT")
    article.like_count += 1
    article.save! 

    ActiveRecord::Base.transaction(requires_new: true) do
      author = article.author.lock!("FOR UPDATE NOWAIT")
      author.articles_liked += 1
      author.save! 
    end
  end

After

After Rails 7 we can use with_lock to create nested transactions
and can specify transaction optional arguments like:-

  article = Article.find("00000000-0000-0000-0000-000000000001")

  article.with_lock do
    author = article.author
    article.like_count += 1
    article.save! 

    # Will create a new transaction called sub-transaction/savepoint.
    # And, in case of error it will rollback to this savepoint and will not rollback parent transaction

    author.with_lock("FOR UPDATE NOWAIT", requires_new: true) do
      author.articles_liked += 1
      author.save! 
    end
  end

Please refer to this PR.



Source link

Leave a reply

Please enter your comment!
Please enter your name here