in_memory_cache 0.0.0 → 1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7550d96162651e202ede0fd0b8672a4439e8eca791c9a9b8aef3f178f19514fe
-  data.tar.gz: cf16e84e42cc8d3edd7b6142222bfd69c663c7a3fdee0ca1ff392b077f9f032a
+  metadata.gz: 74663c00729180e3f82f1d8f6d4988eb8c02d8485e0cb1d80093d8178ad72d70
+  data.tar.gz: db98b4e0bd4c1172e5011896b6a177af5de6c253e4d48d6deccab05ff202eb9f
 SHA512:
-  metadata.gz: cb0341fdba9f8f9e8477ff777bca46a6d33b8728b0554dfc36f187f22c1db3ef2f37141812a3085dfc7895a49049a7ea6f1c277444f327e2fb8f4962f249fb82
-  data.tar.gz: 6068b434a0f8b85d687809d3e4ec1a4ae4eaaf56cb48bac95df853bd6f80391f8ac3b6f3bfea3f4743a8ab4ee5a9179393d96be54182371b534ec124e9e72408
+  metadata.gz: d732a622f11a188fe57085eb0261055551db626e53dd7647d07353b3d13bfc08c27b20bc134b58322f436728c6c1d35ac74866a2ff230be285357096ba6d4956
+  data.tar.gz: a5b15cee0e4f91d2e8c45c22eaa683c5157911c7f009f15efcc556bffac279d555ad736ec6b727eaf029c3c7243582fb3d82fec01b3bef4bbf53de4247465d55

data/lib/memory_cache/memory_cache_store.rb

@@ -1,10 +1,18 @@
 module MemoryCache
   # A processor level in memory cache storage. Data is not across multiple processes. Thread-safe
   class MemoryCacheStore
+
+    MEGABYTE = 1024 * 1024
+
     def initialize
       @data = {}
       @key_access_store = {}
       @lock = Monitor.new
+      @cache_size = 0
+      @max_size = 25 * MEGABYTE
+      @buffer_size = 5 * MEGABYTE
+      @max_wait_loop_for_flush = 100
+      @max_single_value_limit = 2 * MEGABYTE
     end
 
 
@@ -12,8 +20,21 @@ module MemoryCache
     # return true if success, false if any exception
     def write(key, value, options = {})
       synchronize_block do
+        return false unless can_allocate_value?(value)
+        prune_keys(@buffer_size) if is_buffer_limit_reached?
+
+        old_value_present = false
+
+        if key_exists?(key)
+          old_value_present = true
+          old_value = @data[key]
+          old_value_size = compute_value_size(old_value)
+        end
+
         @data[key] = value
         @key_access_store[key] = Time.now.utc.to_i
+        @cache_size -= old_value_size if old_value_present
+        @cache_size += compute_value_size(value)
         return true
       end
     end
@@ -23,14 +44,17 @@ module MemoryCache
     def read(key, options = {})
       synchronize_block do
         @key_access_store[key] = Time.now.utc.to_i
-        return @data[key] ? @data[key].dup : nil
+        return @data[key] ? dup_value(@data[key]) : nil
       end
     end
 
      # delete key from store
     def delete(key, options = {})
+      old_key_present = key_exists?(key)
+      value = @data[key]
       @data.delete(key)
       @key_access_store.delete(key)
+      @cache_size -= compute_value_size(value) if @cache_size > 0 && old_key_present
     end
 
     # Synchronize calls to the cache for thread safety
@@ -46,5 +70,46 @@ module MemoryCache
       end
     end
 
+    def key_exists?(key)
+      @data.key?(key)
+    end
+
+    private
+
+    def prune_keys(new_allocate_size = 0)
+      keys = synchronize_block { @key_access_store.keys.sort { |a, b| @key_access_store[a] <=> @key_access_store[b] } }
+      loop_counter = 0
+      keys.each do |key|
+        delete(key)
+        return if available_size > new_allocate_size
+        if loop_counter >= @max_wait_loop_for_flush
+          clear
+          return
+        end
+        loop_counter += 1
+      end
+    end
+
+    def compute_value_size(value = '')
+      value.to_s.bytesize
+    end
+
+    def available_size
+      size = @max_size - @cache_size
+      size < 0 ? 0 : size
+    end
+
+    def is_buffer_limit_reached?
+      available_size <= @buffer_size
+    end
+
+    def can_allocate_value?(value = '')
+      value.to_s.bytesize < @max_single_value_limit
+    end
+
+    def dup_value(value)
+      value.is_a?(String) ? value.dup : value
+    end
+
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: in_memory_cache
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: '1.1'
 platform: ruby
 authors:
 - kathir