62 lines
1.7 KiB
Ruby
62 lines
1.7 KiB
Ruby
class Keystore < ActiveRecord::Base
|
|
validates_presence_of :key
|
|
|
|
attr_accessible nil
|
|
|
|
def self.get(key)
|
|
Keystore.find_by_key(key)
|
|
end
|
|
|
|
def self.put(key, value)
|
|
if Rails.env == "test"
|
|
Keystore.connection.execute("INSERT OR REPLACE INTO " <<
|
|
"#{Keystore.table_name} (`key`, `value`) VALUES " <<
|
|
"(#{q(key)}, #{q(value)})")
|
|
else
|
|
Keystore.connection.execute("INSERT INTO #{Keystore.table_name} (" +
|
|
"`key`, `value`) VALUES (#{q(key)}, #{q(value)}) ON DUPLICATE KEY " +
|
|
"UPDATE `value` = #{q(value)}")
|
|
end
|
|
|
|
true
|
|
end
|
|
|
|
def self.increment_value_for(key, amount = 1)
|
|
self.incremented_value_for(key, amount)
|
|
end
|
|
|
|
def self.incremented_value_for(key, amount = 1)
|
|
new_value = nil
|
|
|
|
if Rails.env == "test"
|
|
# Tries to do a new insert, if it's not new, it doesn't insert anything.
|
|
# Then update to increment by amount (it starts at 0).
|
|
Keystore.connection.execute("INSERT OR IGNORE INTO " <<
|
|
"#{Keystore.table_name} (`key`, `value`) VALUES " <<
|
|
"(#{q(key)}, 0)")
|
|
Keystore.connection.execute("UPDATE #{Keystore.table_name} " <<
|
|
"SET `value` = `value` + #{q(amount)} " <<
|
|
"WHERE `key` = #{q(key)}");
|
|
|
|
else
|
|
Keystore.connection.execute("INSERT INTO #{Keystore.table_name} (" +
|
|
"`key`, `value`) VALUES (#{q(key)}, #{q(amount)}) ON DUPLICATE KEY " +
|
|
"UPDATE `value` = `value` + #{q(amount)}")
|
|
end
|
|
|
|
return self.value_for(key)
|
|
end
|
|
|
|
def self.decrement_value_for(key, amount = -1)
|
|
self.increment_value_for(key, amount)
|
|
end
|
|
|
|
def self.decremented_value_for(key, amount = -1)
|
|
self.incremented_value_for(key, amount)
|
|
end
|
|
|
|
def self.value_for(key)
|
|
self.get(key).try(:value)
|
|
end
|
|
end
|