Package org.apache.axis.message

Source Code of org.apache.axis.message.SOAPEnvelope

/*
* Copyright 2001-2004 The Apache Software Foundation.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*      http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.axis.message;

import org.apache.axis.AxisFault;
import org.apache.axis.Constants;
import org.apache.axis.Message;
import org.apache.axis.MessageContext;
import org.apache.axis.schema.SchemaVersion;
import org.apache.axis.client.AxisClient;
import org.apache.axis.components.logger.LogFactory;
import org.apache.axis.configuration.NullProvider;
import org.apache.axis.encoding.DeserializationContext;
import org.apache.axis.encoding.DeserializationContextImpl;
import org.apache.axis.encoding.SerializationContext;
import org.apache.axis.soap.SOAPConstants;
import org.apache.axis.utils.Mapping;
import org.apache.axis.utils.Messages;
import org.apache.commons.logging.Log;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;

import javax.xml.namespace.QName;
import javax.xml.soap.SOAPException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.Vector;

/**
* Implementation of a SOAP Envelope
*/
public class SOAPEnvelope extends MessageElement
    implements javax.xml.soap.SOAPEnvelope
{
    protected static Log log =
        LogFactory.getLog(SOAPEnvelope.class.getName());

    private SOAPHeader header;
    private SOAPBody body;

    public Vector trailers = new Vector();
    private SOAPConstants soapConstants;
    private SchemaVersion schemaVersion = SchemaVersion.SCHEMA_2001;

    // This is a hint to any service description to tell it what
    // "type" of message we are.  This might be "request", "response",
    // or anything else your particular service descripton requires.
    //
    // This gets passed back into the service description during
    // deserialization
    public String messageType;

    public SOAPEnvelope()
    {
        this(true, SOAPConstants.SOAP11_CONSTANTS);
    }

    public SOAPEnvelope(SOAPConstants soapConstants)
    {
        this(true, soapConstants);
    }

    public SOAPEnvelope(SOAPConstants soapConstants,
                        SchemaVersion schemaVersion)
    {
        this(true, soapConstants, schemaVersion);
    }

    public SOAPEnvelope(boolean registerPrefixes, SOAPConstants soapConstants)
    {
        this (registerPrefixes, soapConstants, SchemaVersion.SCHEMA_2001);
    }
   
    public SOAPEnvelope(boolean registerPrefixes,
                        SOAPConstants soapConstants,
                        SchemaVersion schemaVersion)
    {   
        // FIX BUG http://nagoya.apache.org/bugzilla/show_bug.cgi?id=18108
        super(Constants.ELEM_ENVELOPE,
               Constants.NS_PREFIX_SOAP_ENV,
               (soapConstants != null) ? soapConstants.getEnvelopeURI() : Constants.DEFAULT_SOAP_VERSION.getEnvelopeURI());

        if (soapConstants == null)
          soapConstants = Constants.DEFAULT_SOAP_VERSION;
        // FIX BUG http://nagoya.apache.org/bugzilla/show_bug.cgi?id=18108       
       
        this.soapConstants = soapConstants;
        this.schemaVersion = schemaVersion;
        header = new SOAPHeader(this, soapConstants);
        body = new SOAPBody(this, soapConstants);

        if (registerPrefixes) {
            if (namespaces == null)
                namespaces = new ArrayList();

            namespaces.add(new Mapping(soapConstants.getEnvelopeURI(),
                                       Constants.NS_PREFIX_SOAP_ENV));
            namespaces.add(new Mapping(schemaVersion.getXsdURI(),
                                       Constants.NS_PREFIX_SCHEMA_XSD));
            namespaces.add(new Mapping(schemaVersion.getXsiURI(),
                                       Constants.NS_PREFIX_SCHEMA_XSI));
        }

        setDirty(true);
    }
   
    public SOAPEnvelope(InputStream input) throws SAXException {
        InputSource is = new InputSource(input);
        // FIX BUG http://nagoya.apache.org/bugzilla/show_bug.cgi?id=18108
        //header = new SOAPHeader(this, soapConstants); // soapConstants = null!
        header = new SOAPHeader(this, Constants.DEFAULT_SOAP_VERSION); // soapConstants = null!
        // FIX BUG http://nagoya.apache.org/bugzilla/show_bug.cgi?id=18108
        DeserializationContext dser = null ;
        AxisClient     tmpEngine = new AxisClient(new NullProvider());
        MessageContext msgContext = new MessageContext(tmpEngine);
        dser = new DeserializationContextImpl(is, msgContext,
                                              Message.REQUEST, this );
        dser.parse();
    }

    /**
     * Get the Message Type (REQUEST/RESPONSE)
     * @return message type
     */
    public String getMessageType()
    {
        return messageType;
    }

    /**
     * Set the Message Type (REQUEST/RESPONSE)
     * @param messageType
     */
    public void setMessageType(String messageType)
    {
        this.messageType = messageType;
    }

    /**
     * Get all the BodyElement's in the soap body
     * @return vector with body elements
     * @throws AxisFault
     */
    public Vector getBodyElements() throws AxisFault
    {
        if (body != null) {
            return body.getBodyElements();
        } else {
            return new Vector();
        }
    }

    /**
     * Return trailers
     * @return vector of some type
     */
    public Vector getTrailers()
    {
        return trailers;
    }

    /**
     * Get the first BodyElement in the SOAP Body
     * @return first Body Element
     * @throws AxisFault
     */
    public SOAPBodyElement getFirstBody() throws AxisFault
    {
        if (body == null) {
            return null;
        } else {
            return body.getFirstBody();
        }
    }

    /**
     * Get Headers
     * @return Vector containing Header's
     * @throws AxisFault
     */
    public Vector getHeaders() throws AxisFault
    {
        if (header != null) {
            return header.getHeaders();
        } else {
            return new Vector();
        }
    }

    /**
     * Get all the headers targeted at a list of actors.
     */
    public Vector getHeadersByActor(ArrayList actors)
    {
        if (header != null) {
            return header.getHeadersByActor(actors);
        } else {
            return new Vector();
        }
    }

    /**
     * Add a HeaderElement
     * @param hdr
     */
    public void addHeader(SOAPHeaderElement hdr)
    {
        if (header == null) {
            header = new SOAPHeader(this, soapConstants);
        }
        hdr.setEnvelope(this);
        header.addHeader(hdr);
        _isDirty = true;
    }

    /**
     * Add a SOAP Body Element
     * @param element
     */
    public void addBodyElement(SOAPBodyElement element)
    {
        if (body == null) {
            body = new SOAPBody(this, soapConstants);
        }
        element.setEnvelope(this);
        body.addBodyElement(element);

        _isDirty = true;
    }

    /**
     * Remove all headers
     */
    public void removeHeaders() {
        header = null;
    }

    /**
     * Set the SOAP Header
     * @param hdr
     */
    public void setHeader(SOAPHeader hdr) {
        header = hdr;
        try {
            header.setParentElement(this);
        } catch (SOAPException ex) {
            // class cast should never fail when parent is a SOAPEnvelope
            log.fatal(Messages.getMessage("exception00"), ex);
        }
    }

    /**
     * Remove a Header Element from SOAP Header
     * @param hdr
     */
    public void removeHeader(SOAPHeaderElement hdr)
    {
        if (header != null) {
            header.removeHeader(hdr);
            _isDirty = true;
        }
    }

    /**
     * Remove the SOAP Body
     */
    public void removeBody() {
        body = null;
    }

    /**
     * Set the soap body
     * @param body
     */
    public void setBody(SOAPBody body) {
        this.body = body;
        try {
            body.setParentElement(this);
        } catch (SOAPException ex) {
            // class cast should never fail when parent is a SOAPEnvelope
            log.fatal(Messages.getMessage("exception00"), ex);
        }
    }

    /**
     * Remove a Body Element from the soap body
     * @param element
     */
    public void removeBodyElement(SOAPBodyElement element)
    {
        if (body != null) {
            body.removeBodyElement(element);
            _isDirty = true;
        }
    }

    /**
     * Remove an element from the trailer
     * @param element
     */
    public void removeTrailer(MessageElement element)
    {
        if (log.isDebugEnabled())
            log.debug(Messages.getMessage("removeTrailer00"));
        trailers.removeElement(element);
        _isDirty = true;
    }

    /**
     * clear the elements in the soap body
     */
    public void clearBody()
    {
        if (body != null) {
            body.clearBody();
            _isDirty = true;
        }
    }

    /**
     * Add an element to the trailer
     * @param element
     */
    public void addTrailer(MessageElement element)
    {
        if (log.isDebugEnabled())
            log.debug(Messages.getMessage("removeTrailer00"));
        element.setEnvelope(this);
        trailers.addElement(element);
        _isDirty = true;
    }

    /**
     * Get a header by name (always respecting the currently in-scope
     * actors list)
     */
    public SOAPHeaderElement getHeaderByName(String namespace,
                                             String localPart)
        throws AxisFault
    {
        return getHeaderByName(namespace, localPart, false);
    }

    /**
     * Get a header by name, filtering for headers targeted at this
     * engine depending on the accessAllHeaders parameter.
     */
    public SOAPHeaderElement getHeaderByName(String namespace,
                                             String localPart,
                                             boolean accessAllHeaders)
        throws AxisFault
    {
        if (header != null) {
            return header.getHeaderByName(namespace,
                                          localPart,
                                          accessAllHeaders);
        } else {
            return null;
        }
    }

    /**
     * Get a body element given its name
     * @param namespace
     * @param localPart
     * @return
     * @throws AxisFault
     */
    public SOAPBodyElement getBodyByName(String namespace, String localPart)
        throws AxisFault
    {
        if (body == null) {
            return null;
        } else {
            return body.getBodyByName(namespace, localPart);
        }
    }

    /**
     * Get an enumeration of header elements given the namespace and localpart
     * @param namespace
     * @param localPart
     * @return
     * @throws AxisFault
     */
    public Enumeration getHeadersByName(String namespace, String localPart)
        throws AxisFault
    {
        return getHeadersByName(namespace, localPart, false);
    }

    /**
     * Return an Enumeration of headers which match the given namespace
     * and localPart.  Depending on the value of the accessAllHeaders
     * parameter, we will attempt to filter on the current engine's list
     * of actors.
     *
     * !!! NOTE THAT RIGHT NOW WE ALWAYS ASSUME WE'RE THE "ULTIMATE
     * DESTINATION" (i.e. we match on null actor).  IF WE WANT TO FULLY SUPPORT
     * INTERMEDIARIES WE'LL NEED TO FIX THIS.
     */
    public Enumeration getHeadersByName(String namespace, String localPart,
                                        boolean accessAllHeaders)
        throws AxisFault
    {
        if (header != null) {
            return header.getHeadersByName(namespace,
                                           localPart,
                                           accessAllHeaders);
        } else {
            return new Vector().elements();
        }
    }

    /** Should make SOAPSerializationException?
     */
    public void outputImpl(SerializationContext context)
        throws Exception
    {
        boolean oldPretty = context.getPretty();
        context.setPretty(true);

        // Register namespace prefixes.
        if (namespaces != null) {
            for (Iterator i = namespaces.iterator(); i.hasNext(); ) {
                Mapping mapping = (Mapping)i.next();
                context.registerPrefixForURI(mapping.getPrefix(),
                                             mapping.getNamespaceURI());
            }
        }

        Enumeration enumeration;

        // Output <SOAP-ENV:Envelope>
        context.startElement(new QName(soapConstants.getEnvelopeURI(),
                                       Constants.ELEM_ENVELOPE), attributes);

        // Output non-SOAPHeader and non-SOAPBody stuff.
        Iterator i = getChildElements();
        while (i.hasNext()) {
            MessageElement element = (MessageElement)i.next();
            if(element instanceof SOAPHeader ||
               element instanceof SOAPBody)
                continue;
            element.output(context);
        }

        // Output headers
        if (header != null) {
            header.outputImpl(context);
        }

        // Output body
        if (body != null) {
            body.outputImpl(context);
        }

        // Output trailers
        enumeration = trailers.elements();
        while (enumeration.hasMoreElements()) {
            MessageElement element = (MessageElement)enumeration.nextElement();
            element.output(context);
            // Output this independent element
        }

        // Output </SOAP-ENV:Envelope>
        context.endElement();

        context.setPretty(oldPretty);
    }

    /**
     * Get the soap constants for this envelope
     * @return
     */
    public SOAPConstants getSOAPConstants() {
        return soapConstants;
    }

    /**
     * Set the soap constants for this envelope
     * @param soapConstants
     */
    public void setSoapConstants(SOAPConstants soapConstants) {
        this.soapConstants = soapConstants;
    }

    /**
     * Get the schema version for this envelope
     * @return
     */
    public SchemaVersion getSchemaVersion() {
        return schemaVersion;
    }
    /**
     * Set the schema version for this envelope
     * @param schemaVersion
     */
    public void setSchemaVersion(SchemaVersion schemaVersion) {
        this.schemaVersion = schemaVersion;
    }

    /**
     * Add a soap body if one does not exist
     * @return
     * @throws SOAPException
     */
    public javax.xml.soap.SOAPBody addBody() throws SOAPException {
        if (body == null) {
            body = new SOAPBody(this, soapConstants);
            _isDirty = true;
            return body;
        } else {
            throw new SOAPException(Messages.getMessage("bodyPresent"));
        }
    }

    /**
     * Add a soap header if one does not exist
     * @return
     * @throws SOAPException
     */
    public javax.xml.soap.SOAPHeader addHeader() throws SOAPException {
        if (header == null) {
            header = new SOAPHeader(this, soapConstants);
            return header;
        } else {
            throw new SOAPException(Messages.getMessage("headerPresent"));
        }
    }

    /**
     * create a Name given the local part
     * @param localName
     * @return
     * @throws SOAPException
     */
    public javax.xml.soap.Name createName(String localName)
        throws SOAPException {
        return new PrefixedQName(null, localName,  null);
    }

    /**
     * Create a name given local part, prefix and uri
     * @param localName
     * @param prefix
     * @param uri
     * @return
     * @throws SOAPException
     */
    public javax.xml.soap.Name createName(String localName,
                                          String prefix,
                                          String uri)
        throws SOAPException {
        return new PrefixedQName(uri, localName, prefix);
    }

    /**
     * Get the soap body
     * @return
     * @throws SOAPException
     */
    public javax.xml.soap.SOAPBody getBody() throws SOAPException {
        return body;
    }

    /**
     * Get the soap header
     * @return
     * @throws SOAPException
     */
    public javax.xml.soap.SOAPHeader getHeader() throws SOAPException {
        return header;
    }

    public void setSAAJEncodingCompliance(boolean comply) {
        this.body.setSAAJEncodingCompliance(comply);
    }
}
TOP

Related Classes of org.apache.axis.message.SOAPEnvelope

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.