如何在has_one这样的关联上加入memcache 补丁

李斯特

cache_fu 好用, 不过我有很多表间关联,我希望某些表的数据,通过关联(has_one,has_many)能够按id统一从memcache中读取或者写入. 该如何呢?
唯一的办法就是 monkey patch了,自己写吧.
第一个版本
require 'active_record/associations/belongs_to_association'
require 'active_record/associations/has_many_association'
module ActiveRecord
class Base
public
def self.has_one_cached *params
has_one *params # call the standard association
assoc_name = params[0]

define_method ("#{assoc_name}_with_cache") do
assoc = self.class.reflections[assoc_name.to_sym]
instance_values[assoc_name.to_s] ||
(assoc.klass.get_cache(instance_values["#{assoc_name}_cached_id"]) if instance_values["#{assoc_name}_cached_id"] ) ||
(a = HasOneAssociation.new(self, assoc)
assoc.klass.get_cache(a.id)
self.instance_variable_set("@#{assoc_name}_cached_id", a.id)
self.instance_variable_set("@#{assoc_name}", a)
)
end
# switch out the method. has_one -> has_one_with_cache
alias_method_chain assoc_name, :cache
end
end
end


代码解释下,捡重要的说吧
1-7行,可以忽略
8 增加一个 has_one_cache的方法,使用起来等同于 has_one
17-23行的三个 "||"
第一个,就是读取AR中的缓存数据,这个没啥说的
第二个"||",是根据instance中保存的instance_values["#{assoc_name}_cached_id"]从memcache中获取对应的数据
第三个"||",就是第一次运行的时候,保存一个cache_id的变量和值,和对AR对象中缓存进行赋值
所以这里实际上是三级的缓存读取方式,先对象缓存,再memcache,如果还没有就直接读取数据库.
不过这里有个很明显的缺陷.要求instance长期存在,这个代码才能起作用.不过rails中对象的生命周期仅限于action,超出这个就没有啥用了.所以上面的代码,到头来只能在偶尔的情况下才能真正的用到memcache
解决办法是有,不过看各位是否喜欢的问题,就是把id也保存到memcache中.呵呵,很土的没有办法的办法.
于是上面的方法就改成了

define_method ("#{assoc_name}_with_cache") do
assoc = self.class.reflections[assoc_name.to_sym]
instance_values[assoc_name.to_s] ||
(assoc.klass.get_cache(instance_values["#{assoc_name}_cached_id"]) if instance_values["#{assoc_name}_cached_id"] ) ||
(id = assoc.klass.get_cache_ex("#{self.class.to_s}:#{self.id}:#{assoc_name}")
assoc.klass.get_cache(id) if id)||
(a = HasOneAssociation.new(self, assoc)
assoc.klass.get_cache(a.id)
assoc.klass.set_to_cache("#{self.class.to_s}:#{self.id}:#{assoc_name}",a.id,assoc.klass.ttl)
self.instance_variable_set("@#{assoc_name}_cached_id", a.id)
self.instance_variable_set("@#{assoc_name}", a)
)
end


增加了一个"||"
(id = assoc.klass.get_cache_ex("#{self.class.to_s}:#{self.id}:#{assoc_name}")
assoc.klass.get_cache(id) if id)

先用memcache中获取id数据,在获取对应的纪录,很囧吧.
不过要看应用场合的,如果你需要访问的数据属于基础数据,永远不变的,或者长期不变的数据,这样的办法能够减轻数据库的访问负担,但是要注意的是"这样的方法不会比直接访问数据库要快"
这里重申下memcache的作用:
"memcache不会使得你的网站访问速度更快,只是为了减轻数据库的访问负担"
另外顺便提下,cache_fu中没有get_cache_ex的函数.这个函数是在robbin的提示下加进来,专门用于非id key的memcache数据读写的.
代码如下

def get_cache_ex(key, timeout = 60 * 30)
reset = cache_reset
data = get_from_cache(key) unless reset
return data unless data.nil?
if block_given?
if data = yield
test = true
if data.instance_of? Array
test = false if data.size ==0
end
set_to_cache(key, data, timeout) if test
end
return data  
end
end


还有,我这里的代码是参考http://groups.google.com/group/acts_as_cached/browse_thread/thread/5fefb2d2355a5048/25ea72ecd0d5a0c9
这个thread写的.
如果有has_many, belongs_to 的需求,还是参考google groups上的帖子来改写吧.
最后感谢下robbin的自动保存功能,真是救命功能阿