diff --git a/export/upstart/master.conf.erb b/export/upstart/master.conf.erb new file mode 100644 index 0000000..dd1a739 --- /dev/null +++ b/export/upstart/master.conf.erb @@ -0,0 +1,13 @@ +pre-start script + +bash << "EOF" + mkdir -p /var/log/<%= app %> + + <% engine.processes.keys.sort.each do |process| %> + <% 1.upto(concurrency[process]).each do |num| %> + start <%=app%>-<%=process%>-<%=num%> + <% end %> + <% end %> +EOF + +end script diff --git a/export/upstart/process.conf.erb b/export/upstart/process.conf.erb new file mode 100644 index 0000000..692c71a --- /dev/null +++ b/export/upstart/process.conf.erb @@ -0,0 +1,5 @@ +stop on stopping <%= app %> +respawn + +chdir <%= engine.directory %> +exec <%= process.command %> >> /var/log/<%=app%>/<%=process.name%>-<%=num%>.log 2>&1 diff --git a/lib/foreman/cli.rb b/lib/foreman/cli.rb index a1be7ea..02a8ed9 100644 --- a/lib/foreman/cli.rb +++ b/lib/foreman/cli.rb @@ -6,46 +6,58 @@ require "thor" class Foreman::CLI < Thor - desc "start [PROCFILE]", "Run the app described in PROCFILE" + class_option :procfile, :type => :string, :aliases => "-p", :desc => "Default: ./Procfile" - def start(procfile="Procfile") - error "#{procfile} does not exist." unless procfile_exists?(procfile) - Foreman::Engine.new(procfile).start + desc "start [PROCESS]", "Start the application, or a specific process" + + method_option :screen, :type => :boolean, :aliases => "-s" + + def start(process=nil) + check_procfile! + + if process + engine.execute(process) + elsif options[:screen] + engine.screen + else + engine.start + end end - desc "execute PROCESS [PROCFILE]", "Run an instance of the specified process from PROCFILE" + desc "export FORMAT LOCATION", "Export the application to another process management format" - def execute(process, procfile="Procfile") - error "#{procfile} does not exist." unless procfile_exists?(procfile) - Foreman::Engine.new(procfile).execute(process) - end + method_option :app, :type => :string, :aliases => "-a" + method_option :concurrency, :type => :string, :aliases => "-c", + :banner => '"alpha=5,bar=3"' - desc "screen [PROCFILE]", "Run the app described in PROCFILE as screen windows" - - def screen(procfile="Procfile") - error "#{procfile} does not exist." unless procfile_exists?(procfile) - Foreman::Engine.new(procfile).screen - end - - desc "export APP [PROCFILE] [FORMAT]", "Export the app described in PROCFILE as APP to another FORMAT" - - def export(app, procfile="Procfile", format="upstart") - error "#{procfile} does not exist." unless procfile_exists?(procfile) + def export(format, location=nil) + check_procfile! formatter = case format when "upstart" then Foreman::Export::Upstart else error "Unknown export format: #{format}." end - formatter.new(Foreman::Engine.new(procfile)).export(app) + formatter.new(engine).export(location, + :name => options[:app], + :concurrency => options[:concurrency] + ) + rescue Foreman::Export::Exception => ex + error ex.message end - desc "scale APP PROCESS AMOUNT", "Change the concurrency of a given process type" +private ###################################################################### - def scale(app, process, amount) - config = Foreman::Configuration.new(app) - error "No such process: #{process}." unless config.processes[process] - config.scale(process, amount) + def check_procfile! + error("Procfile does not exist.") unless File.exist?(procfile) + end + + def engine + @engine ||= Foreman::Engine.new(procfile) + end + + def procfile + options[:procfile] || "./Procfile" end private ###################################################################### diff --git a/lib/foreman/engine.rb b/lib/foreman/engine.rb index db5ca97..1428d67 100644 --- a/lib/foreman/engine.rb +++ b/lib/foreman/engine.rb @@ -83,8 +83,8 @@ private ###################################################################### info stdin.gets, process end end - rescue PTY::ChildExited - # exited + rescue PTY::ChildExited, Interrupt + info "process exiting", process end end end diff --git a/lib/foreman/export.rb b/lib/foreman/export.rb index 4aef006..adb83ae 100644 --- a/lib/foreman/export.rb +++ b/lib/foreman/export.rb @@ -1,6 +1,7 @@ require "foreman" module Foreman::Export + class Exception < ::Exception; end end require "foreman/export/upstart" diff --git a/lib/foreman/export/base.rb b/lib/foreman/export/base.rb new file mode 100644 index 0000000..c7bdde8 --- /dev/null +++ b/lib/foreman/export/base.rb @@ -0,0 +1,48 @@ +require "foreman/configuration" +require "foreman/export" + +class Foreman::Export::Base + + attr_reader :engine + + def initialize(engine) + @engine = engine + end + + def export + raise "export method must be overridden" + end + +private ###################################################################### + + def error(message) + raise Foreman::Export::Exception.new(message) + end + + def say(message) + puts "[foreman export] %s" % message + end + + def export_template(name) + File.read(File.expand_path("../../../../export/#{name}", __FILE__)) + end + + def parse_concurrency(concurrency) + @concurrency ||= begin + pairs = concurrency.to_s.gsub(/\s/, "").split(",") + pairs.inject(Hash.new(1)) do |hash, pair| + process, amount = pair.split("=") + hash.update(process => amount.to_i) + end + end + end + + def write_file(filename, contents) + say "writing: #{filename}" + + File.open(filename, "w") do |file| + file.puts contents + end + end + +end diff --git a/lib/foreman/export/upstart.rb b/lib/foreman/export/upstart.rb index e646202..59bb100 100644 --- a/lib/foreman/export/upstart.rb +++ b/lib/foreman/export/upstart.rb @@ -1,51 +1,42 @@ +require "erb" require "foreman/configuration" -require "foreman/export" +require "foreman/export/base" -class Foreman::Export::Upstart +class Foreman::Export::Upstart < Foreman::Export::Base - attr_reader :engine + def export(location, options={}) + error("Must specify a location") unless location - def initialize(engine) - @engine = engine - end + FileUtils.mkdir_p location - def export(app) - FileUtils.mkdir_p "/etc/foreman" - FileUtils.mkdir_p "/etc/init" + app = options[:app] || File.basename(engine.directory) - config = Foreman::Configuration.new(app) + Dir["#{location}/#{app}*.conf"].each do |file| + say "cleaning up: #{file}" + FileUtils.rm(file) + end - write_file "/etc/init/#{app}.conf", <<-UPSTART_MASTER -pre-start script + concurrency = parse_concurrency(options[:concurrency]) -bash << "EOF" - mkdir -p /var/log/#{app} + master_template = export_template("upstart/master.conf.erb") + master_config = ERB.new(master_template).result(binding) + write_file "#{location}/#{app}.conf", master_config - if [ -f /etc/foreman/#{app}.conf ]; then - source /etc/foreman/#{app}.conf - fi + process_template = export_template("upstart/process.conf.erb") + + engine.processes.values.each do |process| + 1.upto(concurrency[process.name]) do |num| + process_config = ERB.new(process_template).result(binding) + write_file "#{location}/#{app}-#{process.name}-#{num}.conf", process_config + end + end - for process in $( echo "$#{app}_processes" ); do - process_count_config="#{app}_$process" - process_count=${!process_count_config} - - for ((i=1; i<=${process_count:=1}; i+=1)); do - start #{app}-$process NUM=$i - done - done -EOF - -end script + return + write_file "#{location}/#{app}.conf", <<-UPSTART_MASTER UPSTART_MASTER - engine.processes.values.each do |process| - write_file "/etc/init/#{app}-#{process.name}.conf", <<-UPSTART_CHILD -instance $NUM -stop on stopping #{app} -respawn - -chdir #{engine.directory} -exec #{process.command} >>/var/log/#{app}/#{process.name}.log 2>&1 + engine.processes.each do |process| + write_file process_conf, <<-UPSTART_CHILD UPSTART_CHILD end @@ -55,12 +46,4 @@ exec #{process.command} >>/var/log/#{app}/#{process.name}.log 2>&1 config.write end -private ###################################################################### - - def write_file(filename, contents) - File.open(filename, "w") do |file| - file.puts contents - end - end - end diff --git a/spec/foreman/cli_spec.rb b/spec/foreman/cli_spec.rb index 5468a5d..b2114a2 100644 --- a/spec/foreman/cli_spec.rb +++ b/spec/foreman/cli_spec.rb @@ -25,27 +25,6 @@ describe "Foreman::CLI" do end end - describe "execute" do - describe "with a non-existent Procfile" do - it "prints an error" do - mock_error(subject, "Procfile does not exist.") do - dont_allow.instance_of(Foreman::Engine).start - subject.execute("alpha") - end - end - end - - describe "with a Procfile" do - before(:each) { write_procfile } - - it "runs successfully" do - dont_allow(subject).error - mock.instance_of(Foreman::Engine).execute("alpha") - subject.execute("alpha") - end - end - end - describe "export" do describe "with a non-existent Procfile" do it "prints an error" do @@ -62,7 +41,7 @@ describe "Foreman::CLI" do describe "with an invalid formatter" do it "prints an error" do mock_error(subject, "Unknown export format: invalidformatter.") do - subject.export("testapp", "Procfile", "invalidformatter") + subject.export("invalidformatter") end end end @@ -72,33 +51,11 @@ describe "Foreman::CLI" do it "runs successfully" do dont_allow(subject).error - subject.export("testapp") - end - end - end - end - - describe "scale" do - describe "without an existing configuration" do - it "displays an error" do - mock_error(subject, "No such process: alpha.") do - subject.scale("testapp", "alpha", "2") - end - end - end - - describe "with an existing configuration" do - before(:each) { write_foreman_config("testapp") } - - it "scales a process that exists" do - mock.instance_of(Foreman::Configuration).scale("alpha", "2") - subject.scale("testapp", "alpha", "2") - end - - it "errors if a process that does not exist is specified" do - mock_error(subject, "No such process: invalidprocess.") do - dont_allow.instance_of(Foreman::Configuration).scale - subject.scale("testapp", "invalidprocess", "2") + mock.instance_of(Foreman::Export::Upstart).export("/tmp/foo", { + :concurrency => nil, + :name => nil + }) + subject.export("upstart", "/tmp/foo") end end end