Contents

module Freyr
  class ServiceGroup < DelegateClass(Array)
    extend Forwardable
    include TSort
    service_methods = Service.instance_methods - Class.instance_methods
    def_delegators :first, *service_methods
    
    def initialize *services
      @all_services = []
      super(services)
    end

    def find_by_name(n)
      find {|s| s.name == n}
    end

    def update_services
      dependencies = inject([]) do |deps,svc|
        deps | svc.dependencies(true)
      end
      @all_services = dependencies|@_dc_obj
    end

    def inspect
      %Q{#<#{self.class.inspect} #{@_dc_obj.collect{|s| s.name}.join(', ')}>}
    end

    def call_graph
      @call_graph ||= inject(Hash.new {|h,k| h[k]=[]}) do |graph, svc|
        graph.merge(svc.call_graph)
      end
    end
    
    def run
      return [] if empty?

      @call_graph = nil # Make sure it's getting the latest graph
      services = call_order

      services.collect do |service|
        service.name if service.start!
      end
    end
    
    def stop
      changed_names = collect {|s| s.name if s.alive?}.compact
      each do |svc|
        Freyr.logger.debug('stopping service') {svc.name}
        svc.stop!
      end
      
      changed_names
    end
    
    def restart
      names = collect {|s| s.name}
      
      each do |s|
        Freyr.logger.debug('restart service') {s.name}
        s.restart!
        names.delete(s.name)
      end
      
      names
    end

    def tsort_each_node &blk
      call_graph.keys.each(&blk)
    end

    def tsort_each_child(node, &blk)
      call_graph[node].each(&blk)
    end
    alias call_order tsort

  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
freyr-0.5.2	lib/freyr/service_group.rb
freyr-0.5.1	lib/freyr/service_group.rb
freyr-0.5.0	lib/freyr/service_group.rb