docker_toolkit 0.1.2 → 0.1.3

data/lib/docker_toolkit/watcher.rb

@@ -0,0 +1,188 @@
+#!/usr/bin/env ruby
+require 'English'
+
+module DockerToolkit
+
+class Watcher
+  def initialize
+    @lock = Mutex.new
+
+    @procs = []
+    @code = 0
+    @crashed = false
+    @threads = []
+    @who = nil
+    @stopping = false
+  end
+
+  def stop_all
+    @stopping = true
+
+    terminating = @procs.reject do |meta|
+      meta[:handled] || meta[:process].exited?
+    end
+    
+    terminating.each do |meta|
+      ::Process.kill 'TERM', meta[:process].pid
+    end
+
+    terminating.each do |meta|
+      begin
+        meta[:process].poll_for_exit(10)
+      rescue ChildProcess::TimeoutError
+        meta[:process].stop(1)
+        meta[:stdout].close
+        meta[:stderr].close
+      end
+    end
+
+    if !@crashed
+      meta = @procs.detect do |meta|
+        meta[:process].crashed?
+      end
+
+      if meta
+        @crashed = true
+        @code = meta[:process].exit_code
+        @who = meta[:cmd]
+      end
+    end
+
+    @procs.each do |meta|
+      meta[:stdout].close rescue nil
+      meta[:stderr].close rescue nil
+    end
+
+    Thread.new do 
+      sleep 2
+      @threads.each(&:terminate)
+    end
+    
+
+  end
+
+
+  def add *cmd
+    process = ChildProcess.build(*cmd)
+
+    rerr, werr = IO.pipe
+    rout, wout = IO.pipe
+
+    process.io.stdout = wout
+    process.io.stderr = werr
+
+    meta = {
+      cmd: cmd,
+      process: process,
+      stdout: rout,
+      stderr: rerr
+    }
+
+    @threads << Thread.new(meta[:stdout], STDOUT) do |io, out|
+      loop do
+        break unless synchro_readline(io, out)
+      end
+    end
+
+    @threads << Thread.new(meta[:stderr], STDERR) do |io, out|
+      loop do
+        break unless synchro_readline(io, out)
+      end
+    end
+
+    log "Starting #{meta[:cmd]}"
+    meta[:pid] = meta[:process].start.pid
+
+    @procs.push(meta)
+    meta
+  end
+
+  def synchro_readline io, out
+    str = io.gets
+    @lock.synchronize{out.puts str}
+    true
+  rescue => e
+    false
+  end
+
+  def log msg
+    puts "[watcher]: #{msg}"
+  end
+
+  def error msg
+    STDERR.puts "[watcher]: Error: #{msg}"
+  end
+
+
+  def exec &block
+    %w[EXIT QUIT].each do |sig|
+      trap(sig) do
+        stop_all
+      end
+    end 
+
+
+    %w[INT TERM].each do |sig|
+      trap(sig) do
+        log "Catch #{sig}: try exits gracefully.."
+        stop_all
+      end
+    end 
+
+    trap("CLD") do |*args|
+      unhandled = @procs.reject do |meta|
+        meta[:handled] || !meta[:process].exited?
+      end
+
+      unhandled.any? do |meta|
+        log "Child finished"
+        log "  Process[#{meta[:pid]}]: #{meta[:cmd]}"
+        log "    status: #{meta[:process].crashed? ? 'crashed' : 'exited'}"
+        log "    code:   #{meta[:process].exit_code}"
+
+        meta[:handled] = true
+        meta[:stdout].close
+        meta[:stderr].close
+
+        if !@crashed && meta[:process].crashed?
+          @crashed = true
+          @code = meta[:process].exit_code
+          @who = meta[:cmd]
+        end
+
+        if !@stopping
+          log "Try exits gracefully.."
+          stop_all
+        end
+        
+        true
+      end 
+    end
+
+    begin
+      yield(self)
+    rescue => e
+      @crashed = true
+      @code = 1
+      error e.inspect
+      error e.backtrace.last(20).join("\n")
+      log "Try exits gracefully.."
+      stop_all
+    end
+
+    @threads.map(&:join)
+    
+    if @crashed
+      @code = [@code || 0, 1].max
+    end
+
+    exit(@code || 0)
+  end
+
+
+end
+
+end
+
+
+

data/lib/docker_toolkit.rb

@@ -2,9 +2,14 @@ require 'diplomat'
 require 'English'
 require 'optparse'
 
+
+
 require 'docker_toolkit/version'
+require 'docker_toolkit/watcher'
+require 'docker_toolkit/childprocess'
 
 module DockerToolkit
 
 
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: docker_toolkit
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Godko Ivan
@@ -74,6 +74,7 @@ email:
 executables:
 - consul.rb
 - merger.rb
+- terminator.rb
 - waiter.rb
 extensions: []
 extra_rdoc_files: []
@@ -88,6 +89,7 @@ files:
 - README.md
 - bin/consul.rb
 - bin/merger.rb
+- bin/terminator.rb
 - bin/waiter.rb
 - docker/Dockerfile
 - docker/docker-compose.yml
@@ -95,9 +97,38 @@ files:
 - docker/run_tests.sh
 - docker_toolkit.gemspec
 - lib/docker_toolkit.rb
+- lib/docker_toolkit/childprocess.rb
+- lib/docker_toolkit/childprocess/abstract_io.rb
+- lib/docker_toolkit/childprocess/abstract_process.rb
+- lib/docker_toolkit/childprocess/errors.rb
+- lib/docker_toolkit/childprocess/jruby.rb
+- lib/docker_toolkit/childprocess/jruby/io.rb
+- lib/docker_toolkit/childprocess/jruby/process.rb
+- lib/docker_toolkit/childprocess/jruby/pump.rb
+- lib/docker_toolkit/childprocess/tools/generator.rb
+- lib/docker_toolkit/childprocess/unix.rb
+- lib/docker_toolkit/childprocess/unix/fork_exec_process.rb
+- lib/docker_toolkit/childprocess/unix/io.rb
+- lib/docker_toolkit/childprocess/unix/lib.rb
+- lib/docker_toolkit/childprocess/unix/platform/i386-linux.rb
+- lib/docker_toolkit/childprocess/unix/platform/i386-solaris.rb
+- lib/docker_toolkit/childprocess/unix/platform/x86_64-linux.rb
+- lib/docker_toolkit/childprocess/unix/platform/x86_64-macosx.rb
+- lib/docker_toolkit/childprocess/unix/posix_spawn_process.rb
+- lib/docker_toolkit/childprocess/unix/process.rb
+- lib/docker_toolkit/childprocess/version.rb
+- lib/docker_toolkit/childprocess/windows.rb
+- lib/docker_toolkit/childprocess/windows/handle.rb
+- lib/docker_toolkit/childprocess/windows/io.rb
+- lib/docker_toolkit/childprocess/windows/lib.rb
+- lib/docker_toolkit/childprocess/windows/process.rb
+- lib/docker_toolkit/childprocess/windows/process_builder.rb
+- lib/docker_toolkit/childprocess/windows/structs.rb
 - lib/docker_toolkit/version.rb
+- lib/docker_toolkit/watcher.rb
 homepage: https://github.com/RnD-Soft/docker_toolkit
-licenses: []
+licenses:
+- MIT
 metadata: {}
 post_install_message: 
 rdoc_options: []