pazuzu 0.1.2

data/lib/pazuzu/version.rb

@@ -0,0 +1,3 @@
+module Pazuzu
+  VERSION = '0.1.2'
+end

data/lib/pazuzu/worker.rb

@@ -0,0 +1,173 @@
+module Pazuzu
+
+  class Worker
+
+    include Utility::Runnable
+
+    def initialize(application, name)
+      @application = application
+      @name = name
+      @instances = Utility::RunnablePool.new
+      @target_instance_count = 1
+      @dynamic_instance_count = nil
+      @logger = Utility::AnnotatedLogger.new(
+        @application.supervisor.logger, qname)
+      super()
+    end
+
+    def configure!(root_path, command_line, configuration)
+      restart_required = false
+
+      if [:starting, :running].include?(run_state) and root_path != @root_path
+        restart_required = true
+      end
+      @root_path = root_path
+
+      new_command_line = configuration['command_line']
+      new_command_line ||= command_line
+      new_command_line ||= @command_line
+      if new_command_line != @command_line
+        if [:starting, :running].include?(run_state)
+          restart_required = true
+        end
+        @command_line = new_command_line
+      end
+      
+      if restart_required
+        @logger.warn "Root path or command line has changed, instances must be restarted/added manually"
+      end
+
+      new_target_instance_count = configuration['num_instances'].try(:to_i) || 1
+      if new_target_instance_count != @target_instance_count
+        @target_instance_count = new_target_instance_count
+      end
+
+      if [:starting, :running].include?(run_state)
+        adjust_instances!
+      end
+    end
+
+    def qname
+      [@application.name, @name].join('.')
+    end
+
+    def revert_dynamic_instance_count!
+      @dynamic_instance_count = nil
+      adjust_instances!
+    end
+
+    def set_instance_count!(value)
+      @dynamic_instance_count = [0, value].max
+      adjust_instances!
+    end
+
+    def add_instance_count!(value)
+      @dynamic_instance_count ||= @target_instance_count
+      @dynamic_instance_count = [0, @dynamic_instance_count + value].max
+      adjust_instances!
+    end
+
+    def adjust_instances!
+      target_count = @dynamic_instance_count
+      target_count ||= @target_instance_count
+      if @instances.length != target_count
+        if run_state == :running
+          @logger.info "Adjusting from #{@instances.length} to #{target_count} instances"
+        end
+        while @instances.length < target_count
+          instance = Instance.new(self, @instances.length + 1, @root_path, @command_line)
+          @instances.register(instance)
+        end
+        if @instances.length > target_count
+          children_by_index = @instances.children.sort_by { |c| c.index }
+          (children_by_index[target_count..-1] || []).each do |instance|
+            @instances.unregister(instance)
+          end
+        end
+      end
+    end
+
+    # Go through available cgroups and find instances that are orphaned.
+    def recover_orphaned_instances!
+      index = @instances.length + 1
+      loop do
+        instance = Instance.new(self, index, @root_path, @command_line)
+        cgroup = @application.supervisor.cgroup_for_instance(instance)
+        if cgroup.mounted?
+          if instance.attachable?
+            @logger.info "Discovered orphaned instance #{index}, attaching it"
+            @instances.register(instance)
+          end
+          index += 1
+        else
+          break
+        end
+      end
+    end
+
+    # Call to set up a forked process according to the application's
+    # configuration.
+    def setup_spawned_process!
+      @application.setup_spawned_process!
+    end
+
+    def instances
+      @instances.children
+    end
+
+    def log_entries
+      entries = @instances.children.map(&:log_entries)
+      entries.flatten!(1)
+      entries.sort_by! { |(source, time, message)| time }
+      entries
+    end
+
+    attr_reader :name
+    attr_reader :command_line
+    attr_reader :application
+
+    private
+
+      def on_starting
+        @logger.info "Starting"
+        @monitor_thread ||= Thread.start { monitor_instances }
+        adjust_instances!
+        recover_orphaned_instances!
+        @instances.start!
+        runnable_state.started!
+      end
+
+      def on_started
+        @logger.info "Started"
+      end
+
+      def on_stopping
+        @logger.info "Stopping"
+        @instances.stop!
+      end
+
+      def on_stopped
+        @logger.info "Stopped"
+      end
+
+      def on_failed
+      end
+
+      def monitor_instances
+        while [:starting, :running, :stopping].include?(run_state)
+          if runnable_state.state == :stopping
+            unless @instances.children.any? { |w| w.run_state == :running }
+              runnable_state.stopped!
+            end
+          end
+          sleep(1)
+        end
+      rescue Exception => e
+        @logger.error(e.message)
+      ensure
+        @monitor_thread = nil
+      end
+
+  end
+
+end

metadata

@@ -0,0 +1,193 @@
+--- !ruby/object:Gem::Specification
+name: pazuzu
+version: !ruby/object:Gem::Version
+  version: 0.1.2
+  prerelease: 
+platform: ruby
+authors:
+- Alexander Staubo
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.1.0
+- !ruby/object:Gem::Dependency
+  name: statemachine
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.0.1
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: i18n
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: scashin133-syslog_logger
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.7.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.7.3
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Pazuzu is a supervisor daemon that manages pools of application processes
+  as daemons.
+email:
+- alex@origo.no
+executables:
+- pazuzu
+- pazuzud
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Pazuzu.gemspec
+- README.md
+- Rakefile
+- bin/pazuzu
+- bin/pazuzud
+- lib/pazuzu.rb
+- lib/pazuzu/application.rb
+- lib/pazuzu/cgroup.rb
+- lib/pazuzu/command_line/controller.rb
+- lib/pazuzu/control/protocol.rb
+- lib/pazuzu/control/socket_client.rb
+- lib/pazuzu/control/socket_server.rb
+- lib/pazuzu/instance.rb
+- lib/pazuzu/procfiles.rb
+- lib/pazuzu/supervisor.rb
+- lib/pazuzu/supervisor_runner.rb
+- lib/pazuzu/utility/annotated_logger.rb
+- lib/pazuzu/utility/output_tailer.rb
+- lib/pazuzu/utility/process_spawning.rb
+- lib/pazuzu/utility/rate_limiter.rb
+- lib/pazuzu/utility/runnable.rb
+- lib/pazuzu/utility/runnable_pool.rb
+- lib/pazuzu/version.rb
+- lib/pazuzu/worker.rb
+homepage: http://github.com/origo/pazuzu
+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'
+      segments:
+      - 0
+      hash: 261249866598060842
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 261249866598060842
+requirements: []
+rubyforge_project: pazuzu
+rubygems_version: 1.8.22
+signing_key: 
+specification_version: 3
+summary: Pazuzu is a supervisor daemon that manages pools of application processes
+  as daemons.
+test_files: []