Sha256: 250081708937c3c041678c1b2ab4199420c239fb5db667e0fe9dfcd5169672b4

Contents?: true

Size: 1.68 KB

Versions: 5

Compression:

Stored size: 1.68 KB

Contents

require 'mongrel' if Puppet.features.mongrel?

require 'puppet/network/http/mongrel/rest'

class Puppet::Network::HTTP::Mongrel
    def initialize(args = {})
        @listening = false
    end
    
    def listen(args = {})
        raise ArgumentError, ":handlers must be specified." if !args[:handlers] or args[:handlers].empty?
        raise ArgumentError, ":protocols must be specified." if !args[:protocols] or args[:protocols].empty?
        raise ArgumentError, ":address must be specified." unless args[:address]
        raise ArgumentError, ":port must be specified." unless args[:port]
        raise "Mongrel server is already listening" if listening?

        @protocols = args[:protocols]
        @handlers = args[:handlers]
        @server = Mongrel::HttpServer.new(args[:address], args[:port]) 
        setup_handlers

        @listening = true
        @server.run
    end
    
    def unlisten
        raise "Mongrel server is not listening" unless listening?
        @server.stop
        @server = nil
        @listening = false
    end
    
    def listening?
        @listening
    end
    
  private
  
    def setup_handlers
        @protocols.each do |protocol|
            klass = class_for_protocol(protocol)
            @handlers.each do |handler|
                @server.register('/' + handler.to_s, klass.new(:server => @server, :handler => handler))
                @server.register('/' + handler.to_s + 's', klass.new(:server => @server, :handler => handler))
            end
        end
    end
  
    def class_for_protocol(protocol)
        return Puppet::Network::HTTP::MongrelREST if protocol.to_sym == :rest
        raise ArgumentError, "Unknown protocol [#{protocol}]."
    end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version Path
puppet-0.24.9 lib/puppet/network/http/mongrel.rb
puppet-0.24.6 lib/puppet/network/http/mongrel.rb
puppet-0.24.7 lib/puppet/network/http/mongrel.rb
puppet-0.24.5 lib/puppet/network/http/mongrel.rb
puppet-0.24.8 lib/puppet/network/http/mongrel.rb