activerain-mongrel_runit 0.2.3

data/lib/mongrel_runit/servicerunner.rb

@@ -0,0 +1,91 @@
+require 'mongrel_runit/base'
+require 'mongrel_runit/service'
+require 'find'
+
+module MongrelRunit
+  class ServiceRunner < Base
+    include Enumerable 
+      
+    def initialize(config)
+      @config = config
+      @mongrel_services = {}
+      @mongrel_names = []
+      
+      raise ArgumentError, "Must have an application_name!" unless @config.has_key?("application_name")
+      raise ArgumentError, "Must have a port!" unless @config.has_key?("port")
+      raise ArgumentError, "Must have a servers option!" unless @config.has_key?("servers")
+      
+      start_port = @config["port"].to_i
+      port_end = start_port + @config["servers"].to_i - 1
+      start_port.upto(port_end) do |port|
+        local_config = config.dup
+        local_config["port"] = port
+        @mongrel_services[port] = MongrelRunit::Service.new(local_config)
+        @mongrel_names << "mongrel-#{@config["application_name"]}-#{port}"
+      end
+    end
+    
+    def method_missing(method)
+      if self.has_command?(method.id2name) || method.id2name == "create"
+        run(method.id2name)
+      else
+        super
+      end
+    end
+    
+    def run(method)
+      return self.create if method == "create"
+      threads = []
+      response = {}
+      final_status = true
+      self.each do |svc|
+        threads << Thread.new(svc) do |s|
+          output, status = s.send(:run, method)
+          response[svc.config["port"]] = [ output, status ]
+          final_status = status if status == false
+        end
+      end
+      threads.each { |t| t.join }
+      return response, final_status
+    end
+    
+    def create
+  
+      response = {}
+      final_status = true
+      self.each do |svc|
+        output, status = svc.create
+        response[svc.config["port"]] = [ output, status ]
+        final_status = status if status == false
+      end
+      
+      service_path = File.expand_path(@config["runit_service_dir"])
+      sv_path = File.expand_path(@config["runit_sv_dir"])
+      Find.find(sv_path, service_path) do |file|
+        if File.directory?(file) || File.symlink?(file)
+          if file =~ /mongrel-#{@config["application_name"]}-(\d+)$/
+            safe = @mongrel_names.detect { |cm| cm =~ /#{File.basename(file)}$/ }
+            if ! safe
+              output = `rm -r #{file}`
+              raise "Cannot remove stale mongrel config" unless $?.success?
+            end
+          end
+        end
+      end
+      return response, final_status
+    end
+    
+    def [](port)
+      @mongrel_services[port]
+    end
+    
+    def length
+      @mongrel_services.length
+    end
+    
+    def each
+      @mongrel_services.each_value { |value| yield value }
+    end
+    
+  end
+end

data/lib/mongrel_runit/version.rb

@@ -0,0 +1,9 @@
+module MongrelRunit #:nodoc:
+  module VERSION #:nodoc:
+    MAJOR = 0
+    MINOR = 2
+    TINY  = 1
+
+    STRING = [MAJOR, MINOR, TINY].join('.')
+  end
+end

data/test/config/mongrel_runit.yml

@@ -0,0 +1,21 @@
+---
+port: 3000
+address: 127.0.0.1
+pid_file: log/mongrel.pid
+servers: 2
+log_file: foo.log
+environment: production
+cwd: /Users/adam/src/sandbox/mongrel_runit
+timeout: 100
+mime_map: log/mime.map
+docroot: htdocs
+num_procs: 1024
+debug: true
+config_path: /foo/bar/config
+config_script: /foo/bar/config/yml
+user: www-data
+group: www-data
+prefix: /monkey
+runit_sv_dir: /etc/sv
+runit_service_dir: /var/service
+application_name: monkey

data/test/config/mongrel_runit_service.yml

@@ -0,0 +1,12 @@
+---
+port: 3000
+address: 127.0.0.1
+servers: 2
+environment: production
+docroot: htdocs
+num_procs: 1024
+user: www-data
+group: www-data
+application_name: monkey
+env_vars:
+  PATH: "$PATH:/var/lib/gems/1.8/bin"

data/test/tc_config.rb

@@ -0,0 +1,93 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+require "mongrel_runit/config"
+require "yaml"
+
+class TestMongrelRunitConfig < Test::Unit::TestCase
+  YAMLFILE = File.dirname(__FILE__) + "/config/mongrel_runit.yml"
+  ARGS = " -e PRODUCTION"
+  ARGS << " -p 8000"
+  ARGS << " -a 127.0.0.1"
+  ARGS << " -l log/file.log"
+  ARGS << " -P log/file.pid" 
+  ARGS << " -n 1024"
+  ARGS << " -m mime-types.conf"
+  ARGS << " -c /tmp"
+  ARGS << " -r /tmp"
+  ARGS << " --debug"
+  ARGS << " -C /conf/path"
+  ARGS << " -S /conf/path/script"
+  ARGS << " --user www-data" 
+  ARGS << " --group www-data"
+  
+  def setup
+    @file_template = {
+      :port => 3000,
+      :address => '127.0.0.1',
+      :pid_file => 'log/mongrel.pid',
+      :servers => 2,
+      :log_file => 'foo.log',
+      :environment => "production",
+      :cwd => "/Users/adam/src/sandbox/mongrel_runit",
+      :timeout => 100,
+      :mime_map => "log/mime.map",
+      :docroot => "htdocs",
+      :num_procs => 1024,
+      :debug => true,
+      :config_path => "/foo/bar/config",
+      :config_script => "/foo/bar/config/yml",
+      :user => "www-data",
+      :group => "www-data",
+      :prefix => "/monkey",
+      :runit_sv_dir => "/etc/sv",
+      :runit_service_dir => "/var/service",
+      :application_name => "monkey",
+    }
+  end
+  
+  def load_file
+    @from_file = MongrelRunit::Config.load_file(YAMLFILE)
+  end
+  
+  def load_args(args)
+     config = MongrelRunit::Config.load_args(args)
+  end
+  
+  def test_load_args
+    args = [ "-c#{File.expand_path('./config/mongrel_runit.yml')}", "status" ]
+    config = load_args(args)
+    @file_template.each do |key, value|
+      assert_equal(value, config["#{key}"], "#{key} has proper value")
+    end
+  end
+  
+  def test_load_appname_args
+    args = [ 
+      "-c#{File.expand_path('./config/mongrel_runit.yml')}",
+      "-amoo",
+      "status", 
+      ]
+    config = load_args(args)
+    @file_template[:application_name] = "moo"
+    @file_template.each do |key, value|
+      assert_equal(value, config["#{key}"], "#{key} has proper value")
+    end
+    assert_equal("status", config.command, "Command is correct")
+  end
+  
+  def test_load_nothing
+    args = [ "status" ]
+    config = load_args(args)
+    @file_template.each do |key, value|
+      assert_equal(value, config["#{key}"], "#{key} has proper value")
+    end
+    assert_equal("status", config.command, "Command is correct")
+  end
+  
+  def test_load_file
+    load_file
+    @file_template.each do |key, value|
+      assert_equal(value, @from_file["#{key}"], "#{key} has proper value")
+    end
+  end
+end

data/test/tc_service.rb

@@ -0,0 +1,153 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+require "mongrel_runit/config"
+require "mongrel_runit/service"
+require "find"
+
+class TestMongrelRunitService < Test::Unit::TestCase
+  
+  def setup
+    @config = {
+     "port" => "8191",
+     "address" => "127.0.0.1",
+     "servers" => "2",
+     "environment" => "production",
+     "cwd" => File.dirname(__FILE__) + '/testapp',
+     "num_procs" => "1024",
+     #"user" => "www-data",
+     #"group" => "www-data",
+     "runit_sv_dir" => File.dirname(__FILE__) + "/sv",
+     "runit_service_dir" => File.dirname(__FILE__) + "/service",
+     "application_name" => "monkey",
+     "svlogd_config" => <<EOH
+s100000
+n100
+EOH
+    }
+    @service = MongrelRunit::Service.new(@config)
+  end
+  
+  def teardown
+    cleanup
+  end
+  
+  def test_service_create
+    assert_equal(MongrelRunit::Service, @service.class, "Got the right object")
+  end
+  
+  def test_run_testapp
+    @service.create
+    if ENV["RUNSVTEST"]
+      sleep 5
+      output, status = @service.start
+      assert(status, "Service Start")
+      ouput, status = @service.stop
+      assert(status, "Service Stop")
+      output, status = @service.start
+      assert(status, "Service Start")
+    end
+  end
+  
+  def test_make_service_link
+    @service.create
+    link = @service.make_service_link
+    assert(File.symlink?(link), "Created service symlink")
+  end
+  
+  def test_make_sv_dir
+    svlogdir = @service.make_sv_dir
+    assert(File.directory?(svlogdir), "Created sv directory")
+  end
+  
+  def test_check_service_dir
+    dir_exists = @service.check_service_dir
+    assert(dir_exists, "Has a service directory")
+  end
+  
+  def test_make_log
+    runlog = @service.make_log
+    assert_equal(@service.logrunfile, IO.read(runlog), "log/run file created")
+    assert(File.executable?(runlog), "log/run file executable")
+  end
+   
+  def test_make_check
+    checkfile = @service.make_check
+    assert_equal(@service.checkfile, IO.read(checkfile), "check file created")
+    assert(File.executable?(checkfile), "check file executable")
+    new_config = @config.dup
+    new_config["checkfile"] = <<EOH
+foo #{@config['address']} #{@config['port']}
+EOH
+    config_service = MongrelRunit::Service.new(new_config)
+    checkfile = config_service.make_check
+    assert_equal("foo 127.0.0.1 8191\n", config_service.checkfile)
+  end
+  
+  def test_make_log_config
+    logconfig = @service.make_log_config
+    assert_equal(@service.logconfig, IO.read(logconfig)) if logconfig
+  end
+  
+  def test_make_run    
+    no_env_runfile = @service.make_run
+    no_env_file = <<EOH
+#!/bin/sh
+exec \\
+  mongrel_rails start -e production -p 8191 -a 127.0.0.1 -c #{File.dirname(__FILE__) + '/testapp'} -n 1024 2>&1
+EOH
+    env_file = <<EOH
+#!/bin/sh
+exec \\
+  env PATH=$PATH:/var/lib/gems/1.8 \\
+  mongrel_rails start -e production -p 8191 -a 127.0.0.1 -c #{File.dirname(__FILE__) + '/testapp'} -n 1024 2>&1
+EOH
+    command_line_file = <<EOH
+#!/bin/sh
+exec \\
+  moofyboogles -p 8191 2>&1
+EOH
+    chpst_path = `which chpst`.chomp
+    chpst_file = <<EOH
+#!/bin/sh
+exec \\
+  #{chpst_path} -o 3365 \\
+  mongrel_rails start -e production -p 8191 -a 127.0.0.1 -c #{File.dirname(__FILE__) + '/testapp'} -n 1024 2>&1
+EOH
+    assert_equal(no_env_file, @service.runfile)
+    assert_equal(@service.runfile, IO.read(no_env_runfile))
+    
+    env_config = @config.dup
+    env_config["env_vars"] = { "PATH" => "$PATH:/var/lib/gems/1.8" }
+    env_service = MongrelRunit::Service.new(env_config)
+    env_service_runfile = env_service.make_run
+    assert_equal(env_file, env_service.runfile)
+    assert_equal(env_service.runfile, IO.read(env_service_runfile))
+    assert(File.executable?(env_service_runfile), "run file executable")
+    
+    chpst_config = @config.dup
+    chpst_config["chpst"] = "-o 3365"
+    chpst_service = MongrelRunit::Service.new(chpst_config)
+    chpst_service_runfile = chpst_service.make_run
+    assert_equal(chpst_file, chpst_service.runfile)
+    assert_equal(chpst_service.runfile, IO.read(chpst_service_runfile))
+    assert(File.executable?(chpst_service_runfile), "run file executable")
+    
+    command_line_config = @config.dup
+    command_line_config["command_line"] = 'moofyboogles -p #{@config["port"]}'
+    command_line_service = MongrelRunit::Service.new(command_line_config)
+    command_line_runfile = command_line_service.make_run
+    assert_equal(command_line_file, command_line_service.runfile, "Command line runfile works")
+    assert_equal(command_line_service.runfile, IO.read(command_line_runfile), "Command line generated file works")
+  end
+  
+  def cleanup
+   Find.find(@config["runit_sv_dir"], @config["runit_service_dir"]) do |file|
+     if File.directory?(file) || File.symlink?(file)
+       if File.basename(file) =~ /(.+)-(\d+)/
+         `rm -r #{file}`
+       end
+     end
+   end
+  end
+end
+

data/test/tc_servicerunner.rb

@@ -0,0 +1,74 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+require "mongrel_runit/servicerunner"
+require "find"
+
+class TestMongrelRunitServiceRunner < Test::Unit::TestCase
+  
+  def setup
+    @config = {
+     "port" => "8191",
+     "address" => "127.0.0.1",
+     "servers" => "2",
+     "environment" => "production",
+     "cwd" => File.dirname(__FILE__) + '/testapp',
+     "num_procs" => "1024",
+     #"user" => "www-data",
+     #"group" => "www-data",
+     "runit_sv_dir" => File.dirname(__FILE__) + "/sv",
+     "runit_service_dir" => File.dirname(__FILE__) + "/service",
+     "application_name" => "monkey",
+     "svlogd_config" => <<EOH
+s100000
+n100
+EOH
+    }
+    @servicerunner = MongrelRunit::ServiceRunner.new(@config)
+  end
+  
+  def teardown
+    cleanup
+  end
+  
+  def test_service_create
+    assert_equal(MongrelRunit::ServiceRunner, @servicerunner.class, "Got the right object")
+  end
+
+  def test_create
+    result, status = @servicerunner.create
+    @servicerunner.each do |service|
+      assert(result.has_key?(service.config["port"]), "Created #{service.config['port']}")
+      assert(File.directory?(service.svdir), "Created svdir")
+    end 
+  end
+  
+  def test_run_start
+    if ENV["RUNSVTEST"]
+      result, status = @servicerunner.create
+      assert(status, "Created for each service")
+      sleep 5
+      result, status = @servicerunner.start
+      assert(status, "Started each service")
+    end
+  end
+  
+  def test_run_stop
+    if ENV["RUNSVTEST"]
+      result, status = @servicerunner.create
+      assert(status, "Created each service")
+      sleep 5
+      result, status = @servicerunner.stop
+      assert(status, "Stopped each service")
+    end
+  end
+  
+  def cleanup
+    Find.find(@config["runit_sv_dir"], @config["runit_service_dir"]) do |file|
+      if File.directory?(file) || File.symlink?(file)
+        if File.basename(file) =~ /mongrel-(.+)-(\d+)/
+         `rm -r #{file}`
+        end
+      end
+    end
+  end
+end