honcho 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 3429cf3e359e2fc30f07682a77b4759205a9da00
+  data.tar.gz: 03fa822e64a34857dae9ca770abae29a80a17e05
+SHA512:
+  metadata.gz: acb22985e32a0e7dbca8aaa5104b4de987f3b72c4eb9102d3cf8e596217918666fe685d9dbf3400755a2abcd59928ba9425af2b16180a691454b7415986649bd
+  data.tar.gz: 046b0e51b48ee9f5a5a8983c7b5bc3f0ff270f1b9c92fce0fe3f3aa105544843051d6d85b4efdd257063e9ccd1340fadf7544a48af0b9624f4f3ddd0db096ab6

data/bin/honcho

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require_relative '../lib/honcho'
+
+Honcho.new.run

data/lib/honcho/adapters/base.rb

@@ -0,0 +1,26 @@
+class Honcho
+  module Adapters
+    class Base
+      def initialize(config:, redis:)
+        @redis = redis
+        @config = config
+      end
+
+      attr_reader :config, :redis
+
+      def run?
+        work_to_do? || work_being_done?
+      end
+
+      private
+
+      def work_to_do?
+        fail NotImplementedError, "please define #{this.class.name}##{__method__}"
+      end
+
+      def work_being_done?
+        fail NotImplementedError, "please define #{this.class.name}##{__method__}"
+      end
+    end
+  end
+end

data/lib/honcho/adapters/resque.rb

@@ -0,0 +1,22 @@
+class Honcho
+  module Adapters
+    class Resque < Base
+      private
+
+      def work_to_do?
+        queues = redis.smembers("#{namespace}:queues")
+        counts = queues.map { |q| redis.llen("#{namespace}:queue:#{q}") }
+        counts.any?(&:nonzero?)
+      end
+
+      def work_being_done?
+        # No way to tell via redis if work is being done in resque? Booo.
+        false
+      end
+
+      def namespace
+        config.fetch('namespace')
+      end
+    end
+  end
+end

data/lib/honcho/adapters/sidekiq.rb

@@ -0,0 +1,25 @@
+class Honcho
+  module Adapters
+    class Sidekiq < Base
+      private
+
+      def work_to_do?
+        queues = redis.keys("#{namespace}:queue:*")
+        counts = queues.map { |q| redis.llen(q) }
+        counts.any?(&:nonzero?)
+      end
+
+      def work_being_done?
+        processes = redis.smembers("#{namespace}:processes")
+        counts = processes.map do |process|
+          redis.hget("#{namespace}:#{process}", 'busy').to_i
+        end
+        counts.any?(&:nonzero?)
+      end
+
+      def namespace
+        config.fetch('namespace')
+      end
+    end
+  end
+end

data/lib/honcho/adapters.rb

@@ -0,0 +1,3 @@
+require_relative './adapters/base'
+require_relative './adapters/resque'
+require_relative './adapters/sidekiq'

data/lib/honcho.rb

@@ -0,0 +1,181 @@
+#!/usr/bin/env ruby
+
+require 'redis'
+require 'sys/proctable'
+require 'time'
+require 'stringio'
+require 'yaml'
+require_relative './honcho/adapters'
+
+Thread.abort_on_exception = true
+
+class Honcho
+  COLORS = {
+    red:            31,
+    green:          32,
+    yellow:         33,
+    blue:           34,
+    magenta:        35,
+    cyan:           36,
+    bright_black:   30,
+    bright_red:     31,
+    bright_green:   32,
+    bright_yellow:  33,
+    bright_blue:    34,
+    bright_magenta: 35,
+    bright_cyan:    36,
+    bright_white:   37
+  }.freeze
+
+  def initialize
+    @running = {}
+    @stopping = {}
+    @redis = Redis.new
+    @adapters = build_adapters
+    @colors = assign_colors
+  end
+
+  attr_reader :adapters, :running, :stopping, :redis, :colors
+
+  def run
+    trap(:INT)  { term_all && exit }
+    trap(:TERM) { term_all && exit }
+    loop do
+      check_for_work
+      sleep(interval)
+    end
+  end
+
+  private
+
+  def apps
+    config['apps']
+  end
+
+  def command_template
+    config['command_template'] || '%s'
+  end
+
+  def stop_delay
+    config['stop_delay'] || 30
+  end
+
+  def interval
+    config['interval'] || 2
+  end
+
+  def config
+    @config ||= YAML.load_file('honcho.yml')
+  end
+
+  def check_for_work
+    adapters.each do |adapter|
+      if adapter.run?
+        start(adapter)
+      else
+        stop(adapter)
+      end
+    end
+  end
+
+  def start(adapter)
+    stopping.delete(adapter)
+    return if running[adapter]
+    log(adapter.config['name'], "STARTING\n")
+    running[adapter] = start_command(adapter)
+  end
+
+  def start_command(adapter)
+    command = adapter.config['command']
+    Array(command).map do |cmd|
+      rout, wout = IO.pipe
+      pid = spawn(adapter.config['path'], cmd, wout)
+      Thread.new do
+        log(adapter.config['name'], rout.gets) until rout.eof?
+      end
+      [pid, wout]
+    end
+  end
+
+  def spawn(path, cmd, out)
+    Process.spawn(
+      "cd '#{path}' && " + command_template % cmd,
+      pgroup: true,
+      err: out,
+      out: out
+    )
+  end
+
+  def stop(adapter)
+    return unless running[adapter]
+    if should_stop?(adapter)
+      really_stop(adapter)
+    else
+      stopping[adapter] ||= stop_delay
+      stopping[adapter] -= interval
+    end
+  end
+
+  def should_stop?(adapter)
+    stopping[adapter] && stopping[adapter] <= 0
+  end
+
+  def really_stop(adapter)
+    log(adapter.config['name'], "STOPPING\n")
+    stopping.delete(adapter)
+    return unless running[adapter]
+    running[adapter].each do |(pid, wout)|
+      Process.kill('-TERM', pid)
+      wout.close
+    end
+    running.delete(adapter)
+  end
+
+  def term_all
+    running.values.flatten.each do |(pid)|
+      Process.kill('-TERM', pid)
+    end
+  end
+
+  def log(name, message)
+    color = colors[name]
+    $stdout.write("\e[#{color}m#{name.rjust(label_width)}:\e[0m #{message}")
+  end
+
+  def label_width
+    @label_width ||= apps.keys.map(&:size).max
+  end
+
+  def assign_colors
+    color_values = COLORS.values
+    apps.keys.each_with_object({}) do |app, hash|
+      hash[app] = color_values.shift
+    end
+  end
+
+  def build_adapters
+    apps.flat_map do |app, config|
+      config.map do |type, worker_config|
+        adapter = adapter_from_type(type)
+        next if adapter.nil?
+        adapter.new(
+          config: worker_config.merge('name' => app, 'path' => config['path']),
+          redis: redis
+        )
+      end
+    end.compact
+  end
+
+  def adapter_from_type(type)
+    case type
+    when 'sidekiq'
+      Adapters::Sidekiq
+    when 'resque'
+      Adapters::Resque
+    when 'path'
+      nil
+    else
+      fail "Unknown type #{type}"
+    end
+  end
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification
+name: honcho
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Tim Morgn
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-05-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Sidekiq- and Resque-aware process manager (alternative to Foreman)
+email:
+- tim@timmorgan.org
+executables:
+- honcho
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/honcho
+- lib/honcho.rb
+- lib/honcho/adapters.rb
+- lib/honcho/adapters/base.rb
+- lib/honcho/adapters/resque.rb
+- lib/honcho/adapters/sidekiq.rb
+homepage: https://github.com/seven1m/honcho
+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.5.1
+signing_key: 
+specification_version: 4
+summary: Sidekiq- and Resque-aware process manager (alternative to Foreman)
+test_files: []
+has_rdoc: