CV2-6729: N+1 query

app/models/annotations/annotation.rb

@@ -13,6 +13,25 @@ def load
     klass.where(id: self.id).last
   end
 
+  def self.load_edge_nodes(edge_nodes)
+    edge_nodes
+    .group_by { |node| self.annotation_class(node) }
+    .flat_map do |klass, nodes|
+      records = klass.where(id: nodes.map(&:id)).index_by(&:id)
+      nodes.map { |node| records[node.id] }
+    end
+  end
+
+  def self.annotation_class(node)
+    klass = nil
+    begin
+      klass = node.annotation_type.camelize.constantize
+    rescue NameError
+      klass = Dynamic
+    end
+    klass
+  end
+
   def destroy
     dec = self.disable_es_callbacks
     skip_ability = self.skip_check_ability

config/initializers/graphql_extensions.rb

@@ -26,7 +26,7 @@ def sliced_nodes
 
       def edge_nodes
         @edge_nodes ||= paged_nodes
-        @edge_nodes = @edge_nodes.map(&:load) if @field.name == 'annotations'
+        @edge_nodes = Annotation.load_edge_nodes(@edge_nodes) if @field.name == "annotations"
         @edge_nodes
       end