unicorn-wrangler 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1caba80d14d2ba78c0141579ee032bc2ac1f19d7
+  data.tar.gz: b3ee0aa920698aa8225e2d5795e71240b64490b4
+SHA512:
+  metadata.gz: 06a6ac39aa156fc2dded7cfd895fff7870a6ecdc98dbf8170df767c455a735e6f654b60ec5796e2302fabc9b8420047a90935578d6b2be711b1741b84c14d50e
+  data.tar.gz: d60b7bb9fdb697cf39d778deeb9749ce78e61c84f4982db231354a975680f528531a126e81791937e4af934852cfe829f884176b2c2a623e6c551ce4660a420a

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in unicorn-wrangler.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Tom Ward
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,7 @@
+# Unicorn::Wrangler
+
+A simple process to launch and relaunch unicorn.  When launched, it either starts a new unicorn if none is running (found using the given pidfile), or restarts the existing unicorn if one is found.
+
+Usage:
+
+unicorn-wrangler --pidfile /path/to/unicorn.pid --startup-time 60 -- /path/to/unicorn <unicorn-options>

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/unicorn-wrangler

@@ -0,0 +1,22 @@
+#!/usr/bin/env ruby
+
+require 'unicorn/wrangler'
+require 'optparse'
+
+options = {}
+
+command_index = ARGV.index("--")
+command = ARGV[command_index + 1..-1].join(" ")
+
+OptionParser.new do |opts|
+  opts.on("-p", "--pidfile PIDFILE", "Path to the main unicorn PIDFILE") do |pidfile|
+    options[:pidfile] = pidfile
+  end
+
+  opts.on("-s", "--startup-time STARTUP", "Time to wait for server to start") do |time|
+    options[:startup] = time.to_i
+  end
+end.parse!(ARGV[0...command_index])
+
+launcher = Unicorn::Wrangler.new command, options
+launcher.start

data/lib/unicorn/wrangler.rb

@@ -0,0 +1,63 @@
+require "unicorn/wrangler/version"
+
+module Unicorn
+  class Wrangler
+    attr_reader :command, :pidfile, :startup_period, :tick_period, :verbose
+
+    def initialize(command, options = {})
+      @command = command
+      @pidfile = File.expand_path(options[:pidfile] || 'unicorn.pid')
+      @startup_period = options[:startup] || 60
+    end
+
+    def start
+      trap_signals(:QUIT, :INT, :TERM) { exit }
+
+      if unicorn_running?
+        restart_unicorn
+      else
+        Process.spawn(command)
+        wait_for { unicorn_running? }
+      end
+
+      loop do
+        exit unless unicorn_running?
+        sleep 1
+      end
+    end
+
+    private
+
+    def trap_signals(*signals, &block)
+      signals.map(&:to_s).each do |signal|
+        trap(signal) do
+          block.call signal
+        end
+      end
+    end
+
+    def restart_unicorn
+      original_pid = unicorn_pid
+      Process.kill "USR2", unicorn_pid
+      wait_for { unicorn_pid != original_pid }
+      sleep startup_period
+      Process.kill "TERM", original_pid
+    end
+
+    def unicorn_pid
+      File.exist?(pidfile) && File.read(pidfile).to_i
+    end
+
+    def unicorn_running?
+      unicorn_pid && Process.getpgid(unicorn_pid)
+    rescue Errno::ESRCH
+      false
+    end
+
+    def wait_for(&block)
+      until block.call
+        sleep 0.1
+      end
+    end
+  end
+end

data/lib/unicorn/wrangler/version.rb

@@ -0,0 +1,5 @@
+module Unicorn
+  class Wrangler
+    VERSION = "0.0.1"
+  end
+end

data/spec/lib/unicorn/wrangler_spec.rb

@@ -0,0 +1,129 @@
+require 'curl'
+
+describe Unicorn::Wrangler do
+  def wrangler_path
+    Bundler.bin_path + 'unicorn-wrangler'
+  end
+
+  def unicorn_path
+    Bundler.bin_path + 'unicorn'
+  end
+
+  def unicorn_cmd
+    "#{unicorn_path} -p 7516 -c spec/support/unicorn.conf.rb spec/support/config.ru"
+  end
+
+  def start_unicorn
+    Process.spawn unicorn_cmd
+    sleep 1
+  end
+
+  def start_wrangler
+    @wrangler_pid = Process.spawn "#{wrangler_path} --startup-time 1 -p spec/support/unicorn.pid -- #{unicorn_cmd}"
+    sleep 1
+  end
+
+  def unicorn_pid
+    File.exist?('spec/support/unicorn.pid') && File.read('spec/support/unicorn.pid').to_i
+  end
+
+  def unicorn_running?
+    unicorn_pid && Process.getpgid(unicorn_pid)
+  rescue Errno::ESRCH
+    false
+  end
+
+  def wrangler_running?
+    @wrangler_pid && Process.getpgid(@wrangler_pid)
+  rescue Errno::ESRCH
+    false
+  end
+
+  def cleanup_processes
+    `kill -TERM #{unicorn_pid}` if unicorn_pid
+    `kill -TERM #{@wrangler_pid}` if @wrangler_pid
+    sleep 1
+  end
+
+  def perform_request
+    Curl::Easy.perform('http://localhost:7516').body_str
+  end
+
+  context 'in general' do
+    before :each do
+      start_wrangler
+    end
+
+    it 'quits (but leaves unicorn running) on receiving QUIT signal' do
+      Process.kill 'QUIT', @wrangler_pid
+      sleep 1
+      wrangler_running?.should_not be_true
+      unicorn_running?.should be_true
+    end
+
+    it 'quits (but leaves unicorn running) on receiving TERM signal' do
+      Process.kill 'TERM', @wrangler_pid
+      sleep 1
+      wrangler_running?.should_not be_true
+      unicorn_running?.should be_true
+    end
+
+    it 'quits (but leaves unicorn running) on receiving INT signal' do
+      Process.kill 'INT', @wrangler_pid
+      sleep 1
+      wrangler_running?.should_not be_true
+      unicorn_running?.should be_true
+    end
+
+    it 'quits if monitored unicorn process quits' do
+      Process.kill 'TERM', unicorn_pid
+      sleep 3
+      wrangler_running?.should_not be_true
+    end
+
+    it 'continues running after launch (so that process monitoring tools do not restart it)' do
+      sleep 3
+      wrangler_running?.should be_true
+    end
+  end
+
+  context 'when unicorn is not running' do
+    it 'launches unicorn on startup' do
+      start_wrangler
+      unicorn_running?.should be_true
+    end
+  end
+
+  context 'when unicorn is already running' do
+    before :each do
+      start_unicorn
+    end
+
+    it 'restarts unicorn on startup' do
+      original_pid = unicorn_pid
+      start_wrangler
+      sleep 3
+      unicorn_pid.should_not eql(original_pid)
+    end
+
+    it 'continues serving requests from both unicorns startup period' do
+      start_wrangler
+      100.times.map { perform_request }.uniq.size.should eql(2)
+    end
+
+    it 'stops serving requests from original unicorn' do
+      original_response = perform_request
+      start_wrangler
+      sleep 3
+      perform_request.should_not eql(original_response)
+    end
+  end
+
+  before :each do
+    cleanup_processes
+  end
+
+  after :each do
+    cleanup_processes
+  end
+end

data/spec/support/config.ru

@@ -0,0 +1,4 @@
+# A very simple server that always responds with the time it was started.  This
+# is used in tests to check whether the server has started or restarted
+time = Time.now.to_f
+run lambda { |env| [200, {"Content-Type" => "text/plain"}, ["#{time}"]] }

data/spec/support/unicorn.conf.rb

@@ -0,0 +1 @@
+pid File.expand_path("../unicorn.pid", __FILE__)

data/unicorn-wrangler.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'unicorn/wrangler/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "unicorn-wrangler"
+  spec.version       = Unicorn::Wrangler::VERSION
+  spec.authors       = ["Tom Ward"]
+  spec.email         = ["tom@popdog.net"]
+  spec.description   = %q{Wrangles unicorns}
+  spec.summary       = %q{Helps manage unicorn processes}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "unicorn"
+  spec.add_development_dependency "rspec"
+  spec.add_development_dependency "curb"
+end

metadata

@@ -0,0 +1,130 @@
+--- !ruby/object:Gem::Specification
+name: unicorn-wrangler
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Tom Ward
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-06-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: unicorn
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: curb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Wrangles unicorns
+email:
+- tom@popdog.net
+executables:
+- unicorn-wrangler
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/unicorn-wrangler
+- lib/unicorn/wrangler.rb
+- lib/unicorn/wrangler/version.rb
+- spec/lib/unicorn/wrangler_spec.rb
+- spec/support/config.ru
+- spec/support/unicorn.conf.rb
+- unicorn-wrangler.gemspec
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.2
+signing_key: 
+specification_version: 4
+summary: Helps manage unicorn processes
+test_files:
+- spec/lib/unicorn/wrangler_spec.rb
+- spec/support/config.ru
+- spec/support/unicorn.conf.rb