protobuf-nats 0.4.1 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4e8cb63868c843fc5f1117ed16798efe91ab8b9f
-  data.tar.gz: d8deaefee3c498815d6a5d70339c2e9a05768d22
+  metadata.gz: 1a5657826a433d857f0b08fa99f314df065af9be
+  data.tar.gz: 0f4a0ec4a8c239e595251c2894f6e7d7bca8419a
 SHA512:
-  metadata.gz: 8559ded4b5fbbfc35a785b3c22f007db3a36e99725f40e1a9d1ec9a895a4169d2113568651303184deec2aec340e22be442b59872d127d03f1c7d8afcc858343
-  data.tar.gz: e97e4ac6a56e0bf258112790d9d8bcf27b89b2b34ce373b26e06696fc772be9dd93fbd725372146ed644d919b2628ad8d577784f8e7591585038c9876e6a1e97
+  metadata.gz: 541b8c84ffe66f575447b69c84198ed3eedebd996879fa9dc75b3ac210ed899dc526626d7a31f0a526ec68a8407f6a19d69248e6fb7dd4845ef2fc5d41e706a9
+  data.tar.gz: 34e7f770187a6d14d3d88aa4637688c8a5c4fd48e6fcf9dbf7f1a635a3fe0c9e739ecabcb6ccee6c8f0db34a188f58632cff4cf56b06a5fa6459bcbe84393214

data/README.md

@@ -27,6 +27,9 @@ You can also use the following environment variables to tune parameters:
 
 `PB_NATS_SERVER_MAX_QUEUE_SIZE` - The size of the queue in front of your thread pool (default: thread count passed to CLI).
 
+`PB_NATS_SERVER_PAUSE_FILE_PATH` - If this file exists, the server will pause by unsubscribing all services. When the
+file is removed it will resubscribe and restart slow start (default: `nil`).
+
 `PB_NATS_SERVER_SLOW_START_DELAY` - Seconds to wait before adding another round of subscriptions (default 10).
 
 `PB_NATS_SERVER_SUBSCRIPTIONS_PER_RPC_ENDPOINT` - Number of subscriptions to create for each rpc endpoint. This number is

data/lib/protobuf/nats/server.rb

@@ -13,6 +13,7 @@ module Protobuf
 
       def initialize(options)
         @options = options
+        @processing_requests = true
         @running = true
         @stopped = false
 
@@ -59,6 +60,10 @@ module Protobuf
         was_enqueued
       end
 
+      def pause_file_path
+        ::ENV.fetch("PB_NATS_SERVER_PAUSE_FILE_PATH", nil)
+      end
+
       def print_subscription_keys
         logger.info "Creating subscriptions:"
 
@@ -67,7 +72,7 @@ module Protobuf
         end
       end
 
-      def subscribe_to_services
+      def subscribe_to_services_once
         with_each_subscription_key do |subscription_key_and_queue|
           subscriptions << nats.subscribe(subscription_key_and_queue, :queue => subscription_key_and_queue) do |request_data, reply_id, _subject|
             unless enqueue_request(request_data, reply_id)
@@ -100,15 +105,36 @@ module Protobuf
         # We have (X - 1) here because we always subscribe at least once.
         (subscriptions_per_rpc_endpoint - 1).times do
           next unless @running
+          next if paused?
           completed += 1
           sleep slow_start_delay
-          subscribe_to_services
+          subscribe_to_services_once
           logger.info "Slow start adding another round of subscriptions (#{completed}/#{subscriptions_per_rpc_endpoint})..."
         end
 
         logger.info "Slow start finished."
       end
 
+      def detect_and_handle_a_pause
+        case
+        # If we are taking requests and detect a pause file, then unsubscribe.
+        when @processing_requests && paused?
+          @processing_requests = false
+          logger.warn("Pausing server!")
+          unsubscribe
+
+        # If we were paused and the pause file is no longer present, then subscribe again.
+        when !@processing_requests && !paused?
+          logger.warn("Resuming server: resubscribing to all services and restarting slow start!")
+          @processing_requests = true
+          subscribe
+        end
+      end
+
+      def paused?
+        !pause_file_path.nil? && ::File.exist?(pause_file_path)
+      end
+
       def run
         nats.on_reconnect do
           logger.warn "Server NATS connection was reconnected"
@@ -127,19 +153,19 @@ module Protobuf
         end
 
         print_subscription_keys
-        subscribe_to_services
-        yield if block_given?
-        finish_slow_start
+        if paused?
+          yield if block_given?
+        else
+          subscribe { yield if block_given? }
+        end
 
         loop do
           break unless @running
+          detect_and_handle_a_pause
           sleep 1
         end
 
-        logger.info "Unsubscribing from rpc routes..."
-        subscriptions.each do |subscription_id|
-          nats.unsubscribe(subscription_id)
-        end
+        unsubscribe
 
         logger.info "Waiting up to 60 seconds for the thread pool to finish shutting down..."
         thread_pool.shutdown
@@ -155,6 +181,19 @@ module Protobuf
       def stop
         @running = false
       end
+
+      def subscribe
+        subscribe_to_services_once
+        yield if block_given?
+        finish_slow_start
+      end
+
+      def unsubscribe
+        logger.info "Unsubscribing from rpc routes..."
+        subscriptions.each do |subscription_id|
+          nats.unsubscribe(subscription_id)
+        end
+      end
     end
   end
 end

data/lib/protobuf/nats/version.rb

@@ -1,5 +1,5 @@
 module Protobuf
   module Nats
-    VERSION = "0.4.1"
+    VERSION = "0.5.0"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: protobuf-nats
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.5.0
 platform: ruby
 authors:
 - Brandon Dewitt
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-05-23 00:00:00.000000000 Z
+date: 2017-05-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: protobuf