Replace couchbase counters with drb version
This commit is contained in:
@@ -0,0 +1,84 @@
|
||||
class Supplier
|
||||
module Counters
|
||||
extend ActiveSupport::Concern
|
||||
|
||||
def reset_counters!
|
||||
spec = Order.by_supplier_id_and_state(startkey: [id], endkey: [id, {}], reduce: true, group_level: 2)
|
||||
self.class.reset_order_counters_with_spec spec
|
||||
end
|
||||
|
||||
def increment_orders_in_process_count!
|
||||
Qwaiter::Counter.incr orders_in_process_counter_key
|
||||
end
|
||||
|
||||
def increment_orders_placed_count!
|
||||
Qwaiter::Counter.incr orders_placed_counter_key
|
||||
end
|
||||
|
||||
def decrement_orders_in_process_count!
|
||||
Qwaiter::Counter.decr orders_in_process_counter_key
|
||||
end
|
||||
|
||||
def decrement_orders_placed_count!
|
||||
Qwaiter::Counter.decr orders_placed_counter_key
|
||||
end
|
||||
|
||||
def orders_in_process_count
|
||||
Qwaiter::Counter.get orders_in_process_counter_key
|
||||
end
|
||||
|
||||
def orders_placed_count
|
||||
Qwaiter::Counter.get orders_placed_counter_key
|
||||
end
|
||||
|
||||
|
||||
def active_order_count
|
||||
orders_in_process_count + orders_placed_count
|
||||
end
|
||||
|
||||
def orders_in_process_counter_key
|
||||
self.class.orders_in_process_counter_key(id)
|
||||
end
|
||||
|
||||
def orders_placed_counter_key
|
||||
self.class.orders_placed_counter_key(id)
|
||||
end
|
||||
|
||||
module ClassMethods
|
||||
|
||||
def orders_in_process_counter_key(id)
|
||||
"supplier_counter:#{id}:orders_in_process"
|
||||
end
|
||||
|
||||
def orders_placed_counter_key(id)
|
||||
"supplier_counter:#{id}:orders_placed"
|
||||
end
|
||||
|
||||
def reset_counters!
|
||||
# Set all known counters to zero
|
||||
Qwaiter::Couchbase.flush_counters!
|
||||
|
||||
spec = Order.by_supplier_id_and_state(reduce: true, group_level: 2)
|
||||
reset_order_counters_with_spec spec
|
||||
end
|
||||
|
||||
def reset_order_counters_with_spec(spec)
|
||||
# taken from the couch_potato source since we want a model/custom mix here (hmmmm, something that should be fixed in the couchbase version)
|
||||
results = CouchPotato::View::ViewQuery.new(
|
||||
database.couchrest_database,
|
||||
spec.design_document,
|
||||
{spec.view_name => { map: spec.map_function, reduce: spec.reduce_function} },
|
||||
({spec.list_name => spec.list_function} unless spec.list_name.nil?),
|
||||
spec.language
|
||||
).query_view!(spec.view_parameters)
|
||||
Array.wrap(results['rows']).each do |result|
|
||||
supplier_id, state = result['key']
|
||||
case state
|
||||
when 'placed' then Qwaiter::Counter.set orders_placed_counter_key(supplier_id), result['value']
|
||||
when 'active' then Qwaiter::Counter.set orders_in_process_counter_key(supplier_id), result['value']
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
Reference in New Issue
Block a user