Sha256: 9784555d66cd1838a4bb27c9bc91fe33bc758dd056aac265e2b687859c19987d
Contents?: true
Size: 1.73 KB
Versions: 8
Compression:
Stored size: 1.73 KB
Contents
module Friendly
class DataStore
attr_reader :database
def initialize(database)
@database = database
end
def insert(persistable, attributes)
batch? ? batch_insert(persistable, attributes) :
immediate_insert(persistable, attributes)
end
def all(persistable, query)
filtered = dataset(persistable).where(query.conditions)
if query.limit || query.offset
filtered = filtered.limit(query.limit, query.offset)
end
filtered = filtered.order(query.order) if query.order
filtered.map
end
def first(persistable, query)
dataset(persistable).first(query.conditions)
end
def update(persistable, id, attributes)
dataset(persistable).where(:id => id).update(attributes)
end
def delete(persistable, id)
dataset(persistable).where(:id => id).delete
end
def count(persistable, query)
dataset(persistable).where(query.conditions).count
end
def start_batch
Thread.current[:friendly_batch] = Hash.new { |h, k| h[k] = [] }
end
def reset_batch
Thread.current[:friendly_batch] = nil
end
def flush_batch
batch = Thread.current[:friendly_batch]
batch.keys.each do |k|
database.from(k).multi_insert(batch[k], :commit_every => 1000)
end
reset_batch
end
protected
def dataset(persistable)
database.from(persistable.table_name)
end
def immediate_insert(persistable, attributes)
dataset(persistable).insert(attributes)
end
def batch_insert(persistable, attributes)
Thread.current[:friendly_batch][persistable.table_name] << attributes
end
def batch?
Thread.current[:friendly_batch]
end
end
end
Version data entries
8 entries across 8 versions & 2 rubygems