litestack 0.1.5 → 0.1.7

data/lib/litestack/litesupport.rb

@@ -1,4 +1,5 @@
 require 'sqlite3'
+require 'hiredis'
 
 module Litesupport
 
@@ -11,6 +12,11 @@ module Litesupport
     @env ||= detect_environment
   end
   
+  def self.max_contexts
+    return 50 if environment == :fiber || environment == :polyphony
+    5    
+  end
+    
   # identify which environment we are running in
   # we currently support :fiber, :polyphony, :iodine & :threaded
   # in the future we might want to expand to other environments
@@ -33,6 +39,18 @@ module Litesupport
     # we should never reach here
   end
   
+  def self.detect_context
+    if environment == :fiber || environment == :poylphony
+      Fiber.current.storage
+    else
+      Thread.current
+    end
+  end
+  
+  def self.context
+    @ctx ||= detect_context
+  end
+  
   # switch the execution context to allow others to run
   def self.switch
     if self.environment == :fiber
@@ -43,7 +61,7 @@ module Litesupport
       Thread.current.switch_fiber
       true
     else
-      # do nothing in case of thread, switching will auto-happen
+      #Thread.pass
       false
     end   
   end
@@ -71,7 +89,64 @@ module Litesupport
     db = SQLite3::Database.new(path)
     db.busy_handler{ switch || sleep(0.001) }
     db.journal_mode = "WAL"
+    db.instance_variable_set(:@stmts, {})
+    class << db
+      attr_reader :stmts
+    end
     db
   end
   
+  class Mutex
+  
+    def initialize
+      @mutex = Thread::Mutex.new
+    end
+    
+    def synchronize(&block)
+      if Litesupport.environment == :threaded || Litesupport.environment == :iodine
+        @mutex.synchronize{ block.call }
+      else
+        block.call
+      end
+    end
+  
+  end
+  
+  class Pool
+  
+    def initialize(count, &block)
+      @count = count
+      @block = block
+      @resources = []
+      @mutex = Litesupport::Mutex.new
+      @count.times do
+        resource = @mutex.synchronize{ block.call }
+        @resources << [resource, :free]
+      end
+    end
+    
+    def acquire
+      acquired = false
+      result = nil
+      while !acquired do
+        @mutex.synchronize do
+          if resource = @resources.find{|r| r[1] == :free }
+            resource[1] = :busy
+            begin
+              result = yield resource[0]
+            rescue Exception => e
+              raise e
+            ensure
+              resource[1] = :free
+              acquired = true
+            end
+          end
+        end
+        sleep 0.001 unless acquired
+      end
+      result
+    end
+    
+  end
+  
 end  

data/lib/litestack/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Litestack
-  VERSION = "0.1.5"
+  VERSION = "0.1.7"
 end