self_association.rb 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. #---
  2. # Excerpted from "Agile Web Development with Rails",
  3. # published by The Pragmatic Bookshelf.
  4. # Copyrights apply to this code. It may not be used to create training material,
  5. # courses, books, articles, and the like. Contact us if you are in doubt.
  6. # We make no guarantees that this code is fit for any purpose.
  7. # Visit http://www.pragmaticprogrammer.com/titles/rails4 for more book information.
  8. #---
  9. $: << File.dirname(__FILE__)
  10. require "connect"
  11. require "logger"
  12. #ActiveRecord::Base.logger = Logger.new(STDERR)
  13. require "rubygems"
  14. require "active_record"
  15. ActiveRecord::Schema.define do
  16. create_table :employees, :force => true do |t|
  17. t.string :name
  18. t.integer :manager_id
  19. t.integer :mentor_id
  20. end
  21. end
  22. class Employee < ActiveRecord::Base
  23. belongs_to :manager,
  24. :class_name => "Employee",
  25. :foreign_key => "manager_id"
  26. belongs_to :mentor,
  27. :class_name => "Employee",
  28. :foreign_key => "mentor_id"
  29. has_many :mentored_employees,
  30. :class_name => "Employee",
  31. :foreign_key => "mentor_id"
  32. has_many :managed_employees,
  33. :class_name => "Employee",
  34. :foreign_key => "manager_id"
  35. end
  36. Employee.delete_all
  37. adam = Employee.create(:name => "Adam")
  38. beth = Employee.create(:name => "Beth")
  39. clem = Employee.new(:name => "Clem")
  40. clem.manager = adam
  41. clem.mentor = beth
  42. clem.save!
  43. dawn = Employee.new(:name => "Dawn")
  44. dawn.manager = adam
  45. dawn.mentor = clem
  46. dawn.save!
  47. p adam.managed_employees.map {|e| e.name} # => [ "Clem", "Dawn" ]
  48. p adam.mentored_employees # => []
  49. p dawn.mentor.name # => "Clem"