In an ActiveRecord::Base
model, I can reset the state of the model to what it was when I got it from the database with reload
, as long as the attribute I'm setting maps to a table column:
user = User.first
user.email #=> "[email protected]"
user.email = "[email protected]"
user.email #=> "[email protected]"
user.reload
user.email #=> "[email protected]"
But if I add a custom attribute, the only way I've found to have it act the same is like this:
class User < ActiveRecord::Base
attr_accessor :user_agent
def reload
super
self.user_agent = nil
self
end
end
My question is, is there some API to make non-database-column-attributes reset on reload? Something like:
class User < ActiveRecord::Base
# this
reloadable_attr_accessor :user_agent
# or this
def user_agent
@user_agent
end
def user_agent=(value)
set_instance_var_that_resets_on_reload("@user_agent", value)
end
end
Does that exist in Rails somewhere?
ActiveRecord does not provide a way to do this, it can only acts on the model attributes.
That being said, I think a more elegant way to do it would be to loop over the ivars and set them to whatever you like :
class User < ActiveRecord::Base
def reload(options = nil)
super
self.instance_variables.each do |ivar|
next if ivar == '@attributes'
self.instance_variable_set(ivar, nil)
end
end
end
Note that we skip @attributes because AR is taking care of it when you reload the attributes.
Rework Jean-Do's answer slightly. It doesn't break default instance_variables and relations.
after_initialize do
@default_instance_variables = instance_variables
end
def reload(options = nil)
super
self.instance_variables.each do |ivar|
if ivar == :'@default_instance_variables' ||
@default_instance_variables.include?(ivar)
next
end
remove_instance_variable(ivar)
end
self
end
I took gayavat's answer and reworked it into my test_helper.rb
file, because I didn't want to override the usual #reload method.
class ActiveRecord::Base
attr_accessor :default_instance_variables
after_initialize do
@default_instance_variables = instance_variables
end
end
def reload_ivars(obj)
obj.reload
obj.instance_variables.each do |ivar|
if ivar == :'@default_instance_variables' ||
obj.default_instance_variables.include?(ivar)
next
end
obj.send(:remove_instance_variable, ivar)
end
end
When I need to reload something in a test I just call reload_ivars(object)
.
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With