有条件地设置默认值

Question

我试图弄清楚如何根据某些条件设置字段的默认值。

现在，无论房间中有多少人，默认的room_type字段都为double 。

基本上，我的目标是：

def set_room_defaults
  if people_in_room.size == 1
    set room_type = "Single"
  elsif people_in_room.size == 2
    set room_type = "Double"
  elsif people_in_room.size == 3
    set room_type = "Triple"
  end
end

我一直在研究default_value_for gem，希望可以做以下事情：

default_value_for :room_type, (call set_room_default)

我已经读过一点， after_initialize可能与我要寻找的接近？

这样的事情可能吗？ 还是我要解决所有这些错误？

谢谢

Answer 1

从API文档

查找器发现并实例化的每个对象都会触发after_find和after_initialize回调，在实例化新对象之后也会触发after_initialize。

您可以按照以下方式进行操作：

class Room < ApplicationRecord
  after_initialize :set_room_defaults

  def set_room_defaults
    if people_in_room.size == 1
        set room_type = "Single"
    elsif people_in_room.size == 2
        set room_type = "Double"
    elsif people_in_room.size == 3
        set room_type = "Triple"
    end
  end
end

或者您可以将映射存储在Array ：

class Room < ApplicationRecord
  ROOM_TYPES = %w(Single Double Triple).freeze

  after_initialize :set_room_defaults

  def set_room_defaults
    set room_type = ROOM_TYPES[people_in_room.size - 1]
  end
end