将方法添加到Rails ActiveRecord类

时间:2010-12-20 20:45:15

标签: ruby-on-rails ruby migration

在纯Ruby中,这很好用:

class TestSuper
  def foo
    puts "In TestSuper.foo"
  end
end

class TestClass < TestSuper
  def foo
    super
    puts "In TestClass.bar"
  end
end

class TestClass
  def bar
    puts "In TestClass.bar, second definition"
    puts "Calling foo:"
    foo
  end
end

t = TestClass.new
t.foo
t.bar

我可以在TestClass实例上调用foo()和bar()并获得我期望的结果:

In TestSuper.foo
In TestClass.bar
In TestClass.bar, second definition
Calling foo:
In TestSuper.foo
In TestClass.bar

但是,当我在Rails迁移中尝试非常类似的东西时,我会收到错误:

#### my_model.rb ####
puts "In my_model.rb"
class MyModel
  has_many :foo
end

#### my_migration.rb ####
puts "In my_migration.rb"
class MyModel
  def bar
    foo.each{ |f| f.baz }
  end
end

class MyMigration < ActiveRecord::Migration
  def self.up
    MyModel.find(1).bar        
  end

  def self.down
    # Not applicable
  end
end

第一个问题是MyModel.find()消失,除非我明确地在my_migration.rb中扩展了ActiveRecord。否则,它会丢弃超类。

如果我这样做,那么我在MyModel.bar()的foo调用中会出错。

如果我在my_migration.rb中注释掉类(重新)定义,则find()和bar()都可以正常工作。

在调试过程中,我添加了puts语句,以查看每个文件和文件的时间。上课正在执行中。如果MyModel已经定义(我正在my_migration.rb中进行),似乎my_model.rb甚至都没有加载。

那么:为什么会在Rails中发生这种情况,我该如何解决它呢?

3 个答案:

答案 0 :(得分:4)

理论#2:在迁移的顶端

require 'app/models/my_model'

答案 1 :(得分:2)

您需要首先为模型引用一个常量,这将加载您的模型类。然后你可以改变那个班级。以下是做同样事情的3种方法:

MyModel
class MyModel
  def bar
    foo.each{ |f| f.baz }
  end
end
MyModel.module_eval do
  def bar
    foo.each{ |f| f.baz }
  end
end
MyModel.send :include, Module.new {
  def bar
    foo.each{ |f| f.baz }
  end
}

答案 2 :(得分:1)

在你的例子中,我认为你错过了

< ActiveRecord::Base

来自你的第一个定义。

在您的迁移猴子补丁中,您还需要匹配

< ActiveRecord::Base

两个类定义必须匹配