自定義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