vx-consumer 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 610088408ab12fdbc3f8254828174a7b99beb24d
-  data.tar.gz: 50aecdec6d9e6dcf43d7ce495615d31c02b0e9dd
+  metadata.gz: 47b8bcadacba1121ca2956db329217f76ce4e9b6
+  data.tar.gz: 679bba4a1158a235190ac403ef9919d6e198cf44
 SHA512:
-  metadata.gz: 22051318d00559a878bf3c4b28d9fb9427f2c06f9687b0882ac1bfa6fc978b60bed45e9c23538f85da7b8ac53d1593810aaac6b6c8ddb807ff9f7db1ef4522dd
-  data.tar.gz: 423cad5f9a2f17061e64f6c3faeb4431bc5c37cf76951d163c51659bab33592b2edbdfee94fb20e9dfeabe86b335b0910ba846ad14d5747bdc3282e958278616
+  metadata.gz: 2bed198aba1934ca5a6d459a63c438e94193412b145e18b9dbce5217ceebaeeb179d70fc1b2fc2ba6ea8b7ca0dba2e3f846f1cefd57965cf68079177477326eb
+  data.tar.gz: a7d709c2fd93a2b10a5e081b9297580a81e76571a5b61f62496a20af82c1f973505c9486e5a4481b73cec318d19d0845b7ad4d43cd1527746cb0081547709631

data/lib/vx/consumer/subscriber.rb

@@ -19,6 +19,20 @@ module Vx
         in_progress { cancel }
       end
 
+      def try_graceful_shutdown
+        if @lock.try_lock
+          begin
+            instrument('graceful_shutdown_consumer', consumer: vx_consumer_name)
+            cancel
+          ensure
+            @lock.unlock
+          end
+          true
+        else
+          false
+        end
+      end
+
       def in_progress
         @lock.synchronize do
           yield

data/lib/vx/consumer/version.rb

@@ -1,5 +1,5 @@
 module Vx
   module Consumer
-    VERSION = "0.1.5"
+    VERSION = "0.1.6"
   end
 end

data/spec/lib/consumer_spec.rb

@@ -110,7 +110,7 @@ describe Vx::Consumer do
     expect(Bob._collected.map(&:values).flatten.sort).to eq [1,2]
   end
 
-  it "should work with graceful shutdown" do
+  it "should work with graceful_shutdown" do
     Bob.timeout = 1
 
     consumer = Bob.subscribe
@@ -126,6 +126,26 @@ describe Vx::Consumer do
     expect(Bob._collected).to_not be_empty
   end
 
+  it "should work with try_graceful_shutdown" do
+    Bob.timeout = 1
+
+    consumer = Bob.subscribe
+    1.times do |n|
+      Bob.publish a: n
+    end
+
+    sleep 0.1
+    expect(consumer.try_graceful_shutdown).to be_false
+
+    Timeout.timeout(2) do
+      while consumer.running?
+        sleep 0.1
+      end
+    end
+
+    expect(consumer.try_graceful_shutdown).to be_true
+  end
+
   it "running? should be true when consumer process task" do
     Bob.timeout = 1
     consumer = Bob.subscribe

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vx-consumer
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Dmitry Galinsky
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-06-18 00:00:00.000000000 Z
+date: 2014-06-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bunny