foreman-upstart-scaling 0.1

data/data/export/upstart_scaling/master.conf.erb

@@ -0,0 +1,4 @@
+description "<%= app %> processes"
+stop on runlevel [!234]
+stop on starting rc RUNLEVEL=[016]
+

data/data/export/upstart_scaling/process.conf.erb

@@ -0,0 +1,62 @@
+stop on stopping <%= app %>-<%= process.name %>
+
+respawn
+respawn limit 10 5
+kill timeout 20
+
+instance $<%= app.upcase %>_<%= process.name.upcase %>_INSTANCE
+
+# TODO sudo overwrites this those env vars
+env PORT=<%= port %>
+<% engine.environment.each_pair do |var,value| %>
+env <%= var.upcase %>=<%= value %>
+<% end %>
+
+script
+  cd <%= engine.directory %>
+  name="<%= app %>-<%=process.name%>-$<%= app.upcase %>_<%= process.name.upcase %>_INSTANCE"
+
+  # This little dance pipes output into logger with three
+  # goals in mind:
+  #  - Keep the process as a direct child of init (so upstart
+  #    doesn't get confused).
+  #  - Make sure logger isn't a child of the process (so it
+  #    doesn't get confused).
+  #  - Don't leave cruft sitting around in the filesystem.
+
+  # See `man 7 fifo` and `man 7 pipe` for more info.
+  mkdir -p <%= log_root %>
+  stdout_fifo="<%= log_root %>/$name-stdout-log.fifo"
+  stderr_fifo="<%= log_root %>/$name-stderr-log.fifo"
+  rm -f $stdout_fifo
+  rm -f $stderr_fifo
+  mkfifo $stdout_fifo
+  mkfifo $stderr_fifo
+
+  # Start logger reading from the fifos, in the background.
+  # The parens cause the logger process to be reparented to
+  # init, rather than remaining a child of the current proc.
+  #
+  # `setsid` will cause loggers to be in a different session
+  # and process group. If we let the loggers be in the same
+  # process group as the main process, they would be killed
+  # by upstart together with the main process, and all output
+  # would be suppressed during termination.
+  ( exec setsid logger -i -p local0.notice -t $name <$stdout_fifo & )
+  ( exec setsid logger -i -p local0.err    -t $name <$stderr_fifo & )
+
+  # Redirect the current shell's output to the fifos. Now that
+  # we have open fds for those fifos, we don't need the file
+  # system entries any more...
+  exec >$stdout_fifo 2>$stderr_fifo
+
+  # ...so we'll remove it.
+  rm $stdout_fifo
+  rm $stderr_fifo
+
+  # Run the service in the usual way. Our output is already
+  # going to the fifos; that won't change when the new
+  # program executes.
+  exec sudo -H -i -u <%= user %> '<%= process.command %>'
+end script
+

data/data/export/upstart_scaling/process_instances.conf.erb

@@ -0,0 +1,17 @@
+start on starting <%= app %>-<%= process.name %>
+
+task
+
+script
+  <%= app.upcase %>_<%= process.name.upcase %>_SCALE=$(su - <%= user %> -c 'echo $<%= app.upcase %>_<%= process.name.upcase %>_SCALE')
+  [ -z $<%= app.upcase %>_<%= process.name.upcase %>_SCALE ] && <%= app.upcase %>_<%= process.name.upcase %>_SCALE=0
+  export <%= app.upcase %>_<%= process.name.upcase %>_SCALE
+
+  i=1
+  while [ $i -le $<%= app.upcase %>_<%= process.name.upcase %>_SCALE ]; do
+    start <%= app %>-<%= process.name %>-instance <%= app.upcase %>_<%= process.name.upcase %>_INSTANCE=$i
+    i=$(( $i + 1 ))
+  done
+
+end script
+

data/data/export/upstart_scaling/process_master.conf.erb

@@ -0,0 +1,3 @@
+start on starting <%= app %>
+stop on stopping <%= app %>
+

data/lib/foreman/export/upstart_scaling.rb

@@ -0,0 +1,42 @@
+require "erb"
+require "fileutils"
+require "foreman/export"
+
+class Foreman::Export::UpstartScaling < Foreman::Export::Base
+
+  def export
+    error("Must specify a location") unless location
+    FileUtils.mkdir_p location
+    @location = Pathname.new(@location)
+
+    app = self.app || File.basename(engine.directory)
+    user = self.user || app
+    log_root = self.log || "/var/log/#{app}"
+    template_root = self.template || File.expand_path("../../../../data/export/upstart_scaling", __FILE__)
+
+    Dir["#{location}/#{app}*.conf"].each do |file|
+      say "cleaning up: #{file}"
+      FileUtils.rm(file)
+    end
+
+    master_template = export_template("upstart_scaling", "master.conf.erb", template_root)
+    master_config   = ERB.new(master_template).result(binding)
+    write_file "#{location}/#{app}.conf", master_config
+
+    engine.procfile.entries.each do |process|
+      process_master_template = export_template("upstart_scaling", "process_master.conf.erb", template_root)
+      process_master_config = ERB.new(process_master_template).result(binding)
+      write_file "#{location}/#{app}-#{process.name}.conf", process_master_config
+
+      process_instances_template = export_template("upstart_scaling", "process_instances.conf.erb", template_root)
+      process_instances_config = ERB.new(process_instances_template).result(binding)
+      write_file "#{location}/#{app}-#{process.name}-instances.conf", process_instances_config
+
+      process_template = export_template("upstart_scaling", "process.conf.erb", template_root)
+      process_config = ERB.new(process_template).result(binding)
+      write_file "#{location}/#{app}-#{process.name}-instance.conf", process_config
+    end
+  end
+
+end
+

metadata

@@ -0,0 +1,60 @@
+--- !ruby/object:Gem::Specification
+name: foreman-upstart-scaling
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Fabio Kung
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-06-05 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: foreman
+  requirement: &70179280215540 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.46.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70179280215540
+description: Upstart exporter for foreman, that supports worker scaling via env vars.
+email: fabio.kung@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/foreman/export/upstart_scaling.rb
+- data/export/upstart_scaling/master.conf.erb
+- data/export/upstart_scaling/process.conf.erb
+- data/export/upstart_scaling/process_instances.conf.erb
+- data/export/upstart_scaling/process_master.conf.erb
+homepage: http://github.com/fabiokung/foreman-upstart-scaling
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.12
+signing_key: 
+specification_version: 3
+summary: Upstart configuration generator for foreman
+test_files: []