ntswf 2.0.2 → 2.0.3

data/lib/ntswf/base.rb

@@ -5,10 +5,15 @@ module Ntswf
   module Base
     # @param config [Hash] A configuration with the following keys:
     # @option config [String] :access_key_id AWS credential
-    # @option config [Hash] :activity_task_lists The task list names for activities as hash (see also *:unit*)
+    # @option config [Hash] :activity_task_lists
+    #   The task list names for activities as hash (see also *:unit*)
     # @option config [String] :decision_task_list The task list name for decisions
     # @option config [String] :domain The SWF domain name
-    # @option config [Numeric] :execution_version Value allowing clients to reject future execution versions
+    # @option config [Numeric] :execution_version
+    #   Value allowing clients to reject future execution versions
+    # @option config [String] :pidfile
+    #   A path receiving the current PID for looping methods. Causes exit, if
+    #   overwritten by another process. See {Worker#in_subprocess}
     # @option config [Numeric] :subprocess_retries (0) see {Worker#in_subprocess}
     # @option config [String] :secret_access_key AWS credential
     # @option config [String] :unit This worker/client's activity task list key

data/lib/ntswf/worker.rb

@@ -9,9 +9,12 @@ module Ntswf
 
     # Run a method in a separate process.
     # This will ensure the call lives on if the master process is terminated.
-    # If the *:subprocess_retries* configuration is set {StandardError}s during the
+    # If the *:subprocess_retries* configuration is set, {StandardError}s during the
     # method call will be retried accordingly.
+    #
+    # Exits the process if the *:pidfile* configuration is set and the PID file has been modified.
     def in_subprocess(method)
+      raise_on_pidfile_change if @config.pidfile
       $stdout.sync = true
       if child = fork
         srand
@@ -34,5 +37,22 @@ module Ntswf
       log("retrying. exception raised: #{e.message}\n  #{e.backtrace.join("\n  ")}")
       retry
     end
+
+    def raise_on_pidfile_change
+      @pid ||= create_pidfile
+      filed_pid = IO.read(@config.pidfile).strip rescue $!.message
+      if filed_pid != Process.pid.to_s
+        notify("I am a worker, and someone changed *my* PID file. I quit!",
+          pid_file_content: filed_pid,
+          pid_file: @config.pidfile,
+          process_pid: Process.pid,
+        )
+        exit
+      end
+    end
+
+    def create_pidfile
+      IO.write(@config.pidfile, Process.pid)
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ntswf
 version: !ruby/object:Gem::Version
-  version: 2.0.2
+  version: 2.0.3
   prerelease: 
 platform: ruby
 authors: