pgtk 0.16.3 → 0.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f86b9a969a2e41e480e4596035128c5ac1f260eeaa4dfb0625a88316d5c34f5d
-  data.tar.gz: d6b9dfe1dd5717f2884ba63d72b87696ddc0331983527dbb530f74a469961af6
+  metadata.gz: 8a1ac7d09cdc700a6fa51f239d1f14564f14f9cee56773a933e7e7d310291b1a
+  data.tar.gz: 8bcce91182b85e2a592cd3031048b9687b56c7be0587eda199b1c97df944f630
 SHA512:
-  metadata.gz: 7f7433c8b02fae500929e2e62929469eae1511d958fb85d8d7a6ca56fe9010467bac3759b6f77155ec4e4d86f92b532e8f7c5bf57c83f4c6a7ce99caebc63b9d
-  data.tar.gz: ef3a0849da8ac6678a4957868618b6c6350c86fc255285e49efd6ed3cc8dbc5503765a54abed9387d60a90e0d1c1571dd5f35e4f2daad4e8c9b5ada8e141b1d5
+  metadata.gz: '018f2c1cc65ca5b64f27609e4934f47163dcd8c4e582508e995dd332992b18f652de8fef63a9f5d86af969ce306796d2e2db0f490072933ab7a38cd91d02cfa0'
+  data.tar.gz: 16c2631220d3cdad2aef2132213a79c922aae79edad0d34b9b7003bd5eec62977cbf9f3febb20a99b17612e77c074e64b7305078903b76e156e27bb90bfbb69c

data/lib/pgtk/impatient.rb

@@ -59,9 +59,11 @@ class Pgtk::Impatient
   #
   # @param [Pgtk::Pool] pool The pool to decorate
   # @param [Integer] timeout Timeout in seconds for each SQL query
-  def initialize(pool, timeout = 1)
+  # @param [Array<Regex>] off List of regex to exclude queries from checking
+  def initialize(pool, timeout, *off)
     @pool = pool
     @timeout = timeout
+    @off = off
   end
 
   # Get the version of PostgreSQL server.
@@ -78,6 +80,7 @@ class Pgtk::Impatient
   # @return [Array] Result rows
   # @raise [Timeout::Error] If the query takes too long
   def exec(sql, *args)
+    return @pool.exec(sql, *args) if @off.any? { |re| re.match?(sql) }
     start = Time.now
     begin
       Timeout.timeout(@timeout) do

data/lib/pgtk/version.rb

@@ -11,5 +11,5 @@ require_relative '../pgtk'
 # License:: MIT
 module Pgtk
   # Current version of the library.
-  VERSION = '0.16.3'
+  VERSION = '0.17.0'
 end

data/test/test_impatient.rb

@@ -20,7 +20,7 @@ require_relative '../lib/pgtk/impatient'
 class TestImpatient < Pgtk::Test
   def test_takes_version
     fake_pool do |pool|
-      v = Pgtk::Impatient.new(pool).version
+      v = Pgtk::Impatient.new(pool, 1).version
       refute_nil(v)
     end
   end
@@ -28,16 +28,24 @@ class TestImpatient < Pgtk::Test
   def test_interrupts
     fake_pool do |pool|
       assert_raises(Pgtk::Impatient::TooSlow) do
-        Pgtk::Impatient.new(pool, 0.1).exec(
-          'SELECT COUNT(*) FROM generate_series(1, 10000000) AS a'
+        Pgtk::Impatient.new(pool, 0.01).exec(
+          'SELECT COUNT(*) FROM generate_series(1, 1000000) AS a'
         )
       end
     end
   end
 
+  def test_skips_by_regex
+    fake_pool do |pool|
+      Pgtk::Impatient.new(pool, 0.01, /^SELECT.*$/).exec(
+        'SELECT COUNT(*) FROM generate_series(1, 1000000) AS a'
+      )
+    end
+  end
+
   def test_doesnt_interrupt
     fake_pool do |pool|
-      id = Pgtk::Impatient.new(pool).exec(
+      id = Pgtk::Impatient.new(pool, 1).exec(
         'INSERT INTO book (title) VALUES ($1) RETURNING id',
         ['1984']
       ).first['id'].to_i
@@ -47,7 +55,7 @@ class TestImpatient < Pgtk::Test
 
   def test_doesnt_interrupt_in_transaction
     fake_pool do |pool|
-      Pgtk::Impatient.new(pool).transaction do |t|
+      Pgtk::Impatient.new(pool, 1).transaction do |t|
         id = t.exec(
           'INSERT INTO book (title) VALUES ($1) RETURNING id',
           ['1984']

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pgtk
 version: !ruby/object:Gem::Version
-  version: 0.16.3
+  version: 0.17.0
 platform: ruby
 authors:
 - Yegor Bugayenko