diff --git a/app/views/users/calendar.html.haml b/app/views/users/calendar.html.haml index 12446a209f8a37b68c9dee33c21ae127cfd78a32..488f49267c7526cea44b9be010fd64c850d68ee1 100644 --- a/app/views/users/calendar.html.haml +++ b/app/views/users/calendar.html.haml @@ -1,4 +1,7 @@ -%h4 Contributions calendar +%h4 + Contributions calendar + .pull-right + %small Issues, merge requests and push events #cal-heatmap.calendar :javascript new calendar( diff --git a/features/steps/user.rb b/features/steps/user.rb index e6086bfc1c97b373218165c713a626f033d22f80..5939c28a000facd2111fdea180d4f94766952519 100644 --- a/features/steps/user.rb +++ b/features/steps/user.rb @@ -35,7 +35,7 @@ class Spinach::Features::User < Spinach::FeatureSteps step 'I should see contributions calendar' do within '.calendar' do - page.should have_css('.graph-rect.r2.q2') + page.should have_css('.graph-rect.r3.q3') end end diff --git a/lib/gitlab/contributions_calendar.rb b/lib/gitlab/contributions_calendar.rb index ea41644811f80f23fd47e55ca1137612d060736f..79e0a514f9eeb60602677b0c09678a8281c7bf9a 100644 --- a/lib/gitlab/contributions_calendar.rb +++ b/lib/gitlab/contributions_calendar.rb @@ -15,25 +15,20 @@ module Gitlab date_to = Date.today events = Event.contributions.where(author_id: user.id). - where("created_at > ?", date_from).where(project_id: projects) + where("created_at > ?", date_from).where(project_id: projects). + group('date(created_at)'). + select('date(created_at), count(id) as total_amount'). + reorder(nil).map(&:attributes) - grouped_events = events.to_a.group_by { |event| event.created_at.to_date.to_s } dates = (1.year.ago.to_date..(Date.today + 1.day)).to_a dates.each do |date| date_id = date.to_time.to_i.to_s @timestamps[date_id] = 0 + day_events = events.find { |day_events| day_events["date"] == date } - if grouped_events.has_key?(date.to_s) - grouped_events[date.to_s].each do |event| - if event.created_at.to_date == date - if event.issue? || event.merge_request? - @timestamps[date_id] += 1 - elsif event.push? - @timestamps[date_id] += event.commits_count - end - end - end + if day_events + @timestamps[date_id] = day_events["total_amount"] end end