gotime-moneta 0.6.2

data/lib/moneta/datamapper.rb

@@ -0,0 +1,117 @@
+begin
+  gem "dm-core", "0.9.10"
+  require "dm-core"
+rescue LoadError
+  puts "You need the dm-core gem in order to use the DataMapper moneta store"
+  exit
+end
+
+class MonetaHash
+  include DataMapper::Resource
+
+  property :the_key, String, :key => true
+  property :value, Object, :lazy => false
+  property :expires, Time
+
+  def self.value(key)
+    obj = self.get(key)
+    obj && obj.value
+  end
+end
+
+module Moneta
+  class DataMapper
+    class Expiration
+      def initialize(klass, repository)
+        @klass = klass
+        @repository = repository
+      end
+
+      def [](key)
+        if obj = get(key)
+          obj.expires
+        end
+      end
+
+      def []=(key, value)
+        obj = get(key)
+        obj.expires = value
+        obj.save(@repository)
+      end
+
+      def delete(key)
+        obj = get(key)
+        obj.expires = nil
+        obj.save(@repository)
+      end
+
+      private
+      def get(key)
+        repository(@repository) { @klass.get(key) }
+      end
+    end
+
+    def initialize(options = {})
+      @repository = options.delete(:repository) || :moneta
+      ::DataMapper.setup(@repository, options[:setup])
+      repository_context { MonetaHash.auto_upgrade! }
+      @hash = MonetaHash
+      @expiration = Expiration.new(MonetaHash, @repository)
+    end
+
+    module Implementation
+      def key?(key)
+        repository_context { !!@hash.get(key) }
+      end
+
+      def has_key?(key)
+        repository_context { !!@hash.get(key) }
+      end
+
+      def [](key)
+        repository_context { @hash.value(key) }
+      end
+
+      def []=(key, value)
+        repository_context {
+          obj = @hash.get(key)
+          if obj
+            obj.update(key, value)
+          else
+            @hash.create(:the_key => key, :value => value)
+          end
+        }
+      end
+
+      def fetch(key, value = nil)
+        repository_context {
+          value ||= block_given? ? yield(key) : default
+          self[key] || value
+        }
+      end
+
+      def delete(key)
+        repository_context {
+          value = self[key]
+          @hash.all(:the_key => key).destroy!
+          value
+        }
+      end
+
+      def store(key, value, options = {})
+        repository_context { self[key] = value }
+      end
+
+      def clear
+        repository_context { @hash.all.destroy! }
+      end
+
+      private
+      def repository_context
+        repository(@repository) { yield }
+      end
+    end
+    include Implementation
+    include Expires
+  end
+end

data/lib/moneta/file.rb

@@ -0,0 +1,91 @@
+begin
+  require "xattr"
+rescue LoadError
+  puts "You need the xattr gem to use the File moneta store"
+  exit
+end
+require "fileutils"
+
+module Moneta
+  class File
+    class Expiration
+      def initialize(directory)
+        @directory = directory
+      end
+      
+      def [](key)
+        attrs = xattr(key)
+        ret = Marshal.load(attrs.get("moneta_expires"))
+      rescue Errno::ENOENT, SystemCallError
+      end
+      
+      def []=(key, value)
+        attrs = xattr(key)
+        attrs.set("moneta_expires", Marshal.dump(value))
+      end
+      
+      def delete(key)
+        attrs = xattr(key)
+        attrs.remove("moneta_expires")
+      end
+
+      private
+      def xattr(key)
+        ::Xattr.new(::File.join(@directory, key))
+      end
+    end
+    
+    def initialize(options = {})
+      @directory = options[:path]
+      if ::File.file?(@directory)
+        raise StandardError, "The path you supplied #{@directory} is a file"
+      elsif !::File.exists?(@directory)
+        FileUtils.mkdir_p(@directory)
+      end
+      
+      @expiration = Expiration.new(@directory)
+    end
+    
+    module Implementation
+      def key?(key)
+        ::File.exist?(path(key))
+      end
+      
+      alias has_key? key?
+      
+      def [](key)
+        if ::File.exist?(path(key))
+          Marshal.load(::File.read(path(key)))
+        end
+      end
+      
+      def []=(key, value)
+        ::File.open(path(key), "w") do |file|
+          contents = Marshal.dump(value)
+          file.puts(contents)
+        end
+      end
+            
+      def delete(key)
+        value = self[key]
+        FileUtils.rm(path(key))
+        value
+      rescue Errno::ENOENT
+      end
+            
+      def clear
+        FileUtils.rm_rf(@directory)
+        FileUtils.mkdir(@directory)
+      end
+      
+      private
+      def path(key)
+        ::File.join(@directory, key.to_s)
+      end
+    end
+    include Implementation
+    include Defaults
+    include Expires
+    
+  end
+end

data/lib/moneta/lmc.rb

@@ -0,0 +1,52 @@
+begin
+  require "localmemcache"
+rescue LoadError
+  puts "You need the localmemcache gem to use the LMC moneta store"
+  exit
+end
+
+module Moneta
+  class Expiration
+    def initialize(hash)
+      @hash = hash
+    end
+    
+    def [](key)         @hash["#{key}__!__expiration"]          end
+    def []=(key, value) @hash["#{key}__!__expiration"] = value  end
+      
+    def delete(key)
+      key = "#{key}__!__expiration"
+      value = @hash[key]
+      @hash.delete(key)
+      value
+    end
+  end
+  
+  class LMC
+    include Defaults
+
+    module Implementation
+      def initialize(options = {})
+        @hash = LocalMemCache.new(:filename => options[:filename])
+        @expiration = Expiration.new(@hash)
+      end    
+
+      def [](key)         @hash[key]          end
+      def []=(key, value) @hash[key] = value  end
+      def clear()         @hash.clear         end
+
+      def key?(key)
+        @hash.keys.include?(key)
+      end
+
+      def delete(key)
+        value = @hash[key]
+        @hash.delete(key)
+        value
+      end      
+    end
+    include Implementation
+    include StringExpires
+   
+  end
+end

data/lib/moneta/memcache.rb

@@ -0,0 +1,53 @@
+begin
+  require "memcached"
+  MemCache = Memcached
+rescue LoadError
+  require "memcache"
+rescue
+  puts "You need either the `memcached` or `memcache-client` gem to use the Memcache moneta store"
+  exit
+end
+
+module Moneta
+  class Memcache
+    include Defaults
+    
+    def initialize(options = {})
+      @cache = MemCache.new(options.delete(:server), options)
+    end
+
+    def key?(key)
+      !self[key].nil?
+    end
+
+    alias has_key? key?
+
+    def [](key)
+      @cache.get(key)
+    end
+
+    def []=(key, value)
+      store(key, value)
+    end
+
+    def delete(key)
+      value = self[key]
+      @cache.delete(key) if value
+      value
+    end
+
+    def store(key, value, options = {})
+      args = [key, value, options[:expires_in]].compact
+      @cache.set(*args)
+    end
+
+    def update_key(key, options = {})
+      val = self[key]
+      self.store(key, val, options)
+    end
+
+    def clear
+      @cache.flush_all
+    end
+  end
+end

data/lib/moneta/memory.rb

@@ -0,0 +1,11 @@
+module Moneta
+  class Memory < Hash
+    include Expires
+    
+    def initialize(*args)
+      @expiration = {}
+      super
+    end    
+        
+  end
+end

data/lib/moneta/mongodb.rb

@@ -0,0 +1,58 @@
+begin
+  require "mongo"
+rescue LoadError
+  puts "You need the mongo gem to use the MongoDB moneta store"
+  exit
+end
+
+module Moneta
+  class MongoDB
+    include Defaults
+    
+    def initialize(options = {})
+      options = {
+        :host => ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost',
+        :port => ENV['MONGO_RUBY_DRIVER_PORT'] || XGen::Mongo::Driver::Mongo::DEFAULT_PORT,
+        :db => 'cache',
+        :collection => 'cache'
+      }.update(options)
+      conn = XGen::Mongo::Driver::Connection.new(options[:host], options[:port])
+      @cache = conn.db(options[:db]).collection(options[:collection])
+    end
+
+    def key?(key)
+      !!self[key]
+    end
+
+    def [](key)
+      res = @cache.find_first('_id' => key)
+      res = nil if res && res['expires'] && Time.now > res['expires']
+      res && res['data']
+    end
+
+    def []=(key, value)
+      store(key, value)
+    end
+
+    def delete(key)
+      value = self[key]
+      @cache.remove('_id' => key) if value
+      value
+    end
+
+    def store(key, value, options = {})
+      exp = options[:expires_in] ? (Time.now + options[:expires_in]) : nil
+      @cache.update({ '_id' => key }, { '_id' => key, 'data' => value, 'expires' => exp })
+    end
+
+    def update_key(key, options = {})
+      val = self[key]
+      self.store(key, val, options)
+    end
+
+    def clear
+      @cache.clear
+    end
+  end
+end
+

data/lib/moneta/redis.rb

@@ -0,0 +1,67 @@
+begin
+  require "redis"
+rescue LoadError
+  puts "You need the redis gem to use the Redis store"
+  exit
+end
+
+module Moneta
+  class Redis
+    include Defaults
+
+    def initialize(options = {})
+      @cache = ::Redis.new(options)
+    end
+
+    def key?(key)
+      !@cache[key].nil?
+    end
+
+    alias has_key? key?
+
+    def [](key)
+      @cache.get(key)
+    end
+
+    def []=(key, value)
+      store(key, value)
+    end
+
+    def delete(key)
+      value = @cache[key]
+      if @cache.respond_to?(:delete)
+        @cache.delete(key) if value
+      else
+        @cache.del(key) if value
+      end
+      value
+    end
+
+    def store(key, value, options = {})
+      if options.has_key?(:expires_in)
+        if @cache.respond_to?(:setex)
+          @cache.setex(key, options[:expires_in], value)
+        elsif @cache.respond_to?(:set_with_expire)
+          @cache.set_with_expire(key, value, options[:expires_in])
+        else
+          @cache.set(key, value, options[:expires_in])
+        end
+      else
+        @cache.set(key, value)
+      end
+    end
+
+    def update_key(key, options = {})
+      val = @cache[key]
+      self.store(key, val, options)
+    end
+
+    def clear
+      if @cache.respond_to?(:flush_db)
+        @cache.flush_db
+      else
+        @cache.flushdb
+      end
+    end
+  end
+end