root/tags/RELEASE_1_0_0/lib/SOAPStreamHandler.rb

=begin
SOAP4R
Copyright (C) 2000 NAKAMURA Hiroshi.

This program is free software; you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by the Free Software
Foundation; either version 2 of the License, or (at your option) any later
version.

This program is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
PRATICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with
this program; if not, write to the Free Software Foundation, Inc., 675 Mass
Ave, Cambridge, MA 02139, USA.
=end

require 'socket'
require 'timeout'

require 'SOAP'
require 'http-access'
require 'uri'

class SOAPStreamHandler
  public

  attr_reader :endPoint

  def initialize( endPoint )
    @endPoint = endPoint
  end
end


class SOAPHTTPPostStreamHandler < SOAPStreamHandler
  public
  
  MediaType = 'text/xml'

  NofRetry = 10       # [times]
  CallTimeout = 300   # [sec]
  ReadTimeout = 300   # [sec]

  def initialize( endPointUri, proxy = nil )
    super( endPointUri )
    @server = URIModule::URI.create( endPointUri )
    @proxy = proxy
  end

  def send( methodNamespace, methodName, soapString )
    begin
      s = sendPOST( methodNamespace, methodName, soapString )
    rescue PostUnavailableError
      begin
        s = sendMPOST( methodNamespace, methodName, soapString )
      rescue MPostUnavailableError
        raise HTTPStreamError.new( $! )
      end
    end
  end

  private

  def sendPOST( methodNamespace, methodName, soapString )
    retryNo = NofRetry
    drv = nil
    begin
      drv = HTTPAccess.new( @server.host, @server.port, @proxy )
    rescue
      retryNo -= 1
      if retryNo > 0
        puts 'Retrying connection ...' if $DEBUG
        retry
      end
      raise
    end

    action = methodNamespace.dup << '#' << methodName
    requestHeaders = { 'SOAPAction' => action, 'Content-Type' => MediaType }

    begin
      timeout( CallTimeout ) do
        drv.request_post( @server.path, soapString, requestHeaders )
        rh = drv.get_header
        responseHeaders = {}
        rh.each do | header |
          /^([^:]+):\s*([^\r]*)\r?\n/ =~ header
          responseHeaders[ $1.downcase ] = $2
        end
        if ( drv.code == '405' )
          # 405: Method Not Allowed
          raise PostUnavailableError.new( "#{ drv.code }: #{ drv.message }" )
        elsif ( drv.code != '200' )
          raise HTTPStreamError.new( "#{ drv.code }: #{ drv.message }" )
        elsif ( !responseHeaders.has_key?( 'content-type' ))
          raise HTTPStreamError.new( 'Content-type not found.' )
        elsif ( /^#{ MediaType }(?:;.*)?/ !~ responseHeaders[ 'content-type' ] )
          raise HTTPStreamError.new( 'Illegal content-type: ' << responseHeaders[ 'content-type' ] )
        end
      end
    rescue TimeoutError
      raise HTTPStreamError.new( 'Call timeout' )
    end

    receiveString = ''
    begin
      timeout( ReadTimeout ) do
        drv.get_data( 8124 ) do | data |
          receiveString << data
        end
      end
    rescue TimeoutError
      raise HTTPStreamError.new( 'Read timeout' )
    end

    receiveString
  end

  def sendMPOST( methodNamespace, methodName, soapString )
    raise NotImplementError.new()

    s = nil
    if (( status == '501' ) or ( status == '510' ))
      # 501: Not Implemented
      # 510: Not Extended
      raise MPostUnavailableError.new( "Status: #{ status }, Reason-phrase: #{ reason }" )
    elsif ( status != '200' )
      raise HTTPStreamError.new( "#{ status }: #{ reason }" )
    end
  end
end