85 lines
2.5 KiB
Ruby
85 lines
2.5 KiB
Ruby
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
|