Reducing database access.

1 files changed, 10 insertions, 1 deletions

diff --git a/app/models/network/graph.rb b/app/models/network/graph.rb
index 0fe7765b9ea..ea7d188cc38 100644
--- a/app/models/network/graph.rb
+++ b/app/models/network/graph.rb
@@ -2,7 +2,7 @@ require "grit"
 
 module Network
   class Graph
-    attr_reader :days, :commits, :map
+    attr_reader :days, :commits, :map, :notes
 
     def self.max_count
       @max_count ||= 650
@@ -16,10 +16,19 @@ module Network
 
       @commits = collect_commits
       @days = index_commits
+      @notes = collect_notes
     end
 
     protected
 
+    def collect_notes
+      h = Hash.new(0)
+      @project.notes.where('noteable_type = ?' ,"Commit").group('notes.commit_id').select('notes.commit_id, count(notes.id) as note_count').each do |item|
+        h[item["commit_id"]] = item["note_count"]
+      end
+      h
+    end
+
     # Get commits from repository
     #
     def collect_commits