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
|
module Moneta
module Adapters
class Sequel
# @api private
module SQLite
def self.extended(mod)
version = mod.backend.get(::Sequel[:sqlite_version].function)
# See https://sqlite.org/lang_UPSERT.html
mod.instance_variable_set(:@can_upsert, ::Gem::Version.new(version) >= ::Gem::Version.new('3.24.0'))
end
def store(key, value, options = {})
@table.insert_conflict(:replace).insert(config.key_column => key, config.value_column => blob(value))
value
end
def increment(key, amount = 1, options = {})
return super unless @can_upsert
@backend.transaction do
@increment.call(key: key, value: blob(amount.to_s), amount: amount)
Integer(load(key))
end
end
def merge!(pairs, options = {}, &block)
@backend.transaction do
pairs = yield_merge_pairs(pairs, &block) if block_given?
@table.insert_conflict(:replace).import([config.key_column, config.value_column], blob_pairs(pairs).to_a)
end
self
end
protected
def prepare_store
@store = @table
.insert_conflict(:replace)
.prepare(:insert, statement_id(:store), config.key_column => :$key, config.value_column => :$value)
end
def prepare_increment
return super unless @can_upsert
update_expr = (::Sequel[config.value_column].cast(Integer) + :$amount).cast(:blob)
@increment = @table
.insert_conflict(
target: config.key_column,
update: { config.value_column => update_expr },
update_where: ::Sequel.|({ config.value_column => blob("0") },
{ ::Sequel.~(::Sequel[config.value_column].cast(Integer)) => 0 })
)
.prepare(:insert, statement_id(:increment), config.key_column => :$key, config.value_column => :$value)
end
end
end
end
end
|
