Use pluck where posible

dodona-edu · Oct 24, 2024 · 2e997f8 · 2e997f8
1 parent 40633be
commit 2e997f8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/app/controllers/saved_annotations_controller.rb b/app/controllers/saved_annotations_controller.rb
@@ -42,9 +42,9 @@ def new
     authorize SavedAnnotation
     @annotations = AnnotationPolicy::Scope.new(current_user, Annotation.all).resolve
     @annotations = @annotations.where(saved_annotation_id: nil).where(user_id: current_user.id)
-    submissions = Submission.where(id: @annotations.map(&:submission_id).uniq)
-    @courses = Course.where(id: submissions.map(&:course_id).uniq)
-    @exercises = Activity.where(id: submissions.map(&:exercise_id).uniq)
+    submissions = Submission.where(id: @annotations.pluck(:submission_id).uniq)
+    @courses = Course.where(id: submissions.pluck(:course_id).uniq)
+    @exercises = Activity.where(id: submissions.pluck(:exercise_id).uniq)
     @annotations = apply_scopes(@annotations.order_by_created_at(:DESC))
                    .includes(:course).includes(:user).includes(:submission)
                    .paginate(page: parse_pagination_param(params[:page]), per_page: parse_pagination_param(params[:per_page]))