timkaechele/planet_express
1
0
Fork 0
Code

Optimize DiscountBudget for month lookups

Browse Source
This commit is contained in:
Tim Kächele 2021-11-20 21:47:23 +01:00
parent c6b32d66c7
commit ce4075a507
1 changed files with 4 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lib/planet_express/discounts/discount_budget_for_month_repository.rb
View File

@ -6,29 +6,16 @@ module PlanetExpress
attr_reader :data attr_reader :data
def initialize(data: {}) def initialize(data: {})
@data = data @data = Hash.new do |hash, key|
hash[key] = DiscountBudget.new(budget: DEFAULT_BUDGET)
end.merge(data)
end end
def for_month(date) def for_month(date)
return budget_for_month(date) if budget_for_month?(date)
add_budget_for_month(date)
budget_for_month(date)
end
private
def budget_for_month(date)
data[start_of_month(date)] data[start_of_month(date)]
end end
def budget_for_month?(date) private
!budget_for_month(date).nil?
end
def add_budget_for_month(date)
data[start_of_month(date)] = DiscountBudget.new(budget: DEFAULT_BUDGET)
end
def start_of_month(date) def start_of_month(date)
Date.new(date.year, date.month, 1) Date.new(date.year, date.month, 1)