class Timelog < ActiveRecord::Base validates :time_spent, :user, presence: true validate :issuable_id_is_present belongs_to :issue belongs_to :merge_request belongs_to :user def issuable issue || merge_request end private def issuable_id_is_present if issue_id && merge_request_id errors.add(:base, 'Only Issue ID or Merge Request ID is required') elsif issuable.nil? errors.add(:base, 'Issue or Merge Request ID is required') end end end