diff --git a/lib/customer_plan.rb b/lib/customer_plan.rb index c349a546bfb572ee808ef36a427874bb7032f485..f4716a3bbd27e24c25501ab59740e5f7a58ce84f 100644 --- a/lib/customer_plan.rb +++ b/lib/customer_plan.rb @@ -71,9 +71,12 @@ class CustomerPlan def bill_plan(note: nil) EM.promise_fiber do - DB.transaction do + DB.transaction do |db| + next false unless !block_given? || yield(db) + charge_for_plan(note) add_one_month_to_current_plan unless activate_plan_starting_now + true end end end