seesaw 0.1.0 → 0.2.0

data/History.txt

@@ -1,3 +1,9 @@
+== 0.2.0 / 2007-08-19
+
+* added seesaw::switch command to switch to a guaranteed running cluster configuration
+* added seesaw::configure command to autogenerate configuration files
+
+
 == 0.1.0 / 2007-08-18
 
 * 1 major enhancement

data/lib/seesaw/init.rb

@@ -3,46 +3,49 @@ require 'mongrel'
 require 'yaml'
 require 'fileutils'
 require 'seesaw/mongrel_cluster_patch'
+require "erb"
 
 module Seesaw
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
   
   module CommandBase
+    def log(message)
+      puts message
+    end   
+
     def symlink(cluster)
-      p "symlink to #{cluster}"
+      log "symlink to #{cluster}"
       file = File.join(@config_path, @config_files[cluster])
-      p "target: #{file}"
-      FileUtils.ln_sf( file, @config_symlink)
+      system("#{@symlink_cmd} #{file} #{@config_symlink}")
     end
     
     def restart_mongrels(cluster=nil)
-      p "restart mongrels #{cluster}"
+      log "restart mongrels #{cluster}"
       stop_mongrels(cluster)
       start_mongrels(cluster)
     end
     
     def start_mongrels(cluster=nil)
-      p "start mongrels #{cluster}"
+      log "start mongrels #{cluster}"
       cmd = "mongrel_rails seesaw::start"
-      cmd << " --cluster #{cluster}" if cluster
+      cmd << " --cluster #{cluster}" if cluster && cluster != "all"
       system(cmd)
     end
 
     def stop_mongrels(cluster=nil)
-      p "stop mongrels #{cluster}"
+      log "stop mongrels #{cluster}"
       cmd = "mongrel_rails seesaw::stop"
       cmd << " --cluster #{cluster}" if cluster
       system(cmd)
     end
     
     def restart_http(cluster=nil)
-      p "nginx restart"
+      log "webserver restart"
       system(@restart_cmd)
-      sleep 5
     end    
     
     def shutdown_half_cluster(half)
-      p "shutdown half #{half}"
+      log "shutdown half #{half}"
       other_half = half == 1 ? 2 : 1
       symlink(other_half)
       restart_http
@@ -50,14 +53,14 @@ module Seesaw
     end
     
     def switch_to_half_cluster(half)
-      p "switch_to_half_cluster #{half}"
+      log "switch_to_half_cluster #{half}"
       other_half = half == 1 ? 2 : 1
       shutdown_half_cluster(other_half)
       start_mongrels(other_half)
     end
     
     def start_cluster
-      p "start cluster"
+      log "start cluster"
       symlink("all")
       restart_http
     end
@@ -72,6 +75,7 @@ module Seesaw
       @config_symlink = @options["config_symlink"] || "cluster.conf"
       @config_symlink = File.join(@config_path, @config_symlink)
       @restart_cmd = @options["restart_cmd"] || "apachectl graceful"
+      @symlink_cmd = @options["symlink_cmd"] || "ln -sf"
     end
     
     def validate
@@ -87,21 +91,146 @@ module Seesaw
     
   end
   
-  class Migrate < GemPlugin::Plugin "/commands"
+  class Ripple < GemPlugin::Plugin "/commands"
     include Mongrel::Command::Base
     include CommandBase
     
     def run
       parse_options
-      p "c1"
       switch_to_half_cluster(1)
-      p "c2"
       switch_to_half_cluster(2)
-      p "nighty night"
-      sleep 20
-      p "all"
       start_cluster
-      p "done"
+      log "done"
     end      
   end  
-end
+
+  class Switch < GemPlugin::Plugin "/commands"
+    include Mongrel::Command::Base
+    include CommandBase
+
+    def configure 
+      options [
+        ['-C', '--config PATH', "Path to cluster configuration file", :@config_file, "config/seesaw.yml"],        
+        ['', '--cluster CLUSTER', "Switch to cluster configuration CLUSTER: [1|2|all]", :@cluster, 'all']
+      ]
+    end
+    
+    def validate
+      valid?(["1","2","all"].include?(@cluster), "invalid option for --cluster")
+      @valid
+    end
+
+    def run
+      parse_options
+      unless "all" == @cluster
+        @cluster = @cluster.to_i
+        other_half = @cluster == 1 ? 2 : 1
+      end
+      start_mongrels(@cluster)
+      symlink(@cluster)
+      restart_http
+      stop_mongrels(other_half) unless "all" == @cluster
+    end      
+  end
+  
+  class Configure < GemPlugin::Plugin "/commands"
+    include Mongrel::Command::Base    
+    
+    def configure 
+      options [
+        ['-N', '--num-servers INT', "Number of Mongrel servers", :@servers, 2],
+        ['-C', '--config PATH', "Path to seesaw configuration file", :@config_file, 'config/seesaw.yml'],
+        ['-M', '--mongrel PATH', "Path to mongrel cluster configuration file", :@mongrel_config_file, 'config/mongrel_cluster.yml'],
+        ['', '--http_config PATH', "Path to the http server configuration includes", :@http_config_path, "config/http_cluster"],
+        ['', '--server TYPE', "Webserver to generate example configuration for. TYPE can be nginx or apache. Default is nginx.", :@server, 'nginx']
+      ]
+    end
+
+    def validate
+      valid?(['nginx','apache'].include?(@server), "Server must be nginx or apache")
+      valid_dir? File.dirname(@config_file), "Path to config file not valid: #{@config_file}"
+      valid?(File.exist?(@mongrel_config_file), "Path to mongrel cluster config file not valid: #{@mongrel_config_file}")
+      
+      return @valid
+    end
+    
+    def write_config_include(file, ports)
+      
+      log "writing #{file}"
+      
+      ip = "127.0.0.1"
+      server_lines = ""
+      line_template = ERB.new(INCLUDES[@server][:server_line])
+      ports.each do |port|
+        server_lines << line_template.result(binding)
+      end
+      
+      template = ERB.new(INCLUDES[@server][:template])
+      File.open(file, "w") { |f| f.write(template.result(binding)) }
+    end
+
+    def run
+      restart_cmds = {
+        'nginx' => "sudo kill -HUP `cat /opt/local/var/log/nginx.pid`",
+        'apache' => "sudo apachectl graceful"        
+      }
+      
+      @options = {}
+      
+      @options["config_path"] = @http_config_path
+      @options["restart_cmd"] = restart_cmds[@server]
+      @options["symlink_cmd"] = "ln -sf"
+      @options["config_symlink"] = "cluster.conf"
+      @options["config_files"] = {}
+      @options["config_files"]["all"] = "cluster_all.conf"
+      @options["config_files"][1] = "cluster_1.conf"
+      @options["config_files"][2] = "cluster_2.conf"      
+      
+      # write seesaw configuration
+      log "Writing configuration file to #{@config_file}."
+      File.open(@config_file,"w") {|f| f.write(@options.to_yaml)}
+      
+      # read mongrel cluster configuration to find number of servers
+      mongrel_conf = YAML.load(File.read(@mongrel_config_file))
+      
+      # create cluster config directory if it doesn't exist
+      Dir.mkdir(@http_config_path) unless File.exist?(@http_config_path)
+      
+      # write cluster configuration files for appropriate server
+      start_port = mongrel_conf["port"]
+      end_port = mongrel_conf["port"]+mongrel_conf["servers"] - 1
+      mid_port = ((end_port+start_port)/2.0).ceil
+      
+      # full cluster
+      write_config_include(File.join(@http_config_path,@options["config_files"]["all"]), (start_port..end_port))
+      
+      # front half
+      write_config_include(File.join(@http_config_path,@options["config_files"][1]), (start_port..mid_port-1))
+      
+      # back half
+      write_config_include(File.join(@http_config_path,@options["config_files"][2]), (mid_port..end_port))
+      
+      # symlink
+      system("#{@options["symlink_cmd"]} #{File.expand_path(File.join(@http_config_path,@options["config_files"]["all"]))} #{File.join(@http_config_path,@options["config_symlink"])}")
+      
+      log "Don't forget to include #{File.expand_path("#{@http_config_path}/cluster.conf")} in your webserver configuration."
+    end  
+  end
+  
+  INCLUDES = {}
+  INCLUDES['nginx'] = {}
+  INCLUDES['apache'] = {}
+  INCLUDES['nginx'][:template]=<<-TEMPLATE
+upstream mongrel_pack {
+<%= server_lines %>}
+  TEMPLATE
+  INCLUDES['nginx'][:server_line] = "  server <%=ip%>:<%=port%>;\n"
+
+  INCLUDES['apache'][:template]=<<-TEMPLATE
+<Proxy balancer://mongrel_pack>
+<%= server_lines %></Proxy>
+  TEMPLATE
+  INCLUDES['apache'][:server_line] = "  BalancerMember http://<%=ip%>:<%=port%>;\n"
+    
+end
+

data/lib/seesaw/mongrel_cluster_patch.rb

@@ -287,7 +287,7 @@ module Seesaw
     
   end
   
-  class Configure < GemPlugin::Plugin "/commands"
+  class ConfigureMongrels < GemPlugin::Plugin "/commands"
     include ExecBase
     
     def configure 

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.9.4
 specification_version: 1
 name: seesaw
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
-date: 2007-08-18 00:00:00 +10:00
+  version: 0.2.0
+date: 2007-08-19 00:00:00 +10:00
 summary: Ripple-restart a mongrel cluster with no downtime
 require_paths: 
 - lib