45 lines
1.6 KiB
Ruby
45 lines
1.6 KiB
Ruby
|
#
|
||
|
|
# A wrapper around redis that namespaces keys with the current site id
|
||
|
|
#
|
||
|
|
class DiscourseRedis
|
||
|
|
|
||
|
|
def initialize
|
||
|
|
@config = YAML::load(File.open("#{Rails.root}/config/redis.yml"))[Rails.env]
|
||
|
|
redis_opts = {:host => @config['host'], :port => @config['port'], :db => @config['db']}
|
||
|
|
@redis = Redis.new(redis_opts)
|
||
|
|
end
|
||
|
|
|
||
|
|
# prefix the key with the namespace
|
||
|
|
def method_missing(meth, *args, &block)
|
||
|
|
if @redis.respond_to?(meth)
|
||
|
|
@redis.send(meth, *args, &block)
|
||
|
|
else
|
||
|
|
super
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
# Proxy key methods through, but prefix the keys with the namespace
|
||
|
|
%w(append blpop brpop brpoplpush decr decrby del exists expire expireat get getbit getrange getset hdel
|
||
|
|
hexists hget hgetall hincrby hincrbyfloat hkeys hlen hmget hmset hset hsetnx hvals incr incrby incrbyfloat
|
||
|
|
lindex linsert llen lpop lpush lpushx lrange lrem lset ltrim mget move mset msetnx persist pexpire pexpireat psetex
|
||
|
|
pttl rename renamenx rpop rpoplpush rpush rpushx sadd scard sdiff set setbit setex setnx setrange sinter
|
||
|
|
sismember smembers sort spop srandmember srem strlen sunion ttl type watch zadd zcard zcount zincrby
|
||
|
|
zrange zrangebyscore zrank zrem zremrangebyrank zremrangebyscore zrevrange zrevrangebyscore zrevrank zrangebyscore).each do |m|
|
||
|
|
class_eval %{
|
||
|
|
def #{m}(*args)
|
||
|
|
args[0] = "\#\{DiscourseRedis.namespace\}:\#\{args[0]\}"
|
||
|
|
@redis.#{m}(*args)
|
||
|
|
end
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.namespace
|
||
|
|
RailsMultisite::ConnectionManagement.current_db
|
||
|
|
end
|
||
|
|
|
||
|
|
def url
|
||
|
|
"redis://#{@config['host']}:#{@config['port']}/#{@config['db']}"
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|