ohm models with zset

ohm models with zset

module Ohm
  class Model

    def self.zset(name, model = nil)
      attr_zset_reader(name, model)
      collections << name unless collections.include?(name)
    end

    def self.attr_zset_reader(name, model)
      define_method(name) do
        instance_variable_get("@#{name}") ||
                instance_variable_set("@#{name}", Attributes::ZSet.new(db, key(name), model))
      end
    end
  end
end

module Ohm
  module Attributes
    class ZSet < Collection

      def zadd(score, value)
        db.zadd(key, score, value.id)
      end

      def delete(value)
        db.zrem(key, value)
      end

      def find(min, max)
        db.zrangebyscore(key, min, max).each.collect do |key|
          model[key]
        end
      end
    end

  end
end