重构Rails中的红宝石方法

Question

我在Rails应用程序的通知模型中有以下代码。

我想重构一下。 肯定有问题的是“检查对象是否存在，然后对其进行处理”部分。 我想我可以try但我不知道它是如何工作的。 斯比能否告诉我该怎么做（ try或者try更好的方法）？

关于其余代码还有其他意见吗？

#check and decrease chat notification that happens between 2 given users (since 1v1 chat notification number max limit is 1)
def self.decreasing_chat_notification_number(current_user, user)
  if self.between_chat_recipient(current_user, user).unchecked.any?
    notification = self.between_chat_recipient(current_user, user).unchecked.first
    checking_and_decreasing_notification(notification)
  end
end

#check and decrease all the task notifications that happens between 2 given users
def self.decreasing_task_notification_number(current_user, user)
  if self.task.between_other_recipient(current_user, user).unchecked
    self.task.between_other_recipient(current_user, user).unchecked.each do |notification|
      checking_and_decreasing_notification(notification)
    end
  end
end

#check and decrease all the post notifications that happened on a given post
def self.decreeasing_post_notification_number(current_user, post)
  if self.post.this_post_comments(current_user, post).unchecked
    self.post.this_post_comments(current_user, post).unchecked.each do |notification|
      checking_and_decreasing_notification(notification)
    end
  end
end

private

def check_notification #chat notification gets checked
  if self.checked_at == nil
    update_attributes(checked_at: Time.zone.now)
  end
end

def checking_and_decreasing_notification(notification)
  notification.check_notification
  current_user.decrease_new_other_notifications
  current_user.decreased_other_number_pusher
end

范围：

scope :not_chat, -> { where.not(notifiable_type: "Message") }
scope :chat, -> { where(notifiable_type: "Message") }
scope :task, -> { where(notifiable_type: "Task") }
scope :post, -> { where(notifiable_type: "Post") }
scope :checked, -> { where.not(checked_at: nil) }
scope :unchecked, -> { where(checked_at: nil) }
scope :this_post_comments, -> (recipient_id, post_id) do
  where("notifications.recipient_id = ? AND notifications.notifiable_id = ?", recipient_id, post_id)
end
scope :between_chat_recipient, -> (recipient_id, sender_id) do
  where("notifications.recipient_id = ? AND notifications.sender_id = ? AND notifications.notifiable_type = ?", recipient_id, sender_id, "Message")
end
scope :between_other_recipient, -> (recipient_id, sender_id) do
  where("notifications.recipient_id = ? AND notifications.sender_id = ?", recipient_id, sender_id)
end

更新：

在current_user上，所有内容均作为实例方法而不是Notification.class_method进行调用。 顺便说一句。 用户表具有"new_chat_notification",default: 0和"new_other_notification", default: 0属性，用于计算当前通知的数量。

user.rb

  #check and decrease chat notification that happens between 2 given users (max 1)
  def decreasing_chat_notification_number(user)
    notification = self.notifications.between_chat_recipient(user).unchecked.first
    self.checking_and_decreasing_notification(notification) if notification.present?
  end

  #check and decrease task notifications that happens between 2 given users
  def decreasing_task_notification_number(user)
    self.notifications.task.between_other_recipient(user).unchecked.each do |notification|
      self.checking_and_decreasing_notification(notification)
    end
  end

  #check and decrease the post notification that belongs to a given post
  def decreasing_post_notification_number(post_id)
    self.notifications.this_post_comments(post_id).unchecked.each do |notification|
      self.checking_and_decreasing_notification(notification)
    end
  end

  def checking_and_decreasing_notification(notification)
    notification.check_notification
    if notification.notifiable_type == "Message"
      self.decrease_new_chat_notifications
      self.decreased_chat_number_pusher
    else
      self.decrease_new_other_notifications
      self.decreased_other_number_pusher
    end
  end

  def decrease_new_chat_notifications
    decrement!(:new_chat_notification) if self.new_chat_notification > 0
  end


  def decrease_new_other_notifications
    decrement!(:new_other_notification) if self.new_other_notification > 0
  end

  def decreased_chat_number_pusher
    number = self.new_chat_notification
    Pusher['private-'+ self.id.to_s].trigger('new_chat_notification', {:number => number})
  end

  def decreased_other_number_pusher
    number = self.new_other_notification
    Pusher['private-'+ self.id.to_s].trigger('new_other_notification', {:number => number})
  end

notification.rb

scope :not_chat, -> { where.not(notifiable_type: "Message") }
scope :chat, -> { where(notifiable_type: "Message") }
scope :task, -> { where(notifiable_type: "Task") }
scope :post, -> { where(notifiable_type: "Post") }
scope :checked, -> { where.not(checked_at: nil) }
scope :unchecked, -> { where(checked_at: nil) }

scope :this_post_comments, -> (post_id) do
  where("notifications.notifiable_id = ?", post_id)
end

scope :between_chat_recipient, -> (sender_id) do
  where("notifications.sender_id = ? AND notifications.notifiable_type = ?", sender_id, "Message")
end

scope :between_other_recipient, -> (sender_id) do
  where("notifications.sender_id = ? AND notifications.notifiable_type != ?", sender_id, "Message")
end

def check_notification #chat notification gets checked
  update_attribute(:checked_at, Time.zone.now) if self.checked_at.nil?
end

Answer 1

您可以使用try尝试向对象发送消息。 如果对象响应，则将执行该对象，否则将被忽略。

string = "foo"
string.try(:length) # => 3
nil.try(:length) => nil

我建议你try! 代替。 如果接收方不响应消息并且不nil ，它将提高

但这对您没有太大帮助。

如果您进行迭代，则无需检查是否存在。 您可以进行迭代，如果关联为“空”，则将不会发生任何事情。

因此，例如，您可以重写decreeasing_post_notification_number ：

def self.decreeasing_post_notification_number(current_user, post)
  self.post.this_post_comments(current_user, post).unchecked.each do |notification|
    checking_and_decreasing_notification(notification)
  end
end

对于IFF decreasing_chat_notification_number _聊天_通知_编号同样适用，可以为each调用而不是仅first checking_and_decreasing_notification _通知_ decreasing_chat_notification_number _通知

由于我们在这里看不到完整的代码，因此我必须采取一些假设。 该代码看起来不是OO（您具有将所有必需值作为参数接收的self方法）。 我建议您将其重构为：

• 使此方法成为相关模型的行为（方法）
• 提供涵盖单个业务运营的服务

顺便说一句：很酷，您使用Time.zone.now ，而不仅仅是Time.now ！

Answer 2

我会在助手内部移动支票：

def checking_and_decreasing_notification(notification)
  return if notification.nil? # here we defend from bad input

  notification.check_notification
  current_user.decrease_new_other_notifications
  current_user.decreased_other_number_pusher
end

现在可以安全地将其称为：

def self.decreasing_chat_notification_number(current_user, user)
    # first below is safe, since `unchecked` returns a relation
    #   first called on relation will return nil on empty set
    checking_and_decreasing_notification \
      self.between_chat_recipient(current_user, user).unchecked.first
  end
end

旁注 ： if的后缀形式在这里可能会更优雅：

def check_notification #chat notification gets checked
  update_attributes(checked_at: Time.zone.now) if self.checked_at.nil?
end