自定义belongs_to属性编写器

Question

我正在处理的一个应用程序试图在不使用多态性的情况下使用多态的概念。

class User 
has_many :notes 
end 

class Customer 
has_many :notes 
end 

class Note 
belongs_to :user 
belongs_to :customer 
end 

固有我们有笔记两列：user_id和customer_id，现在这里的坏事是有可能的便条到现在已经有一个customer_id和user_id在同一时间，这是我不想要的。

我知道一个简单/更好的方法是使笔记表多态，但有一些限制，阻止我现在这样做。

我想知道是否有一些自定义的方式来覆盖这些关联，以确保分配一个时，另一个未分配。

这里是我试过的：当使用

def user_id=(id) 
    super 
    write_attribute('customer_id', nil) 
end 

def customer_id=(id) 
    super 
    write_attribute('user_id', nil) 
end 

这不起作用：

note.customer=customer or 
note.update(customer: customer) 

，但工程使用的时候：

note.update(customer_id: 12) 

基本上，我需要一个可以适用于这两种情况，而无需编写4种方法：

def user_id=(id) 

end 

def customer_id=(id) 

end 

def customer=(id) 

end 

def user=(id) 

end 

Answer 1

我宁愿使用ActiveRecord回调来实现这样的结果。

class Note 

    belongs_to :user 
    belongs_to :customer 

    before_save :correct_assignment 

    # ... your code ... 

    private 

    def correct_assignment 
    if user_changed? 
     self.customer = nil 
    elsif customer_changed? 
     self.user = nil 
    end 
    end 

end