diode 1.3.3 → 1.3.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1a336c0eda5cf34b05fe4d4fe013dc181af28163588ede63ba741c76c8e976b9
-  data.tar.gz: e84ea6a14a4b6a8682e0ce74593c23fe828bf36ac19d5072b5a534a4c44eb632
+  metadata.gz: 9ea759707873eb1e4b875b88f3592a9943bda41f0e7e2b734a02f97907a3418e
+  data.tar.gz: 57be21fb601cdb86a63b82f4bb61d1ea7950f1b308099303425cadcf16be99bb
 SHA512:
-  metadata.gz: c86e2f8c48f2c053f4f236113de844f811b42058f9d62895cb00b1db397c22ec221bc577366674a35ecc9c463ab8296c378871f8d7ea280427930035eeb2c056
-  data.tar.gz: b31be1338d45b4999c3688778aacdc7946140444a0293f30cf0c986a3747fd7105e59f687ca782c98e2b79595faa12f3c03bd436a84e74828266d98942f97548
+  metadata.gz: 3adb8476510156d01dff10b3b66130a1dc91e65710416c1a4ee07fb43513758172efca9701b876e7d6c5c87e09561842d437c5445c4a03fc352737ac3e5b9f37
+  data.tar.gz: 34dceb159914710b19af8e4683de8f361206574aabd6dfaf0fb9b29d76b23a82f7b482af2370d98c0e244823c1b1f697643d48301fd0d91b6c48b10b5307daa6

data/lib/diode/pgate.rb

@@ -0,0 +1,71 @@
+# pgate - Postgresql connection pooling
+require 'pg'
+
+module Diode
+class Pgate
+  attr_accessor(:max, :min, :connHash)
+  attr_reader(:pool, :inuse)
+
+  def initialize(connectionHash, max=10, min=1)
+    @connHash = connectionHash
+    @max = max || 10
+    @min = min || 1
+    @pool = []
+    @inuse = []
+    @running = true
+    Thread.new() {
+      curate() while @running
+    }
+  end
+
+  def assign()
+    conn = nil
+    return nil unless @running
+    if @pool.empty?
+      conn = PG::Connection.new(@connHash) if @inuse.size < @max
+      @inuse << conn
+      return(conn)
+    else
+      conn = @pool.shift()
+      return(assign()) unless ok?(conn) # recurse if bad connection
+      @inuse << conn
+      return(conn)
+    end
+  end
+
+  def release(conn) # return conn to pool if its alive, else discard it
+    @inuse.delete(conn) if @inuse.include?(conn)
+    @pool << conn if ok?(conn)
+  end
+
+  def ok?(conn)
+    return false unless conn.respond_to?(:conninfo)
+    begin
+      conn.conninfo()
+      return(true)
+    rescue PG::ConnectionBad => e
+      return(false)
+    end
+  end
+
+  def curate()
+    if @pool.size + @inuse.size > @max
+      unless @pool.empty?
+        conn = @pool.shift()
+        conn.close
+      end
+    end
+    if @pool.size + @inuse.size < @min
+      @pool << conn = PG::Connection.new(@connHash)
+    end
+    sleep(30)
+  end
+
+  def shutdown()
+    @running = false
+    (@pool + @inuse).each{|c| c.close() }
+  end
+
+end
+end
+# load 'pgate.rb'; h = {host: "127.0.0.1", port: 7000, dbname: 'admin', user: 'richard', password: 'redacted24'};  pool = Diode::Pgate.new(h)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: diode
 version: !ruby/object:Gem::Version
-  version: 1.3.3
+  version: 1.3.4
 platform: ruby
 authors:
 - Kjell Koda
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-04-20 00:00:00.000000000 Z
+date: 2024-06-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json
@@ -48,6 +48,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/diode/pgate.rb
 - lib/diode/request.rb
 - lib/diode/response.rb
 - lib/diode/server.rb