uc 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 5a3884c1ef367b6d4e8a60f72d8b50a27060e41d
+  data.tar.gz: bf05917d1e72924de65d6ec0fddc720cdbfc5ae4
+SHA512:
+  metadata.gz: 4f120d3edaf991f4a799fa4130d76c7df56f7f3406e17c0155be8775f9c2fd78b34e606efa00b44f003bb1f1f83519f008542373bb9d78dae6da49d93a93ffeb
+  data.tar.gz: 538bf7be85f3052d4ad570aa65124e7e5c9d4b8092caf1a6263df6a03a6c79b084c62b2ff1a1118e19bad1b935fdea3cc47f0e726a85e8934cf0bd41c1893251

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 uc.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Neeraj
+
+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,29 @@
+# Uc
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'uc'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install uc
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it ( http://github.com/<my-github-username>/uc/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

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

data/bin/uc

@@ -0,0 +1,51 @@
+#!/usr/bin/env ruby
+lib = File.expand_path(File.dirname(__FILE__) + '/../lib')
+$LOAD_PATH.unshift(lib) if File.directory?(lib) && !$LOAD_PATH.include?(lib)
+
+require 'optparse'
+require 'uc/error'
+require 'uc/server'
+
+options = {}
+opts_parser = OptionParser.new do |opts| 
+    
+    banner = [] 
+    banner << "Usage: uc [options] command"
+    banner << "Available commands: "
+    banner << " start stop restart status "
+    banner << " rolling-restart "
+
+    opts.banner = banner
+
+    opts.on("-d", "--app-dir [APP DIR]" , "Set app dir") do |d| 
+      path = Pathname.new(d)
+      raise ::Uc::Error, "app_dir path must be absolute" if path.relative?
+      options[:app_dir] = path
+    end 
+
+    opts.on_tail("-h", "--help", "Show this message") do 
+      puts  opts
+    end 
+end
+
+begin
+  opts_parser.parse!(ARGV)
+
+  app_dir = options[:app_dir] || Dir.pwd
+  server = ::Uc::Server.new(app_dir)
+  command = ARGV.shift
+
+  case command
+  when "start", "stop", "restart", "status"
+    server.send command.to_sym
+  when "rr", "rolling-restart"
+    server.rolling_restart ARGV.shift
+  else
+    puts "Invalid command"
+    abort
+  end
+
+rescue OptionParser::InvalidOption, OptionParser::MissingArgument, ::Uc::Error => e
+  puts "ERROR #{e.message}"
+  abort
+end

data/lib/uc/error.rb

@@ -0,0 +1,4 @@
+module Uc
+  class Error < StandardError
+  end
+end

data/lib/uc/logger.rb

@@ -0,0 +1,21 @@
+require 'logger'
+module Uc
+  module Logger
+   
+    def self.logger
+      @logger ||= begin
+        logger = ::Logger.new(STDOUT)
+        logger.formatter = proc do |severity, datetime, progname, msg|
+          "#{msg}\n"
+        end
+        logger.level = ::Logger::INFO
+        logger
+      end
+    end 
+
+    def logger
+      ::Uc::Logger.logger
+    end
+
+  end
+end

data/lib/uc/mqueue.rb

@@ -0,0 +1,70 @@
+require 'posix_mq'
+require 'uc/logger'
+module Uc
+  class Mqueue
+    include ::Uc::Logger
+ 
+    attr_reader :name 
+
+    def initialize(name)
+      @name = name
+    end
+
+    def setup
+      attr = ::POSIX_MQ::Attr.new(0,100,100)       # o_readonly, maxmsg, msgsize
+      puts name
+      ::POSIX_MQ.new("/#{name}", :rw, 0700, attr)
+      make_empty
+    end
+
+    def reader
+      @reader ||= ::POSIX_MQ.new("/#{name}", :r)
+    end
+
+    def nb_reader
+      mq = ::POSIX_MQ.new("/#{name}", :r)
+      mq.nonblock = true
+      return mq
+    end
+
+    def writer
+      POSIX_MQ.new("/#{name}", IO::WRONLY)
+    end
+
+    def nb_writer
+      writer = POSIX_MQ.new("/#{name}", IO::WRONLY)
+      writer.nonblock = true
+      return writer
+    end
+
+    def watch(loglevel: :info, msg: "success", err_msg: "error", &block)
+      setup
+      make_empty
+      yield
+      wait_for_fin(msg, err_msg, loglevel)
+    end
+
+    def wait_for_fin(msg, err_msg, loglevel)
+      message = ""
+      timeout = 30
+      while message != "fin"
+        reader.receive(message, timeout)
+        logger.send(loglevel, message) if message != "fin"
+      end
+      logger.info msg
+    rescue Errno::ETIMEDOUT
+      logger.info "#{timeout}s timeout reached while waiting for message"
+      raise ::Uc::Error, err_msg
+    end
+
+    def make_empty
+      mq = nb_reader
+      while true do
+        mq.receive
+      end
+    rescue Errno::EAGAIN
+      return
+    end
+
+  end
+end

data/lib/uc/server.rb

@@ -0,0 +1,114 @@
+require 'uc/logger'
+require 'uc/shell_helper'
+require 'uc/mqueue'
+require 'uc/unicorn'
+require 'uc/unicorn_config'
+require 'uc/error'
+
+module Uc
+  class Server
+
+    include ::Uc::ShellHelper
+    include ::Uc::Logger
+
+    attr_reader :uconfig
+
+    def initialize(app_dir)
+      @uconfig = ::Uc::UnicornConfig.new(app_dir)
+    end
+
+    def app_env(&block)
+      uconfig.dirs_checked? || uconfig.check_dirs
+      yield if block
+    end
+
+    def start 
+      app_env do
+        if server_running?
+          logger.info "unicorn already running pid #{pid}"
+          return
+        end
+
+        cmd %{unicorn -c #{uconfig.config_path} -D}, return_output: false,
+          error_msg: "error starting unicorn"
+      end
+    end
+
+    def stop
+      app_env do
+        if not server_running?
+          puts "unicorn not running"
+          return
+        end
+
+        kill(pid, 30)
+      end
+    end
+
+    def kill(pid, timeout)
+      Process.kill(:TERM, pid)
+      logger.debug "TERM signal sent to #{pid}"
+      (1..timeout).each do
+        if not process_running? pid
+          logger.info "Stopped #{pid}"
+          return
+        end
+        sleep 1
+      end
+      Process.kill(9, pid)
+      sleep 1
+      logger.info "Killed #{pid}"
+    end
+
+    def status
+      status = ( server_running? ? "Running pid #{pid}" : "Stopped")  
+      puts status
+    end
+
+    def restart
+      stop
+      start
+    end
+
+    def rolling_restart(queue = nil)
+      app_env
+      if not server_running?
+        start
+        return
+      end
+      queue ||= get_queue_name
+      raise ::Uc::Error, "argument missing mq name"  if (queue.nil? || queue.empty?)
+      mq = ::Uc::Mqueue.new(queue)
+      mq.watch err_msg: "server may not have restarted successfully" do
+        Process.kill("USR2", pid)
+      end
+    end
+
+    def process_running?(pid)
+      return false if pid <= 0
+      Process.getpgid pid
+      return true
+    rescue Errno::ESRCH
+        return false
+    end
+
+    def server_running?
+      process_running? pid
+    end
+
+    def get_queue_name
+      Dir.chdir uconfig.app_dir do
+        return ::Uc::Unicorn.get_queue_name
+      end
+    end
+
+    def read_pid
+      uconfig.read_pid
+    end
+
+    def pid
+      @pid ||= read_pid
+    end
+
+  end
+end

data/lib/uc/shell_helper.rb

@@ -0,0 +1,19 @@
+module Uc
+  module ShellHelper
+
+    def cmd(command, error_msg: nil, return_output: false)
+      puts "Running #{command}"
+      if return_output
+        output = `#{command}`
+      else
+        output = ""
+        system "#{command} 2>&1"
+      end 
+      return_value = $?.exitstatus
+      error_msg ||= "Non zero exit for \"#{command}\""
+      raise ::Uc::Error, error_msg if return_value !=0 
+      return output
+    end 
+
+  end 
+end

data/lib/uc/unicorn.rb

@@ -0,0 +1,56 @@
+require 'uc/mqueue'
+module Uc
+  module Unicorn
+
+    def self.queue_file
+      "config/unicorn_mq"
+    end
+
+    def self.rolling_restart(server, worker, queue_name: nil, sleep_secs: 1)
+      queue_name ||= get_queue_name
+      old_pid = "#{server.config[:pid]}.oldbin"
+      if old_pid != server.pid
+        begin
+          sig = (worker.nr + 1) >= server.worker_processes ? :QUIT : :TTOU
+          Process.kill(sig, File.read(old_pid).to_i)
+
+          mq = ::Uc::Mqueue.new(queue_name)
+          writer = mq.nb_writer
+          writer.send("started worker #{worker.nr + 1}")
+          if sig == :QUIT
+            writer.send("fin")
+          end
+          sleep sleep_secs
+        rescue Errno::ENOENT, Errno::ESRCH, Errno::EAGAIN => e
+        end
+     end
+    end
+
+    def self.clean_env
+      ENV.delete "BUNDLE_BIN_PATH"
+      ENV.delete "RUBYLIB"
+      ENV.delete "RUBYOPT"
+      ENV.delete "GEM_HOME"
+      ENV.delete "GEM_PATH"
+    end
+
+
+    def self.get_queue_name
+      queue_name = queue_name_from_file
+      queue_name ||= "unicorn_#{Process.uid}"
+      return queue_name
+    end
+
+    def self.queue_name_from_file
+      queue_file = Pathname.new self.queue_file
+      queue_name = nil
+      if queue_file.readable?
+        queue_name = File.read(queue_file).chomp
+        queue_name = (queue_name.empty? ? nil : queue_name)
+      end
+      return queue_name
+    end
+
+
+  end
+end

data/lib/uc/version.rb

@@ -0,0 +1,3 @@
+module Uc
+  VERSION = "0.0.1"
+end

data/lib/uc.rb

@@ -0,0 +1,5 @@
+require "uc/version"
+
+module Uc
+  # Your code goes here...
+end

data/uc.gemspec

@@ -0,0 +1,23 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'uc/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "uc"
+  spec.version       = Uc::VERSION
+  spec.authors       = ["Neeraj"]
+  spec.summary       = %q{Unicorn controller}
+  spec.description   = %q{Unicorn controller}
+  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.5"
+  spec.add_development_dependency "rake"
+  spec.add_runtime_dependency "posix_mq", "~> 2.1"
+end

metadata

@@ -0,0 +1,101 @@
+--- !ruby/object:Gem::Specification
+name: uc
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Neeraj
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-05-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !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: posix_mq
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+description: Unicorn controller
+email: 
+executables:
+- uc
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/uc
+- lib/uc.rb
+- lib/uc/error.rb
+- lib/uc/logger.rb
+- lib/uc/mqueue.rb
+- lib/uc/server.rb
+- lib/uc/shell_helper.rb
+- lib/uc/unicorn.rb
+- lib/uc/version.rb
+- uc.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.2.0
+signing_key: 
+specification_version: 4
+summary: Unicorn controller
+test_files: []