/
optimistic_locking.rb
83 lines (73 loc) · 2.59 KB
/
optimistic_locking.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
module Sequel
require 'plugins/instance_filters'
module Plugins
# This plugin implements a simple database-independent locking mechanism
# to ensure that concurrent updates do not override changes. This is
# best implemented by a code example:
#
# class Person < Sequel::Model
# plugin :optimistic_locking
# end
# p1 = Person[1]
# p2 = Person[1]
# p1.update(:name=>'Jim') # works
# p2.update(:name=>'Bob') # raises Sequel::Plugins::OptimisticLocking::Error
#
# In order for this plugin to work, you need to make sure that the database
# table has a lock_version column (or other column you name via the lock_column
# class level accessor) that defaults to 0.
#
# This plugin relies on the instance_filters plugin.
# You must have require_modification on the class and instances
# that use this plugin.
module OptimisticLocking
# Exception class raised when trying to update or destroy a stale object.
Error = InstanceFilters::Error
# Load the instance_filters plugin into the model.
def self.apply(model, opts={})
model.plugin :instance_filters
end
# Set the lock_column to the :lock_column option, or :lock_version if
# that option is not given.
def self.configure(model, opts={})
model.lock_column = opts[:lock_column] || :lock_version
end
module ClassMethods
# The column holding the version of the lock
attr_accessor :lock_column
# Copy the lock_column value into the subclass
def inherited(subclass)
super
subclass.lock_column = lock_column
end
end
module InstanceMethods
# Add the lock column instance filter to the object before destroying it.
def before_destroy
lock_column_instance_filter
super
end
# Add the lock column instance filter to the object before updating it.
def before_update
lock_column_instance_filter
super
end
private
# Add the lock column instance filter to the object.
def lock_column_instance_filter
lc = model.lock_column
instance_filter(lc=>send(lc))
end
# Only update the row if it has the same lock version, and increment the
# lock version.
def _update(columns)
lc = model.lock_column
lcv = send(lc)
columns[lc] = lcv + 1
super
send("#{lc}=", lcv + 1)
end
end
end
end
end