Contents

class TestLab
  class Node

    module Lifecycle

      # Iterates an array of arrays calling the specified method on all the
      # collections of objects
      def call_collections(collections, method_name)
        collections.each do |collection|
          call_methods(collection, method_name)
        end
      end

      # Calls the specified method on all the objects supplied
      def call_methods(objects, method_name)
        objects.each do |object|
          # @ui.stdout.puts(format_message(format_object(object, (method_name == :setup ? :green : :red))))

          if object.respond_to?(method_name)
            object.send(method_name)
          end
        end
      end

      # Bootstrap the node
      def node_setup
        node_setup_template = File.join(self.class.template_dir, 'node-setup.erb')
        self.ssh.bootstrap(ZTK::Template.render(node_setup_template))
      end

      # Setup the node.
      def setup
        @ui.logger.debug { "Node Setup: #{self.id} " }

        # @ui.stdout.puts(format_message(format_object(self, :green)))

        please_wait(:ui => @ui, :message => format_object_action(self, 'Setup', :green)) do
          node_setup

          if self.components.include?('resolv')
            build_resolv_conf
          end

          if self.components.include?('bind')
            bind_setup
          end
        end

        call_collections([self.networks, self.routers, self.containers], :setup)

        if self.components.include?('bind')
          bind_reload
        end

        true
      end

      # Teardown the node.
      def teardown
        @ui.logger.debug { "Node Teardown: #{self.id} " }

        # @ui.stdout.puts(format_message(format_object(self, :red)))

        call_collections([self.containers, self.routers, self.networks], :teardown)

        please_wait(:ui => @ui, :message => format_object_action(self, 'Teardown', :red)) do
        end

        true
      end

    end

  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
testlab-0.2.0	lib/testlab/node/lifecycle.rb