socialforge/app/models/forge_message.rb

class ForgeMessage < ActiveRecord::Base
  # 公共表中活动类型,命名规则:TYPE_OF_{类名}_ACT
  TYPE_OF_ISSUE_ACT = "Issue"
  TYPE_OF_MESSAGE_ACT = "Message"
  TYPE_OF_ATTACHMENT_ACT = "Attachment"
  TYPE_OF_DOCUMENT_ACT = "Document"
  TYPE_OF_JOURNAL_ACT = "Journal"
  TYPE_OF_WIKI_ACT = "Wiki"
  TYPE_OF_NEWS_ACT = "News"

  attr_accessible :forge_message_id, :forge_message_type, :project_id, :user_id, :viewed, :secret_key

  belongs_to :forge_message ,:polymorphic => true
  belongs_to :project
  belongs_to :user
  has_many :message_alls, :class_name => 'MessageAll',:as =>:message, :dependent => :destroy

  validates :user_id,presence: true
  validates :project_id,presence: true
  validates :forge_message_id,presence: true
  validates :forge_message_type, presence: true
  after_create :add_user_message

  def add_user_message
    if MessageAll.where("message_type = '#{self.class.to_s}' and message_id = '#{self.id}'").first.nil?
      self.message_alls << MessageAll.new(:user_id => self.user_id)
    end
  end
end