Sha256: 5473c7c1eaf975bf3a1e4ce52f4b64fa4568726c2be163697146b08e7192867c
Contents?: true
Size: 1.29 KB
Versions: 8
Compression:
Stored size: 1.29 KB
Contents
require 'much-plugin'
module Ardb
module UseDbDefault
include MuchPlugin
plugin_included do
extend ClassMethods
include InstanceMethods
@ardb_use_db_default_attrs = []
around_create :ardb_allow_db_to_default_attrs
end
module ClassMethods
def ardb_use_db_default_attrs
@ardb_use_db_default_attrs
end
def use_db_default(*attrs)
@ardb_use_db_default_attrs += attrs.map(&:to_s)
@ardb_use_db_default_attrs.uniq!
end
end
module InstanceMethods
private
def ardb_allow_db_to_default_attrs
# this allows the attr to be defaulted by the DB, this keeps
# activerecord from adding the attr into the sql `INSERT`, which will
# make the DB default its value
unchanged_names = self.class.ardb_use_db_default_attrs.reject do |name|
self.send("#{name}_changed?")
end
unchanged_names.each{ |name| @attributes.delete(name) }
yield
# we have to go and fetch the attr value from the DB, otherwise
# activerecord doesn't know the value that the DB used
scope = self.class.where(:id => self.id)
unchanged_names.each do |name|
@attributes[name] = scope.pluck(name).first
end
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems