Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Ruby on rails - Reference the same model twice?

Is it possible to set up a double relationship in activerecord models via the generate scaffold command?

For example, if I had a User model and a PrivateMessage model, the private_messages table would need to keep track of both the sender and recipient.

Obviously, for a single relationship I would just do this:

ruby script/generate scaffold pm title:string content:string user:references 

Is there a similar way to set up two relations?

Also, is there anyway to set up aliases for the relations?

So rather than saying:

@message.user 

You can use something like:

@message.sender or @message.recipient

Any advice would be greatly appreciated.

Thanks.

like image 496
Dan Avatar asked Jan 13 '10 14:01

Dan


1 Answers

Here's a complete answer to this issue, in case people visiting this question are new to Ruby on Rails and having a hard time putting everything together (as I was when I first looked into this).

Some parts of the solution take place in your Migrations and some in your Models:

Migrations

class CreatePrivateMessages < ActiveRecord::Migration   def change     create_table :private_messages do |t|       t.references :sender       t.references :recipient     end     # Rails 5+ only: add foreign keys     add_foreign_key :private_messages, :users, column: :sender_id, primary_key: :id     add_foreign_key :private_messages, :users, column: :recipient_id, primary_key: :id   end end 

Here you are specifying that there are two columns in this table that will be referred to as :sender and :recipient and which hold references to another table. Rails will actually create columns called 'sender_id' and 'recipient_id' for you. In our case they will each reference rows in the Users table, but we specify that in the models, not in the migrations.

Models

class PrivateMessage < ActiveRecord::Base   belongs_to :sender, :class_name => 'User'   belongs_to :recipient, :class_name => 'User' end 

Here you are creating a property on the PrivateMessage model named :sender, then specifying that this property is related to the User class. Rails, seeing the "belongs_to :sender", will look for a column in your database called "sender_id", which we defined above, and use that to store the foreign key. Then you're doing the exact same thing for the recipient.

This will allow you to access your Sender and Recipient, both instances of the User model, through an instance of the PrivateMessage model, like this:

@private_message.sender.name @private_message.recipient.email 

Here is your User Model:

class User < ActiveRecord::Base   has_many :sent_private_messages, :class_name => 'PrivateMessage', :foreign_key => 'sender_id'   has_many :received_private_messages, :class_name => 'PrivateMessage', :foreign_key => 'recipient_id' end 

Here you are creating a property on the User Model named :sent_private_messages, specifying that this property is related to the PrivateMessage Model, and that the foreign key on the PrivateMessage model which relates it to this property is called 'sender_id'. Then you are doing the same thing for received private messages.

This allows you to get all of a users sent or received private messages by doing something like this:

@user.sent_private_messages @user.received_private_messages 

Doing either of these will return an array of instances of the PrivateMessage model.

....

like image 143
Richard Jones Avatar answered Sep 21 '22 03:09

Richard Jones