课程讨论区优化

app/controllers/boards_controller.rb

@@ -70,31 +70,16 @@ class BoardsController < ApplicationController
   def show
     # 讨论区消息状态更新（已读和未读）
     if @project
-      query_forge_messages = @board.messages
-      query_forge_messages.each do |query_forge_message|
-        query = query_forge_message.forge_messages
-        query.each do |forge_message|
-          if User.current.id == forge_message.user_id
-            forge_message.update_attributes(:viewed => true)
-          end
-        end
-      end
+      ForgeMessage.where("user_id =? and project_id =? and viewed =?", User.current.id, @project.id, 0).update_all(:viewed => true)
       # 更新@消息为已读
-      @project.boards.each do |board|
-        board.messages.each do |m|
-          User.current.at_messages.unviewed('Message', m.id).each {|x| x.viewed!}
-        end
-      end
+      # 注释掉这句是因为和上面的方法代码重复，一个类型的消息，已经更新就不需要再更新
+      # @project.boards.each do |board|
+      #   board.messages.each do |m|
+      #     User.current.at_messages.unviewed('Message', m.id).each {|x| x.viewed!}
+      #   end
+      # end
     elsif @course
-      query_course_messages = @board.messages
-      query_course_messages.each do |query_course_message|
-        query = query_course_message.course_messages
-        query.each do |course_message|
-          if User.current.id == course_message.user_id
-            course_message.update_attributes(:viewed => true)
-          end
-        end
-      end
+      CourseMessage.where("user_id =? and course_id =? and viewed =?", User.current.id, @course.id, 0).update_all(:viewed => true)
     end
 
     sort_init 'updated_on', 'desc'