Изменение типа класса ActiveRecord в Rails с наследованием одной таблицы - PullRequest
Новая
       0

Изменение типа класса ActiveRecord в Rails с наследованием одной таблицы

25 голосов
/ 16 июля 2010

У меня есть два типа классов: 

BaseUser < ActiveRecord::Base

и 

User < BaseUser

, которые действует с помощью системы аутентификации Authlogic.Это наследование реализовано с использованием Single Table Inheritance

. Если регистрируется новый пользователь, я регистрирую его как пользователя.Однако, если у меня уже есть BaseUser с тем же адресом электронной почты, я бы хотел изменить этого BaseUser на User в базе данных без простого копирования всех данных пользователю из BaseUser и создания нового пользователя (то есть с новымЯ бы).Это возможно?Спасибо.

Ответы [ 3 ]

77 голосов
/ 24 декабря 2012

Ответ Стива работает, но, поскольку экземпляр имеет класс BaseUser, когда вызывается save, проверки и обратные вызовы, определенные в User, не будут выполняться.Возможно, вы захотите преобразовать экземпляр, используя метод gets : 

user = BaseUser.where(email: "user@example.com").first_or_initialize
user = user.becomes(User) # convert to instance from BaseUser to User
user.type = "User"
user.save!
18 голосов
/ 16 июля 2010

Вы можете просто установить поле типа на «Пользователь» и сохранить запись.Объект в памяти все равно будет отображаться как BaseUser, но при следующей перезагрузке объект в памяти будет пользователем 

>> b=BaseUser.new
>> b.class # = BaseUser

# Set the Type. In-Memory object is still a BaseUser
>> b.type='User'
>> b.class # = BaseUser
>> b.save

# Retrieve the records through both models (Each has the same class)

>> User.find(1).class # = User
>> BaseUser.find(1).class # User
4 голосов
/ 29 сентября 2014

Основываясь на других ответах, я ожидал, что это будет работать в Rails 4.1: 

  def update
    @company = Company.find(params[:id])
    # This separate step is required to change Single Table Inheritance types
    new_type = params[:company][:type]
    if new_type != @company.type && Company::COMPANY_TYPES.include?(new_type)
      @company.becomes!(new_type.constantize)
      @company.type = new_type
      @company.save!
    end

    @company.update(company_params)
    respond_with(@company)
  end

Это не так, поскольку изменение типа не будет сохраняться.Вместо этого я выбрал менее элегантный подход, который работает правильно: 

  def update
    @company = Company.find(params[:id])
    # This separate step is required to change Single Table Inheritance types
    new_type = params[:company][:type]
    if new_type != @company.type && Company::COMPANY_TYPES.include?(new_type)
      @company.update_column :type, new_type
    end

    @company.update(company_params)
    respond_with(@company)
  end

И вот тесты контроллера, которые я использовал для подтверждения решения: 

  describe 'Single Table Inheritance (STI)' do

    class String
      def articleize
        %w(a e i o u).include?(self[0].to_s.downcase) ? "an #{self}" : "a #{self}"
      end
    end

    Company::COMPANY_TYPES.each do |sti_type|
      it "a newly assigned Company of type #{sti_type} " \
        "should be #{sti_type.articleize}" do
        post :create, { company: attributes_for(:company, type: sti_type) },
             valid_session
        expect(assigns(:company)).to be_a(sti_type.constantize)
      end
    end

    Company::COMPANY_TYPES.each_index do |i|
      sti_type, next_sti_type = Company::COMPANY_TYPES[i - 1],
                                Company::COMPANY_TYPES[i]
      it "#{sti_type.articleize} changed to type #{next_sti_type} " \
        "should be #{next_sti_type.articleize}" do
        company = Company.create! attributes_for(:company, type: sti_type)
        put :update, { id: company.to_param, company: { type: next_sti_type } },
            valid_session
        reloaded_company = Company.find(company.to_param)
        expect(reloaded_company).to be_a(next_sti_type.constantize)
      end
    end
  end
...