mongrel-cluster-refresh 0.0.1

data/README.markdown

@@ -0,0 +1,36 @@
+# mongrel-cluster-refresh
+
+Send graceful restart signal to most resource intensive processes in a mongrel cluster. cluster::refresh finds the cluster processes by inspecting the mongrel_cluster.yml then looking up the process' MEM and CPU usage and sorting them by MEM usage, CPU usage and process ID. Then the top N processes specified, or 1/3 of the cluster by default, are sent the USR2 signal which instructs the mongrel process to restart itself.
+
+## Usage
+
+* gem install mongrel-cluster-refresh -s http://gemcutter.org
+* mongrel cluster::refresh -C /path/to/mongrel_cluster.yml
+
+## Options
+
+* -C, --config PATH - Path to cluster configuration file; defaults to config/mongrel_cluster.yml in the current working directory
+* -R, --refresh NUM - Number of mongrels to refresh; defaults to 1/3 of the <tt>servers</tt> directive in your mongrel_cluster.yml
+* --clean - Remove orphaned pid files
+
+## License
+
+Copyright (c) 2009 Ryan Carmelo Briones &lt;<ryan.briones@brionesandco.com>&gt;
+
+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/Rakefile

@@ -0,0 +1,23 @@
+require 'rubygems'
+require 'rake/gempackagetask'
+ 
+spec = Gem::Specification.new do |s|
+  s.name = 'mongrel-cluster-refresh'
+  s.version = '0.0.1'
+  s.summary = 'Send graceful restart signal to most resource intensive processes in a mongrel cluster'
+  s.files = FileList['[A-Z]*', 'lib/init.rb']
+  s.has_rdoc = false
+  s.author = 'Ryan Carmelo Briones'
+  s.email = 'ryan.briones@brionesandco.com'
+  s.homepage = 'http://brionesandco.com/ryanbriones'
+end
+ 
+package_task = Rake::GemPackageTask.new(spec) {}
+ 
+task :build_gemspec do
+  File.open("#{spec.name}.gemspec", "w") do |f|
+    f.write spec.to_ruby
+  end
+end
+ 
+task :default => [:build_gemspec, :gem]

data/lib/init.rb

@@ -0,0 +1,105 @@
+require 'rubygems'
+require 'gem_plugin'
+require 'mongrel'
+require 'yaml'
+
+module Cluster
+  class Refresh < GemPlugin::Plugin '/commands'
+    include Mongrel::Command::Base
+
+    def configure
+      options [
+               ['-C', '--config PATH', 'Path to cluster configuration file', :@config_file, "config/mongrel_cluster.yml"],
+               ['-R', '--refresh NUM', 'Number of mongrels to refresh', :@refresh_num, nil],
+               ['', '--clean', "Remove orphaned pid files", :@clean, false],
+              ]
+    end
+
+    def validate
+      valid_exists?(@config_file, "Configuration file does not exist")
+      @valid
+    end
+    
+    def run
+      read_config
+      collect_mongrel_stats
+      refresh
+    end
+
+    def read_config
+      @options = {
+        "port" => 3000,
+        "pid_file" => "tmp/pids/mongrel.pid",
+        "servers" => 2
+      }
+
+      conf = YAML.load_file(@config_file)
+      @options.merge!(conf) if conf
+
+      third_of_servers = @options["servers"].to_i/3
+      @refresh_num ||= (third_of_servers.zero? ? 1 : third_of_servers)
+    end
+
+    def collect_mongrel_stats
+      start_port = @options["port"].to_i
+      end_port = start_port + @options["servers"].to_i - 1
+      
+      mongrel_instance_base = {
+        :port => nil,
+        :pid => nil,
+        :cpu => nil,
+        :mem => nil
+      }
+      
+      @mongrels = (start_port..end_port).inject([]) do |mongrels, port|
+        mongrel = mongrel_instance_base.dup
+        begin
+          mongrel[:port] = port
+          mongrel[:pid] = File.read(port_pid_file(port)).strip.to_i
+          
+          process_string = `ps -p #{mongrel[:pid]} -o pcpu= -o pmem=`
+
+          if(process_string.nil? || process_string.strip == "")
+            if @clean
+              File.unlink(port_pid_file(port))
+              puts "missing process: removing #{port_pid_file(port)}"
+            end
+            
+            next(mongrels)
+          end
+
+          cpu, mem = process_string.split(" ").map { |field| field.strip }
+          
+          mongrel[:cpu] = cpu.to_f
+          mongrel[:mem] = mem.to_f
+
+          mongrels << mongrel
+        ensure
+          mongrels
+        end
+      end
+
+      @mongrels = @mongrels.sort_by { |mongrel| [-mongrel[:mem], -mongrel[:cpu], mongrel[:pid]] }
+    end
+
+    def refresh
+      @mongrels.first(@refresh_num).each do |mongrel|
+        puts "restarting mongrel on port #{mongrel[:port]}"
+        Process.kill("USR2", mongrel[:pid])
+      end
+    end
+
+    def port_pid_file(port)
+      pid_file_ext = File.extname(@options["pid_file"])
+      pid_file_base = File.basename(@options["pid_file"], pid_file_ext)
+
+      path = []
+      path << @options["cwd"] if @options["cwd"]
+      path << File.dirname(@options["pid_file"])
+      pid_file_dir = File.join(path)
+      
+      pid_file = [pid_file_base, port].join(".") + pid_file_ext
+      File.join(pid_file_dir, pid_file)
+    end
+  end
+end

metadata

@@ -0,0 +1,57 @@
+--- !ruby/object:Gem::Specification 
+name: mongrel-cluster-refresh
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Ryan Carmelo Briones
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-26 00:00:00 -04:00
+default_executable: 
+dependencies: []
+
+description: 
+email: ryan.briones@brionesandco.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- README.markdown
+- Rakefile
+- lib/init.rb
+has_rdoc: true
+homepage: http://brionesandco.com/ryanbriones
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Send graceful restart signal to most resource intensive processes in a mongrel cluster
+test_files: []
+