dtk-node-agent 0.7.7 → 0.8.0

data/mcollective_additions/plugins/v2.2/agent/sync_agent_code.ddl

@@ -1,10 +0,0 @@
-metadata    :name        => "sync agents code",
-            :description => "Agent to sync agent's code from git repo",
-            :author      => "Reactor8",
-            :license     => "",
-            :version     => "",
-            :url         => "",
-            :timeout     => 600
-action "run", :description => "sync agents code" do
-end
-

data/mcollective_additions/plugins/v2.2/agent/sync_agent_code.rb

@@ -1,85 +0,0 @@
-require 'rubygems'
-require 'puppet'
-require 'grit'
-require 'tempfile'
-require 'fileutils'
-
-AGENT_MCOLLECTIVE_LOCATION = "#{::MCollective::Config.instance.libdir.join}/mcollective/agent/"
-
-module MCollective
-  module Agent
-    class Sync_agent_code < RPC::Agent
-      def initialize()
-        super()
-        @log = Log.instance
-        @reply_data = nil
-      end
-      
-      def old_run_action
-        ret ||= Response.new()
-        #git_server_url = request[:git_server_url]
-        raise "git server is not set in facts" unless git_server_url = Facts["git-server"]
-        branch = request[:branch]
-        cmd_opts = {:raise => true, :timeout => 60}
-        
-        begin
-          # Amar: if git repo dir exists pull code, otherwise (first converge case) clone agent's project from git
-          if File.directory?(AgentGitPath)
-            grit_repo = ::Grit::Repo.new(AgentGitPath)
-            grit_repo.git.send(:pull, cmd_opts)
-            @log.info("Latest agent code pulled from GIT Repositoy.")
-          else
-            clone_args = [git_server_url, AgentGitPath]
-            clone_args += ["-b", branch] if branch
-            ::Grit::Git.new("").clone(cmd_opts, *clone_args)
-            @log.info("Agent project successfully cloned from GIT Repository.")            
-          end
-          # Copy latest agents to mcollective agent's directory
-          agents = Dir.glob("#{AgentGitPath}/mcollective_additions/plugins/v2.2/agent/*")
-          FileUtils.cp_r(agents, AgentMcollectivePath)
-          @log.info("Agent files copied to mcollective destination.")
-          # System call to restart mcollective
-          system("sudo /etc/init.d/mcollective restart")
-          @log.info("mcollective system restart completed.")
-
-          ret.set_status_succeeded!()
-        rescue Exception => e
-          @log.error("Error in syncing agent's code with GIT Repository: '#{git_server_url}'. Error: #{e}")
-          ret.set_status_failed!()
-          error_info = { :errors => { :message => "Error in syncing agent's code with GIT Repository: '#{git_server_url}'." } }
-          ret.merge!(error_info)
-        end
-
-        return ret
-      end
-
-   end
-      #TODO: this should be common accross Agents
-      class Response < Hash
-        def initialize(hash={})
-          super()
-          self.merge!(hash)
-          self[:status] = :unknown unless hash.has_key?(:status)
-        end
-
-        def to_hash()
-          Hash.new.merge(self)
-        end
-
-        def failed?()
-          self[:status] == :failed
-        end
-
-        def set_status_failed!()
-          self[:status] = :failed
-        end
-        def set_status_succeeded!()
-          self[:status] = :succeeded
-        end
-        def set_dynamic_attributes!(dynamic_attributes)
-          self[:dynamic_attributes] = dynamic_attributes
-        end
-      end
-  end
-end
-

data/mcollective_additions/plugins/v2.2/agent/tail.ddl

@@ -1,11 +0,0 @@
-metadata    :name        => "tailing log",
-            :description => "Agent to tail log",
-            :author      => "Reactor8",
-            :license     => "",
-            :version     => "",
-            :url         => "",
-            :timeout     => 2
-action "get_log", :description => "returns log content" do
-end
-action "grep", :description => "returns log content from multiple nodes" do
-end

data/mcollective_additions/plugins/v2.2/agent/tail.rb

@@ -1,67 +0,0 @@
-module MCollective
-  module Agent
-    class Tail < RPC::Agent
-
-      # number of lines that will be returned on first request
-      BATCH_SIZE_OF_LOG = 50
-
-      action "get_log" do
-        begin
-          unless File.exists? request[:log_path]
-            reply[:data]  =    { :error => "File #{request[:log_path]} not found on given node."}
-            reply[:pbuilderid] = Facts["pbuilderid"]
-            reply[:status] = :ok
-            return
-          end
-
-          # returns total number of lines in file, one is to start next iteration with new line
-          last_line  = `wc -l #{request[:log_path]} | awk '{print $1}'`.to_i + 1
-          # if there is start line from CLI request we use it, if not we take last BATCH_SIZE_OF_LOG lines
-          if request[:start_line].empty?
-            # If BATCH_SIZE_OF_LOG is bigger than last_line, then start line will be 0
-            start_line = (last_line > BATCH_SIZE_OF_LOG) ? last_line-BATCH_SIZE_OF_LOG : 0
-          else
-            start_line = request[:start_line]
-          end
-          
-          # returns needed lines
-          if (request[:grep_option].nil? || request[:grep_option].empty?)
-            output = `tail -n +#{start_line} #{request[:log_path]}`
-          else
-            output = `tail -n +#{start_line} #{request[:log_path]} | grep #{request[:grep_option]}`
-          end
-
-          reply[:data]      = { :output => output, :last_line => last_line }
-          reply[:pbuilderid] = Facts["pbuilderid"]
-          reply[:status]    = :ok
-        rescue Exception => e
-          Log.error e
-        end
-      end
-
-      action "grep" do
-        begin
-          unless File.exists? request[:log_path]
-            reply[:data]  =    { :error => "File #{request[:log_path]} not found on given node."}
-            reply[:pbuilderid] = Facts["pbuilderid"]
-            reply[:status] = :ok
-            return
-          end          
-          
-          # returns needed lines
-          if (request[:stop_on_first_match].empty? || request[:stop_on_first_match].nil? || request[:stop_on_first_match].eql?('false'))
-            output = `more #{request[:log_path]} | grep #{request[:grep_pattern]}`
-          else
-            output = `more #{request[:log_path]} | grep #{request[:grep_pattern]} | tail -1`
-          end
-          reply[:data]      = { :output => output}
-          reply[:pbuilderid] = Facts["pbuilderid"]
-          reply[:status]    = :ok
-        rescue Exception => e
-          Log.error e
-        end
-      end
-
-    end
-  end
-end

data/mcollective_additions/plugins/v2.2/audit/logfile.rb

@@ -1,26 +0,0 @@
-module MCollective
-  module RPC
-    # An audit plugin that just logs to a file
-    #
-    # You can configure which file it logs to with the setting
-    #
-    #   plugin.rpcaudit.logfile
-
-    class Logfile<Audit
-      require 'pp'
-
-      def audit_request(request, connection)
-        logfile = Config.instance.pluginconf["rpcaudit.logfile"] || "/var/log/mcollective-audit.log"
-
-        now = Time.now
-        # Already told timezone to be in UTC so we don't look it up again
-        # This avoids platform specific timezone representation issues
-        now_iso8601 = now.utc.strftime("%Y-%m-%d %H:%M:%S UTC")
-
-        File.open(logfile, "a") do |f|
-          f.puts("[#{now_iso8601}] reqid=#{request.uniqid}: reqtime=#{request.time} caller=#{request.caller}@#{request.sender} agent=#{request.agent} action=#{request.action} data=#{request.data.pretty_print_inspect}")
-        end
-      end
-    end
-  end
-end

data/mcollective_additions/plugins/v2.2/connector/r8stomp.rb

@@ -1,238 +0,0 @@
-require 'stomp'
-
-module MCollective
-    module Connector
-        # Handles sending and receiving messages over the Stomp protocol
-        #
-        # This plugin supports version 1.1 or 1.1.6 and newer of the Stomp rubygem
-        # the versions between those had multi threading issues.
-        #
-        # For all versions you can configure it as follows:
-        #
-        #    connector = stomp
-        #    plugin.stomp.host = stomp.your.net
-        #    plugin.stomp.port = 6163
-        #    plugin.stomp.user = you
-        #    plugin.stomp.password = secret
-        #
-        # All of these can be overriden per user using environment variables:
-        #
-        #    STOMP_SERVER, STOMP_PORT, STOMP_USER, STOMP_PASSWORD
-        #
-        # Version 1.1.6 onward support supplying multiple connections and it will
-        # do failover between these servers, you can configure it as follows:
-        #
-        #     connector = stomp
-        #     plugin.stomp.pool.size = 2
-        #
-        #     plugin.stomp.pool.host1 = stomp1.your.net
-        #     plugin.stomp.pool.port1 = 6163
-        #     plugin.stomp.pool.user1 = you
-        #     plugin.stomp.pool.password1 = secret
-        #     plugin.stomp.pool.ssl1 = true
-        #
-        #     plugin.stomp.pool.host2 = stomp2.your.net
-        #     plugin.stomp.pool.port2 = 6163
-        #     plugin.stomp.pool.user2 = you
-        #     plugin.stomp.pool.password2 = secret
-        #     plugin.stomp.pool.ssl2 = false
-        #
-        # Using this method you can supply just STOMP_USER and STOMP_PASSWORD
-        # you have to supply the hostname for each pool member in the config.
-        # The port will default to 6163 if not specified.
-        #
-        # In addition you can set the following options but only when using
-        # pooled configuration:
-        #
-        #     plugin.stomp.pool.initial_reconnect_delay = 0.01
-        #     plugin.stomp.pool.max_reconnect_delay = 30.0
-        #     plugin.stomp.pool.use_exponential_back_off = true
-        #     plugin.stomp.pool.back_off_multiplier = 2
-        #     plugin.stomp.pool.max_reconnect_attempts = 0
-        #     plugin.stomp.pool.randomize = false
-        #     plugin.stomp.pool.timeout = -1
-        #
-        # For versions of ActiveMQ that supports message priorities
-        # you can set a priority, this will cause a "priority" header
-        # to be emitted if present:
-        #
-        #     plugin.stomp.priority = 4
-        #
-        class Stomp<Base
-          attr_reader :connection
-
-          #modifications so connect at initialization and there is not multiple connections per thread
-          def initialize
-            @config = Config.instance
-            @subscriptions = []
-            ###BEGIN R8 CUSTOMIZATION 
-            if Thread.current[:stomp_client]
-              @connection = Thread.current[:stomp_client]
-            else
-              connect()
-              Thread.current[:stomp_client] = @connection
-            end
-            ###END R8 CUSTOMIZATION 
-          end
-
-          def disconnect
-            Log.debug("Disconnecting from Stomp")
-            @connection.disconnect
-          end
-
-            # Connects to the Stomp middleware
-            def connect
-                if @connection
-                    Log.debug("Already connection, not re-initializing connection")
-                    return
-                end
-                begin
-                    host = nil
-                    port = nil
-                    user = nil
-                    password = nil
-                    @base64 = get_bool_option("stomp.base64", false)
-                    @msgpriority = get_option("stomp.priority", 0).to_i
-
-                    # Maintain backward compat for older stomps
-                    unless @config.pluginconf.include?("stomp.pool.size")
-                        host = get_env_or_option("STOMP_SERVER", "stomp.host")
-                        port = get_env_or_option("STOMP_PORT", "stomp.port", 6163).to_i
-                        user = get_env_or_option("STOMP_USER", "stomp.user")
-                        password = get_env_or_option("STOMP_PASSWORD", "stomp.password")
-
-                        Log.debug("Connecting to #{host}:#{port}")
-                        @connection = ::Stomp::Connection.new(user, password, host, port, true)
-                    else
-                        pools = @config.pluginconf["stomp.pool.size"].to_i
-                        hosts = []
-
-                        1.upto(pools) do |poolnum|
-                            host = {}
-
-                            host[:host] = get_option("stomp.pool.host#{poolnum}")
-                            host[:port] = get_option("stomp.pool.port#{poolnum}", 6163).to_i
-                            host[:login] = get_env_or_option("STOMP_USER", "stomp.pool.user#{poolnum}")
-                            host[:passcode] = get_env_or_option("STOMP_PASSWORD", "stomp.pool.password#{poolnum}")
-                            host[:ssl] = get_bool_option("stomp.pool.ssl#{poolnum}", false)
-
-                            Log.debug("Adding #{host[:host]}:#{host[:port]} to the connection pool")
-                            hosts << host
-                        end
-
-                        raise "No hosts found for the STOMP connection pool" if hosts.size == 0
-
-                        connection = {:hosts => hosts}
-
-                        # Various STOMP gem options, defaults here matches defaults for 1.1.6 the meaning of
-                        # these can be guessed, the documentation isn't clear
-                        connection[:initial_reconnect_delay] = get_option("stomp.pool.initial_reconnect_delay", 0.01).to_f
-                        connection[:max_reconnect_delay] = get_option("stomp.pool.max_reconnect_delay", 30.0).to_f
-                        connection[:use_exponential_back_off] = get_bool_option("stomp.pool.use_exponential_back_off", true)
-                        connection[:back_off_multiplier] = get_bool_option("stomp.pool.back_off_multiplier", 2).to_i
-                        connection[:max_reconnect_attempts] = get_option("stomp.pool.max_reconnect_attempts", 0)
-                        connection[:randomize] = get_bool_option("stomp.pool.randomize", false)
-                        connection[:backup] = get_bool_option("stomp.pool.backup", false)
-                        connection[:timeout] = get_option("stomp.pool.timeout", -1).to_i
-
-                        @connection = ::Stomp::Connection.new(connection)
-                    end
-                rescue Exception => e
-                    raise("Could not connect to Stomp Server: #{e}")
-                end
-            end
-
-            # Receives a message from the Stomp connection
-            def receive
-                Log.debug("Waiting for a message from Stomp")
-                msg = @connection.receive
-
-                # STOMP puts the payload in the body variable, pass that
-                # into the payload of MCollective::Request and discard all the
-                # other headers etc that stomp provides
-                if @base64
-                    Request.new(SSL.base64_decode(msg.body))
-                else
-                    Request.new(msg.body)
-                end
-            end
-
-            # Sends a message to the Stomp connection
-            def send(target, msg)
-                Log.debug("Sending a message to Stomp target '#{target}'")
-
-                msg = SSL.base64_encode(msg) if @base64
-
-                # deal with deprecation warnings in newer stomp gems
-                if @connection.respond_to?("publish")
-                    @connection.publish(target, msg, msgheaders)
-                else
-                    @connection.send(target, msg, msgheaders)
-                end
-            end
-
-            # Subscribe to a topic or queue
-            def subscribe(source)
-                unless @subscriptions.include?(source)
-                    Log.debug("Subscribing to #{source}")
-                    @connection.subscribe(source)
-                    @subscriptions << source
-                end
-            end
-
-            # Subscribe to a topic or queue
-            def unsubscribe(source)
-                Log.debug("Unsubscribing from #{source}")
-                @connection.unsubscribe(source)
-                @subscriptions.delete(source)
-            end
-
-                      private
-            def msgheaders
-                headers = {}
-                headers = {"priority" => @msgpriority} if  (@msgpriority and @msgpriority > 0)
-
-                return headers
-            end
-
-            # looks in the environment first then in the config file
-            # for a specific option, accepts an optional default.
-            #
-            # raises an exception when it cant find a value anywhere
-            def get_env_or_option(env, opt, default=nil)
-                return ENV[env] if ENV.include?(env)
-                return @config.pluginconf[opt] if @config.pluginconf.include?(opt)
-                return default if default
-
-                raise("No #{env} environment or plugin.#{opt} configuration option given")
-            end
-
-            # looks for a config option, accepts an optional default
-            #
-            # raises an exception when it cant find a value anywhere
-            def get_option(opt, default=nil)
-                return @config.pluginconf[opt] if @config.pluginconf.include?(opt)
-                return default if default
-
-                raise("No plugin.#{opt} configuration option given")
-            end
-
-            # gets a boolean option from the config, supports y/n/true/false/1/0
-            def get_bool_option(opt, default)
-                return default unless @config.pluginconf.include?(opt)
-
-                val = @config.pluginconf[opt]
-
-                if val =~ /^1|yes|true/
-                    return true
-                elsif val =~ /^0|no|false/
-                    return false
-                else
-                    return default
-                end
-            end
-        end
-    end
-end
-
-# vi:tabstop=4:expandtab:ai

data/mcollective_additions/plugins/v2.2/connector/stomp.rb

@@ -1,349 +0,0 @@
-require 'stomp'
-
-module MCollective
-  module Connector
-    # Handles sending and receiving messages over the Stomp protocol
-    #
-    # This plugin supports version 1.1 or 1.1.6 and newer of the Stomp rubygem
-    # the versions between those had multi threading issues.
-    #
-    # For all versions you can configure it as follows:
-    #
-    #    connector = stomp
-    #    plugin.stomp.host = stomp.your.net
-    #    plugin.stomp.port = 6163
-    #    plugin.stomp.user = you
-    #    plugin.stomp.password = secret
-    #
-    # All of these can be overriden per user using environment variables:
-    #
-    #    STOMP_SERVER, STOMP_PORT, STOMP_USER, STOMP_PASSWORD
-    #
-    # Version 1.1.6 onward support supplying multiple connections and it will
-    # do failover between these servers, you can configure it as follows:
-    #
-    #     connector = stomp
-    #     plugin.stomp.pool.size = 2
-    #
-    #     plugin.stomp.pool.host1 = stomp1.your.net
-    #     plugin.stomp.pool.port1 = 6163
-    #     plugin.stomp.pool.user1 = you
-    #     plugin.stomp.pool.password1 = secret
-    #     plugin.stomp.pool.ssl1 = true
-    #
-    #     plugin.stomp.pool.host2 = stomp2.your.net
-    #     plugin.stomp.pool.port2 = 6163
-    #     plugin.stomp.pool.user2 = you
-    #     plugin.stomp.pool.password2 = secret
-    #     plugin.stomp.pool.ssl2 = false
-    #
-    # Using this method you can supply just STOMP_USER and STOMP_PASSWORD
-    # you have to supply the hostname for each pool member in the config.
-    # The port will default to 6163 if not specified.
-    #
-    # In addition you can set the following options but only when using
-    # pooled configuration:
-    #
-    #     plugin.stomp.pool.initial_reconnect_delay = 0.01
-    #     plugin.stomp.pool.max_reconnect_delay = 30.0
-    #     plugin.stomp.pool.use_exponential_back_off = true
-    #     plugin.stomp.pool.back_off_multiplier = 2
-    #     plugin.stomp.pool.max_reconnect_attempts = 0
-    #     plugin.stomp.pool.randomize = false
-    #     plugin.stomp.pool.timeout = -1
-    #
-    # You can set the initial connetion timeout - this is when your stomp server is simply
-    # unreachable - after which it would failover to the next in the pool:
-    #
-    #     plugin.stomp.pool.connect_timeout = 30
-    #
-    # For versions of ActiveMQ that supports message priorities
-    # you can set a priority, this will cause a "priority" header
-    # to be emitted if present:
-    #
-    #     plugin.stomp.priority = 4
-    #
-    class Stomp<Base
-      # Older stomp gems do not have these error classes, in order to be able to
-      # handle these exceptions if they are present and still support older gems
-      # we're assigning the constants to a dummy exception that will never be thrown
-      # by us.  End result is that the code catching these exceptions become noops on
-      # older gems but on newer ones they become usable and handle those new errors
-      # intelligently
-      class DummyError<RuntimeError; end
-
-      ::Stomp::Error = DummyError unless defined?(::Stomp::Error)
-      ::Stomp::Error::NoCurrentConnection = DummyError unless defined?(::Stomp::Error::NoCurrentConnection)
-      ::Stomp::Error::DuplicateSubscription = DummyError unless defined?(::Stomp::Error::DuplicateSubscription)
-
-      # Class for Stomp 1.1.9 callback based logging
-      class EventLogger
-        def on_connecting(params=nil)
-          Log.info("Connection attempt %d to %s" % [params[:cur_conattempts], stomp_url(params)])
-        rescue
-        end
-
-        def on_connected(params=nil)
-          Log.info("Conncted to #{stomp_url(params)}")
-        rescue
-        end
-
-        def on_disconnect(params=nil)
-          Log.info("Disconnected from #{stomp_url(params)}")
-        rescue
-        end
-
-        def on_connectfail(params=nil)
-          Log.info("Connection to #{stomp_url(params)} failed on attempt #{params[:cur_conattempts]}")
-        rescue
-        end
-
-        def on_miscerr(params, errstr)
-          Log.error("Unexpected error on connection #{stomp_url(params)}: #{errstr}")
-        rescue
-        end
-
-        def on_ssl_connecting(params)
-          Log.info("Performing SSL connection to #{stomp_url(params)}")
-        rescue
-        end
-
-        def on_ssl_connected(params)
-          Log.info("Connected SSL socket #{stomp_url(params)}")
-        rescue
-        end
-
-        def stomp_url(params)
-          "%s://%s@%s:%d" % [ params[:cur_ssl] ? "stomp+ssl" : "stomp", params[:cur_login], params[:cur_host], params[:cur_port]]
-        end
-      end
-
-      attr_reader :connection
-
-      def initialize
-        Log.info("MCollective 2.2.x will be the last to fully support the 'stomp' connector, please migrate to the 'activemq' or 'rabbitmq' connector")
-
-        @config = Config.instance
-        @subscriptions = []
-      end
-
-      # Connects to the Stomp middleware
-      def connect(connector = ::Stomp::Connection)
-        if @connection
-          Log.debug("Already connection, not re-initializing connection")
-          return
-        end
-
-        begin
-          host = nil
-          port = nil
-          user = nil
-          password = nil
-          @base64 = get_bool_option("stomp.base64", false)
-          @msgpriority = get_option("stomp.priority", 0).to_i
-
-          # Maintain backward compat for older stomps
-          unless @config.pluginconf.include?("stomp.pool.size")
-            host = get_env_or_option("STOMP_SERVER", "stomp.host")
-            port = get_env_or_option("STOMP_PORT", "stomp.port", 6163).to_i
-            user = get_env_or_option("STOMP_USER", "stomp.user")
-            password = get_env_or_option("STOMP_PASSWORD", "stomp.password")
-
-            Log.debug("Connecting to #{host}:#{port}")
-            @connection = connector.new(user, password, host, port, true)
-          else
-            pools = @config.pluginconf["stomp.pool.size"].to_i
-            hosts = []
-
-            1.upto(pools) do |poolnum|
-              host = {}
-
-              host[:host] = get_option("stomp.pool.host#{poolnum}")
-              host[:port] = get_option("stomp.pool.port#{poolnum}", 6163).to_i
-              host[:login] = get_env_or_option("STOMP_USER", "stomp.pool.user#{poolnum}")
-              host[:passcode] = get_env_or_option("STOMP_PASSWORD", "stomp.pool.password#{poolnum}")
-              host[:ssl] = get_bool_option("stomp.pool.ssl#{poolnum}", false)
-
-              Log.debug("Adding #{host[:host]}:#{host[:port]} to the connection pool")
-              hosts << host
-            end
-
-            raise "No hosts found for the STOMP connection pool" if hosts.size == 0
-
-            connection = {:hosts => hosts}
-
-            # Various STOMP gem options, defaults here matches defaults for 1.1.6 the meaning of
-            # these can be guessed, the documentation isn't clear
-            connection[:initial_reconnect_delay] = get_option("stomp.pool.initial_reconnect_delay", 0.01).to_f
-            connection[:max_reconnect_delay] = get_option("stomp.pool.max_reconnect_delay", 30.0).to_f
-            connection[:use_exponential_back_off] = get_bool_option("stomp.pool.use_exponential_back_off", true)
-            connection[:back_off_multiplier] = get_bool_option("stomp.pool.back_off_multiplier", 2).to_i
-            connection[:max_reconnect_attempts] = get_option("stomp.pool.max_reconnect_attempts", 0).to_i
-            connection[:randomize] = get_bool_option("stomp.pool.randomize", false)
-            connection[:backup] = get_bool_option("stomp.pool.backup", false)
-            connection[:timeout] = get_option("stomp.pool.timeout", -1).to_i
-            connection[:connect_timeout] = Integer(get_option("stomp.pool.connect_timeout", 30))
-            connection[:reliable] = true
-
-            connection[:logger] = EventLogger.new
-
-            @connection = connector.new(connection)
-          end
-        rescue Exception => e
-          raise("Could not connect to Stomp Server: #{e}")
-        end
-      end
-
-      # Receives a message from the Stomp connection
-      def receive
-        Log.debug("Waiting for a message from Stomp")
-
-        # When the Stomp library > 1.2.0 is mid reconnecting due to its reliable connection
-        # handling it sets the connection to closed.  If we happen to be receiving at just
-        # that time we will get an exception warning about the closed connection so handling
-        # that here with a sleep and a retry.
-        begin
-          msg = @connection.receive
-        rescue ::Stomp::Error::NoCurrentConnection
-          sleep 1
-          retry
-        end
-
-        Message.new(msg.body, msg, :base64 => @base64, :headers => msg.headers)
-      end
-
-      # Sends a message to the Stomp connection
-      def publish(msg)
-        msg.base64_encode! if @base64
-
-        raise "Cannot set specific reply to targets with the STOMP plugin" if msg.reply_to
-
-        if msg.type == :direct_request
-          msg.discovered_hosts.each do |node|
-            target = make_target(msg.agent, msg.type, msg.collective, node)
-
-            Log.debug("Sending a direct message to STOMP target '#{target}'")
-
-            publish_msg(target, msg.payload)
-          end
-        else
-          target = make_target(msg.agent, msg.type, msg.collective)
-
-          Log.debug("Sending a broadcast message to STOMP target '#{target}'")
-
-          publish_msg(target, msg.payload)
-        end
-      end
-
-      # Subscribe to a topic or queue
-      def subscribe(agent, type, collective)
-        source = make_target(agent, type, collective)
-
-        unless @subscriptions.include?(source)
-          Log.debug("Subscribing to #{source}")
-          @connection.subscribe(source)
-          @subscriptions << source
-        end
-      rescue ::Stomp::Error::DuplicateSubscription
-        Log.debug("Received subscription for #{source[:name]} but already had a subscription, ignoring")
-      end
-
-      # Actually sends the message to the middleware
-      def publish_msg(target, msg)
-        # deal with deprecation warnings in newer stomp gems
-        if @connection.respond_to?("publish")
-          @connection.publish(target, msg, msgheaders)
-        else
-          @connection.send(target, msg, msgheaders)
-        end
-      end
-
-      # Subscribe to a topic or queue
-      def unsubscribe(agent, type, collective)
-        source = make_target(agent, type, collective)
-
-        Log.debug("Unsubscribing from #{source}")
-        @connection.unsubscribe(source)
-        @subscriptions.delete(source)
-      end
-
-      # Disconnects from the Stomp connection
-      def disconnect
-        Log.debug("Disconnecting from Stomp")
-        @connection.disconnect
-      end
-
-      def msgheaders
-        headers = {}
-        headers = {"priority" => @msgpriority} if @msgpriority > 0
-
-        return headers
-      end
-
-      # looks in the environment first then in the config file
-      # for a specific option, accepts an optional default.
-      #
-      # raises an exception when it cant find a value anywhere
-      def get_env_or_option(env, opt, default=nil)
-        return ENV[env] if ENV.include?(env)
-        return @config.pluginconf[opt] if @config.pluginconf.include?(opt)
-        return default if default
-
-        raise("No #{env} environment or plugin.#{opt} configuration option given")
-      end
-
-      # looks for a config option, accepts an optional default
-      #
-      # raises an exception when it cant find a value anywhere
-      def get_option(opt, default=nil)
-        return @config.pluginconf[opt] if @config.pluginconf.include?(opt)
-        return default if default
-
-        raise("No plugin.#{opt} configuration option given")
-      end
-
-      # gets a boolean option from the config, supports y/n/true/false/1/0
-      def get_bool_option(opt, default)
-        return default unless @config.pluginconf.include?(opt)
-
-        val = @config.pluginconf[opt]
-
-        if val =~ /^1|yes|true/
-          return true
-        elsif val =~ /^0|no|false/
-          return false
-        else
-          return default
-        end
-      end
-
-      def make_target(agent, type, collective, target_node=nil)
-        raise("Unknown target type #{type}") unless [:directed, :broadcast, :reply, :request, :direct_request].include?(type)
-        raise("Unknown collective '#{collective}' known collectives are '#{@config.collectives.join ', '}'") unless @config.collectives.include?(collective)
-
-        prefix = @config.topicprefix
-
-        case type
-          when :reply
-            suffix = :reply
-          when :broadcast
-            suffix = :command
-          when :request
-            suffix = :command
-          when :direct_request
-            agent = nil
-            prefix = @config.queueprefix
-            suffix = Digest::MD5.hexdigest(target_node)
-          when :directed
-            agent = nil
-            prefix = @config.queueprefix
-            # use a md5 since hostnames might have illegal characters that
-            # the middleware dont understand
-            suffix = Digest::MD5.hexdigest(@config.identity)
-        end
-
-        ["#{prefix}#{collective}", agent, suffix].compact.join(@config.topicsep)
-      end
-    end
-  end
-end