# NewRelic Agent instrumentation for WebServices

# Note Action Web Service is removed from default package in rails 2.0
if defined? ActionWebService

  # instrumentation for Web Service martialing - XML RPC
  ActionWebService::Protocol::XmlRpc::XmlRpcProtocol.class_eval do
    add_method_tracer :decode_request, "WebService/Xml Rpc/XML Decode"
    add_method_tracer :encode_request, "WebService/Xml Rpc/XML Encode"
    add_method_tracer :decode_response, "WebService/Xml Rpc/XML Decode"
    add_method_tracer :encode_response, "WebService/Xml Rpc/XML Encode"
  end

  # instrumentation for Web Service martialing - Soap
  ActionWebService::Protocol::Soap::SoapProtocol.class_eval do
    add_method_tracer :decode_request, "WebService/Soap/XML Decode"
    add_method_tracer :encode_request, "WebService/Soap/XML Encode"
    add_method_tracer :decode_response, "WebService/Soap/XML Decode"
    add_method_tracer :encode_response, "WebService/Soap/XML Encode"
  end

  ActionController::Base.class_eval do
    if method_defined? :perform_invocation
      add_method_tracer :perform_invocation, 'WebService/#{controller_name}/#{args.first}'
    end
  end if defined? ActionController::Base
end