soapserializer.java

Xfire文件 用于开发web service 的一个开源工具 很好用的

package org.codehaus.xfire.soap;

import java.util.List;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;

import org.codehaus.xfire.MessageContext;
import org.codehaus.xfire.exchange.AbstractMessage;
import org.codehaus.xfire.exchange.InMessage;
import org.codehaus.xfire.exchange.MessageSerializer;
import org.codehaus.xfire.exchange.OutMessage;
import org.codehaus.xfire.fault.XFireFault;
import org.codehaus.xfire.util.jdom.StaxSerializer;
import org.jdom.Element;

public class SoapSerializer
    implements MessageSerializer
{
    public static final String SERIALIZE_PROLOG = "xfire.serializeProlog";
    private MessageSerializer serializer;

    public SoapSerializer(MessageSerializer serializer)
    {
        this.serializer = serializer;
    }

    public MessageSerializer getSerializer()
    {
        return serializer;
    }

    /**
     * Sends a message wrapped in a SOAP Envelope and Body.
     * 
     * @param message
     * @param writer
     * @param context
     * @throws XFireFault
     */
    public void writeMessage(OutMessage message, XMLStreamWriter writer, MessageContext context)
        throws XFireFault
    {
        try
        {
            QName env = message.getSoapVersion().getEnvelope();

            boolean serializeProlog = Boolean.TRUE.equals(context.getProperty(SoapSerializer.SERIALIZE_PROLOG));
            if (Boolean.TRUE.equals(context.getProperty(SoapSerializer.SERIALIZE_PROLOG)))
                writer.writeStartDocument(message.getEncoding(), "1.0");
            
            writer.setPrefix(env.getPrefix(), env.getNamespaceURI());
            writer.setPrefix(SoapConstants.XSD_PREFIX, SoapConstants.XSD);
            writer.setPrefix(SoapConstants.XSI_PREFIX, SoapConstants.XSI_NS);
            writer.writeStartElement(env.getPrefix(),
                                     env.getLocalPart(),
                                     env.getNamespaceURI());
            writer.writeNamespace(env.getPrefix(), env.getNamespaceURI());
            
            writer.writeNamespace(SoapConstants.XSD_PREFIX, SoapConstants.XSD);
            writer.writeNamespace(SoapConstants.XSI_PREFIX, SoapConstants.XSI_NS);

            if (message.getHeader() != null && message.getHeader().getContentSize() > 0)
            {
                QName header = message.getSoapVersion().getHeader();
                writer.writeStartElement(header.getPrefix(),
                                         header.getLocalPart(),
                                         header.getNamespaceURI());
                
                writeHeaders(message, writer);
                
                writer.writeEndElement();
            }
            
            QName body = message.getSoapVersion().getBody();
            writer.writeStartElement(body.getPrefix(),
                                     body.getLocalPart(),
                                     body.getNamespaceURI());
    
            serializer.writeMessage(message, writer, context);
    
            writer.writeEndElement();
            writer.writeEndElement();
            
            if (serializeProlog)
                writer.writeEndDocument();
    
            writer.flush();
        }
        catch (XMLStreamException e)
        {
            throw new XFireFault("Couldn't write message.", e, XFireFault.RECEIVER);
        }
    }

    protected void writeHeaders(AbstractMessage msg, XMLStreamWriter writer)
            throws XMLStreamException
    {
        StaxSerializer ser = new StaxSerializer();

        List elements = msg.getHeader().getChildren();
        for (int i = 0; i < elements.size(); i++)
        {
            Element e = (Element) elements.get(i);
            
            ser.writeElement(e, writer);
        }
    }

    public void readMessage(InMessage message, MessageContext context)
        throws XFireFault
    {
        throw new UnsupportedOperationException();
    }
}