CODEWITHSUNDEEP
ruby-on-railsruby-on-rails-4relational-databasemodels
Derk153
Derk153

Reputation: 682

Rails 4. Two foreign keys in one model

I have two models:

#Account migration

create_table :accounts do |t|
  t.string :email
  t.string :password

  t.timestamps
end

#ProjectInvitations migration

create_table :project_invitations do |t|t
  t.integer :sender_account_id #belongs_to :account
  t.string :recipient_first_name
  t.string :recipient_last_name
  t.string :recipient_email_string
  t.integer :recipient_account_id #belongs_to :account
  t.string :status

  t.timestamps
end

How can I add relations to model Account through foreign keys sender_account_id and recipient_account_id?

Upvotes: 2

Views: 203

Answers (2)

Derk153
Derk153

Reputation: 682

This code works good for me:

class ProjectInvitation < ActiveRecord::Base

  belongs_to :sender,    :class_name => Account, :foreign_key => :sender_account_id
  belongs_to :recipient, :class_name => Account, :foreign_key => :recipient_account_id
end

class Account < ActiveRecord::Base
  has_many: :sent_invitations,     class_name: ProjectInvitation, foreign_key: :sender_account_id
  has_many: :received_invitations, class_name: ProjectInvitation, foreign_key: :recipient_account_id
end

Console testing logs:

2.1.2 :001 > a = Account.create email:"test"
   (0.3ms)  SAVEPOINT active_record_1
  SQL (0.9ms)  INSERT INTO "accounts" ("created_at", "email", "id", "updated_at") VALUES ($1, $2, $3, $4) RETURNING "id"  [["created_at", "2014-07-31 08:34:26.767247"], ["email", "test"], ["id", "7d73c3c0-1e59-4988-a589-ee56223be88b"], ["updated_at", "2014-07-31 08:34:26.767247"]]
   (0.2ms)  RELEASE SAVEPOINT active_record_1
 => #<Account id: #<UUID:0x3ffb32085120 UUID:7d73c3c0-1e59-4988-a589-ee56223be88b>, user_id: nil, email: "test", password: nil, created_at: "2014-07-31 08:34:26", updated_at: "2014-07-31 08:34:26">
2.1.2 :002 > i = ProjectInvitation.create status:'pending'
   (0.2ms)  SAVEPOINT active_record_1
  SQL (0.3ms)  INSERT INTO "project_invitations" ("created_at", "id", "status", "updated_at") VALUES ($1, $2, $3, $4) RETURNING "id"  [["created_at", "2014-07-31 08:35:02.956242"], ["id", "7bbfe6b5-fffa-46e0-b92c-a2d0d7aa0469"], ["status", "pending"], ["updated_at", "2014-07-31 08:35:02.956242"]]
   (0.1ms)  RELEASE SAVEPOINT active_record_1
 => #<ProjectInvitation id: #<UUID:0x3ffb31fab678 UUID:7bbfe6b5-fffa-46e0-b92c-a2d0d7aa0469>, project_id: nil, sender_account_id: nil, recipient_first_name: nil, recipient_last_name: nil, recipient_email_string: nil, recipient_account_id: nil, status: "pending", created_at: "2014-07-31 08:35:02", updated_at: "2014-07-31 08:35:02">
2.1.2 :003 > i.sender = a
 => #<Account id: #<UUID:0x3ffb32085120 UUID:7d73c3c0-1e59-4988-a589-ee56223be88b>, user_id: nil, email: "test", password: nil, created_at: "2014-07-31 08:34:26", updated_at: "2014-07-31 08:34:26">
2.1.2 :004 > i.save
   (0.3ms)  SAVEPOINT active_record_1
  SQL (3.1ms)  UPDATE "project_invitations" SET "sender_account_id" = $1, "updated_at" = $2 WHERE "project_invitations"."id" = '7bbfe6b5-fffa-46e0-b92c-a2d0d7aa0469'  [["sender_account_id", "7d73c3c0-1e59-4988-a589-ee56223be88b"], ["updated_at", "2014-07-31 08:35:30.298393"]]
   (0.2ms)  RELEASE SAVEPOINT active_record_1
 => true
2.1.2 :005 > a.sent_invitations
  ProjectInvitation Load (0.3ms)  SELECT "project_invitations".* FROM "project_invitations"  WHERE "project_invitations"."sender_account_id" = $1  [["sender_account_id", "7d73c3c0-1e59-4988-a589-ee56223be88b"]]
 => #<ActiveRecord::Associations::CollectionProxy [#<ProjectInvitation id: #<UUID:0x3ffb3371dc84 UUID:7bbfe6b5-fffa-46e0-b92c-a2d0d7aa0469>, project_id: nil, sender_account_id: #<UUID:0x3ffb3371d5cc UUID:7d73c3c0-1e59-4988-a589-ee56223be88b>, recipient_first_name: nil, recipient_last_name: nil, recipient_email_string: nil, recipient_account_id: nil, status: "pending", created_at: "2014-07-31 08:35:02", updated_at: "2014-07-31 08:35:30">]>

Upvotes: 0

Alex Tatarnikov
Alex Tatarnikov

Reputation: 353

class ProjectInvitation < ActiveRecord::Base

  belongs_to :sender,    class_name: Account, foreign_key: :sender_account_id
  belongs_to :recipient, class_name: Account, foreign_key: :recipient_account_id
end

class Account < ActiveRecord::Base
  has_many: :sent_invitations,     class_name: ProjectInvitation, foreign_key: :sender_account_id
  has_many: :received_invitations, class_name: ProjectInvitation, foreign_key: :recipient_account_id
end

Upvotes: 3

Related Questions